| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- import random
- import time
- from django.core.management.base import BaseCommand
- from faker import Factory
- from ....acl.cache import clear_acl_cache
- from ....categories.models import Category, RoleCategoryACL
- from ....core.management.progressbar import show_progress
- class Command(BaseCommand):
- help = "Creates fake categories for dev and testing purposes."
- def add_arguments(self, parser):
- parser.add_argument(
- "categories",
- help="number of categories to create",
- nargs="?",
- type=int,
- default=5,
- )
- parser.add_argument(
- "minlevel",
- help="min. level of created categories",
- nargs="?",
- type=int,
- default=0,
- )
- def handle(self, *args, **options):
- items_to_create = options["categories"]
- min_level = options["minlevel"]
- categories = Category.objects.all_categories(True)
- copy_acl_from = list(Category.objects.all_categories())[0]
- categories = categories.filter(level__gte=min_level)
- fake = Factory.create()
- message = "Creating %s fake categories...\n"
- self.stdout.write(message % items_to_create)
- created_count = 0
- start_time = time.time()
- show_progress(self, created_count, items_to_create)
- while created_count < items_to_create:
- parent = random.choice(categories)
- new_category = Category()
- if random.randint(1, 100) > 75:
- new_category.set_name(fake.catch_phrase().title())
- else:
- new_category.set_name(fake.street_name())
- if random.randint(1, 100) > 50:
- if random.randint(1, 100) > 80:
- new_category.description = "\r\n".join(fake.paragraphs())
- else:
- new_category.description = fake.paragraph()
- new_category.insert_at(parent, position="last-child", save=True)
- copied_acls = []
- for acl in copy_acl_from.category_role_set.all():
- copied_acls.append(
- RoleCategoryACL(
- role_id=acl.role_id,
- category=new_category,
- category_role_id=acl.category_role_id,
- )
- )
- if copied_acls:
- RoleCategoryACL.objects.bulk_create(copied_acls)
- created_count += 1
- show_progress(self, created_count, items_to_create, start_time)
- clear_acl_cache()
- total_time = time.time() - start_time
- total_humanized = time.strftime("%H:%M:%S", time.gmtime(total_time))
- message = "\n\nSuccessfully created %s fake categories in %s"
- self.stdout.write(message % (created_count, total_humanized))
|
