lib.itmens/users/management/commands/refresh_following.py
2023-01-11 19:13:02 -05:00

19 lines
562 B
Python

from django.core.management.base import BaseCommand
from users.models import User
from datetime import timedelta
from django.utils import timezone
from tqdm import tqdm
class Command(BaseCommand):
help = "Refresh following data for all users"
def handle(self, *args, **options):
count = 0
for user in tqdm(User.objects.all()):
user.following = user.get_following_ids()
if user.following:
count += 1
user.save(update_fields=["following"])
print(f"{count} users updated")