from django.core.management.base import BaseCommand
from django.utils.translation import activate
from service.models import Service,LinkedService
from country.models import Country
from exposition.models import Exposition
from conference.models import Conference


class Command(BaseCommand):
    def handle(self, *args, **options):
        activate('ru')
        LinkedService.objects.all().delete()
        services = Service.objects.all()
        for service in services:
            linked = LinkedService()
            linked.service = service
            linked.save()
            if Country.objects.language().filter(services=getattr(Country.services, service.url)).count() == Country.objects.language().count():
                linked.all_countries = True
            else:
                linked.countries = Country.objects.language().filter(services=getattr(Country.services, service.url))
            linked.expositions = Exposition.objects.language().filter(services=getattr(Exposition.services, service.url))
            linked.conferences = Conference.objects.language().filter(services=getattr(Conference.services, service.url))
            linked.save()