27 lines
938 B
Python
27 lines
938 B
Python
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)}'))
|