diff --git a/idhub/management/__init__.py b/idhub/management/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/idhub/management/commands/__init__.py b/idhub/management/commands/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/idhub/management/commands/initial_datas.py b/idhub/management/commands/initial_datas.py new file mode 100644 index 0000000..71506c6 --- /dev/null +++ b/idhub/management/commands/initial_datas.py @@ -0,0 +1,39 @@ +from django.core.management.base import BaseCommand, CommandError +from django.contrib.auth import get_user_model +from idhub.models import Organization + + +User = get_user_model() + + +class Command(BaseCommand): + help = "Insert minimum datas for the project" + + def handle(self, *args, **kwargs): + admin = 'admin@example.org' + pw_admin = '1234' + + user = 'user1@example.org' + pw_user = '1234' + organization = [ + ("ExO", "https://verify.exo.cat"), + ("Somos Connexión", "https://verify.somosconexion.coop") + ] + + # self.create_admin_users(admin, pw_admin) + self.create_users(user, pw_user) + for o in organization: + self.create_organizations(*o) + + def create_admin_users(self, email, password): + User.objects.create_superuser(email=email, password=password) + + + def create_users(self, email, password): + u= User.objects.create(email=email, password=password) + u.set_password(password) + u.save() + + + def create_organizations(self, name, url): + Organization.objects.create(name=name, url=url) diff --git a/initial_datas.py b/initial_datas.py deleted file mode 100644 index 01cd9ea..0000000 --- a/initial_datas.py +++ /dev/null @@ -1,41 +0,0 @@ -import os -import django -os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'trustchain_idhub.settings') # noqa -django.setup() # noqa - -from django.contrib.auth import get_user_model -from idhub.models import Organization - - -User = get_user_model() - - -def create_admin_users(email, password): - User.objects.create_superuser(email=email, password=password) - - -def create_users(email, password): - u= User.objects.create(email=email, password=password) - u.set_password(password) - u.save() - - -def create_organizations(): - Organization.objects.create(name="ExO", url="https://verify.exo.cat") - Organization.objects.create(name="Somos Connexión", url="https://verify.somosconexion.coop") - - -def main(): - admin = 'admin@example.org' - pw_admin = '1234' - - user = 'user1@example.org' - pw_user = '1234' - - # create_admin_users(admin, pw_admin) - create_users(user, pw_user) - create_organizations() - - -if __name__ == '__main__': - main()