from django.core.management.base import BaseCommand

from university.models import User
from university.utils.avatar import generate_lettere_avatar_url


class Command(BaseCommand):
    help = "Generate lettere avatar_url for all users who do not have avatar."

    def handle(self, *args, **options):
        for user in User.objects.filter(avatar_url__isnull=True):
            random_url = generate_lettere_avatar_url(user.username)
            user.avatar_url = random_url
            user.save(update_fields=['avatar_url'])
            print('{}: {}'.format(user.username, user.avatar_url))