add_confirmed_at command

This commit is contained in:
alex 2020-01-31 11:25:38 +03:00
parent c3cc87507e
commit d3bf68ab09

View File

@ -0,0 +1,26 @@
from django.core.management.base import BaseCommand
from tqdm import tqdm
from account.models import User, UserRole, Role
from transfer.models import OwnershipAffs, Accounts
from establishment.models import Establishment
class Command(BaseCommand):
help = """Add confirmed date to User."""
def handle(self, *args, **kwarg):
update_users = []
old_users = Accounts.objects.filter(confirmed_at__isnull=False)
for old_user in tqdm(old_users, desc='find users for update confirmed_at field'):
try:
user = User.objects.get(email=old_user.email)
except User.DoesNotExist:
continue
else:
user.confirmed_at = old_user.confirmed_at
update_users.append(user)
User.objects.bulk_update(update_users, ['confirmed_at', ])
self.stdout.write(self.style.WARNING(f'Updated users: {len(update_users)}'))