| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- import random
- import time
- from django.contrib.auth import get_user_model
- from django.core.management.base import BaseCommand
- from ....core.management.progressbar import show_progress
- User = get_user_model()
- class Command(BaseCommand):
- help = "Adds random followers for testing purposes"
- def handle(self, *args, **options):
- total_users = User.objects.count()
- message = "Adding fake followers to %s users...\n"
- self.stdout.write(message % total_users)
- total_followers = 0
- processed_count = 0
- start_time = time.time()
- show_progress(self, processed_count, total_users)
- for user in User.objects.iterator():
- user.followed_by.clear()
- if random.randint(1, 100) > 10:
- processed_count += 1
- show_progress(self, processed_count, total_users)
- continue # 10% active users
- users_to_add = random.randint(1, total_users / 5)
- random_queryset = User.objects.exclude(id=user.id).order_by("?")
- while users_to_add > 0:
- new_follower = random_queryset[:1][0]
- if not new_follower.is_following(user):
- user.followed_by.add(new_follower)
- users_to_add -= 1
- total_followers += 1
- processed_count += 1
- show_progress(self, processed_count, total_users)
- self.stdout.write("\nSyncing models...")
- for user in User.objects.iterator():
- user.followers = user.followed_by.count()
- user.following = user.follows.count()
- user.save(update_fields=["followers", "following"])
- total_time = time.time() - start_time
- total_humanized = time.strftime("%H:%M:%S", time.gmtime(total_time))
- message = "\nSuccessfully added %s fake followers in %s"
- self.stdout.write(message % (total_followers, total_humanized))
|
