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)}'))