from django.core.management.base import BaseCommand

from ...lib.qr_code_gen import generate_qr_code
from ...models import Profile


class Command(BaseCommand):
    help = 'Generate new QR codes'

    def handle(self, *args, **kwargs):
        for profile in Profile.objects.all():
            if profile.pin:
                profile.qr_code = generate_qr_code(profile.pin)
                profile.save()