Browse Source

tweaked faking users a little, added total time spent on generation to final message

Rafał Pitoń 9 years ago
parent
commit
4449ef96d6

+ 5 - 2
misago/faker/management/commands/createfakecategories.py

@@ -41,7 +41,7 @@ class Command(BaseCommand):
         message = 'Creating %s fake categories...\n'
         message = 'Creating %s fake categories...\n'
         self.stdout.write(message % fake_cats_to_create)
         self.stdout.write(message % fake_cats_to_create)
 
 
-        message = '\n\nSuccessfully created %s fake categories'
+        message = '\n\nSuccessfully created %s fake categories in %s'
 
 
         created_count = 0
         created_count = 0
         start_time = time.time()
         start_time = time.time()
@@ -82,4 +82,7 @@ class Command(BaseCommand):
                 self, created_count, fake_cats_to_create, start_time)
                 self, created_count, fake_cats_to_create, start_time)
 
 
         acl_version.invalidate()
         acl_version.invalidate()
-        self.stdout.write(message % created_count)
+
+        total_time = time.time() - start_time
+        total_humanized = time.strftime('%H:%M:%S', time.gmtime(total_time))
+        self.stdout.write(message % (created_count, total_humanized))

+ 6 - 2
misago/faker/management/commands/createfakefollowers.py

@@ -1,4 +1,5 @@
 import random
 import random
+import time
 
 
 from django.contrib.auth import get_user_model
 from django.contrib.auth import get_user_model
 from django.core.management.base import BaseCommand
 from django.core.management.base import BaseCommand
@@ -16,11 +17,12 @@ class Command(BaseCommand):
         message = 'Adding fake followers to %s users...\n'
         message = 'Adding fake followers to %s users...\n'
         self.stdout.write(message % total_users)
         self.stdout.write(message % total_users)
 
 
-        message = '\nSuccessfully added %s fake followers'
+        message = '\nSuccessfully added %s fake followers in %s'
 
 
         total_followers = 0
         total_followers = 0
         processed_count = 0
         processed_count = 0
 
 
+        start_time = time.time()
         show_progress(self, processed_count, total_users)
         show_progress(self, processed_count, total_users)
         for user in User.objects.iterator():
         for user in User.objects.iterator():
             user.followed_by.clear()
             user.followed_by.clear()
@@ -48,4 +50,6 @@ class Command(BaseCommand):
             user.following = user.follows.count()
             user.following = user.follows.count()
             user.save(update_fields=['followers', 'following'])
             user.save(update_fields=['followers', 'following'])
 
 
-        self.stdout.write(message % total_followers)
+        total_time = time.time() - start_time
+        total_humanized = time.strftime('%H:%M:%S', time.gmtime(total_time))
+        self.stdout.write(message % (total_followers, total_humanized))

+ 4 - 2
misago/faker/management/commands/createfakethreads.py

@@ -36,7 +36,7 @@ class Command(BaseCommand):
 
 
         self.stdout.write('Creating fake threads...\n')
         self.stdout.write('Creating fake threads...\n')
 
 
-        message = '\nSuccessfully created %s fake threads'
+        message = '\nSuccessfully created %s fake threads in %s'
 
 
         created_threads = 0
         created_threads = 0
         start_time = time.time()
         start_time = time.time()
@@ -149,4 +149,6 @@ class Command(BaseCommand):
             category.synchronize()
             category.synchronize()
             category.save()
             category.save()
 
 
-        self.stdout.write(message % created_threads)
+        total_time = time.time() - start_time
+        total_humanized = time.strftime('%H:%M:%S', time.gmtime(total_time))
+        self.stdout.write(message % (created_threads, total_humanized))

+ 16 - 5
misago/faker/management/commands/createfakeusers.py

@@ -11,6 +11,7 @@ from faker import Factory
 
 
 from misago.core.management.progressbar import show_progress
 from misago.core.management.progressbar import show_progress
 from misago.users.models import Rank
 from misago.users.models import Rank
+from misago.users.avatars import dynamic, gallery, get_avatar_hash
 
 
 
 
 class Command(BaseCommand):
 class Command(BaseCommand):
@@ -33,7 +34,7 @@ class Command(BaseCommand):
         message = 'Creating %s fake user accounts...\n'
         message = 'Creating %s fake user accounts...\n'
         self.stdout.write(message % fake_users_to_create)
         self.stdout.write(message % fake_users_to_create)
 
 
-        message = '\n\nSuccessfully created %s fake user accounts'
+        message = '\n\nSuccessfully created %s fake user accounts in %s'
 
 
         created_count = 0
         created_count = 0
         start_time = time.time()
         start_time = time.time()
@@ -44,9 +45,17 @@ class Command(BaseCommand):
                     'rank': random.choice(ranks),
                     'rank': random.choice(ranks),
                 }
                 }
 
 
-                User.objects.create_user(fake.first_name(), fake.email(),
-                                         'pass123', set_default_avatar=True,
-                                         **kwargs)
+                user = User.objects.create_user(
+                    fake.first_name(), fake.email(), 'pass123',
+                    set_default_avatar=False, **kwargs)
+
+                if random.randint(0, 100) > 90:
+                    dynamic.set_avatar(user)
+                else:
+                    gallery.set_random_avatar(user)
+
+                user.avatar_hash = get_avatar_hash(user)
+                user.save(update_fields=['avatar_hash'])
             except (ValidationError, IntegrityError):
             except (ValidationError, IntegrityError):
                 pass
                 pass
             else:
             else:
@@ -54,4 +63,6 @@ class Command(BaseCommand):
                 show_progress(
                 show_progress(
                     self, created_count, fake_users_to_create, start_time)
                     self, created_count, fake_users_to_create, start_time)
 
 
-        self.stdout.write(message % created_count)
+        total_time = time.time() - start_time
+        total_humanized = time.strftime('%H:%M:%S', time.gmtime(total_time))
+        self.stdout.write(message % (created_count, total_humanized))