from django.core.management.base import BaseCommand

from establishment.models import Establishment, SocialNetwork
from transfer.models import EstablishmentInfos


class Command(BaseCommand):
    help = 'Add social links values from old db to new db'

    def handle(self, *args, **kwargs):
        count = 0

        queryset = EstablishmentInfos.objects.exclude(
            establishment_id__isnull=True
        ).values_list('id', 'establishment_id', 'facebook', 'twitter', 'instagram')

        for id, es_id, facebook, twitter, instagram in queryset:
            try:
                establishment = Establishment.objects.get(old_id=es_id)
            except Establishment.DoesNotExist:
                continue
            except Establishment.MultipleObjectsReturned:
                establishment = Establishment.objects.filter(old_id=es_id).first()
            else:
                if facebook:
                    if 'facebook.com/' not in facebook:
                        facebook = 'https://www.facebook.com/' + facebook
                    obj, _ = SocialNetwork.objects.get_or_create(
                        old_id=id,
                        establishment=establishment,
                        title='facebook',
                        url=facebook,
                    )
                    count += 1
                if twitter:
                    if 'twitter.com/' not in twitter:
                        twitter = 'https://www.twitter.com/' + twitter
                    obj, _ = SocialNetwork.objects.get_or_create(
                        old_id=id,
                        establishment=establishment,
                        title='twitter',
                        url=twitter,
                    )
                    count += 1
                if instagram:
                    if 'instagram.com/' not in instagram:
                        instagram = 'https://www.instagram.com/' + instagram
                    obj, _ = SocialNetwork.objects.get_or_create(
                        old_id=id,
                        establishment=establishment,
                        title='instagram',
                        url=instagram,
                    )
                    count += 1

        self.stdout.write(self.style.WARNING(f'Created/updated {count} objects.'))