diff --git a/djangoldp/factories.py b/djangoldp/factories.py new file mode 100644 index 0000000000000000000000000000000000000000..172e289b5b49acfaf4a80e711ba249cd77c2d3af --- /dev/null +++ b/djangoldp/factories.py @@ -0,0 +1,14 @@ +import factory +from django.contrib.auth.models import User +from django.db.models.signals import post_save + +@factory.django.mute_signals(post_save) +class UserFactory(factory.django.DjangoModelFactory): + class Meta: + model = User + + username = factory.Faker('user_name') + first_name = factory.Faker('first_name') + last_name = factory.Faker('last_name') + email = factory.Faker('email') + password = factory.PostGenerationMethodCall('set_password', 'totototo') diff --git a/djangoldp/management/commands/__init__.py b/djangoldp/management/commands/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/djangoldp/management/commands/mock_user.py b/djangoldp/management/commands/mock_user.py new file mode 100644 index 0000000000000000000000000000000000000000..30b4373e8dcd7d0bed9c4a8277bac1a3fbd49c4b --- /dev/null +++ b/djangoldp/management/commands/mock_user.py @@ -0,0 +1,13 @@ +from django.core.management.base import BaseCommand, CommandError +from djangoldp.factories import UserFactory + +class Command(BaseCommand): + help = 'Mock data' + + def add_arguments(self, parser): + parser.add_argument('--size', type=int, default=0, help='Number of user to create') + + def handle(self, *args, **options): + UserFactory.create_batch(size=options['size']); + + self.stdout.write(self.style.SUCCESS('Successful data mock install'))