from django.db import models
from rex.models.module import BasicModule
from rex.models.location import Country
from rest_framework import viewsets, permissions
from rex.models.module import BasicModuleSerializer


####################
# CountryVisaAdministrative
###################

class CountryVisaAdministrative(BasicModule):
    country = models.OneToOneField(
        Country, on_delete=models.CASCADE, related_name='country_visa_administrative', primary_key=True)


class CountryVisaAdministrativeSerializer(BasicModuleSerializer):
    class Meta:
        model = CountryVisaAdministrative
        fields = '__all__'


# TODO switch to dict
class CountryVisaAdministrativeViewSet(viewsets.ModelViewSet):
    permission_classes = (permissions.DjangoModelPermissions,)
    queryset = CountryVisaAdministrative.objects.all()  # pylint: disable=E1101
    serializer_class = CountryVisaAdministrativeSerializer


#####################
#####################


####################
# CountryHealthInsurance
###################


class CountryHealthInsurance(BasicModule):
    country = models.OneToOneField(
        Country, on_delete=models.CASCADE, related_name='country_health_insurances', primary_key=True)


class CountryHealthInsuranceSerializer(BasicModuleSerializer):
    class Meta:
        model = CountryHealthInsurance
        fields = '__all__'


# TODO switch to dict
class CountryHealthInsuranceViewSet(viewsets.ModelViewSet):
    permission_classes = (permissions.DjangoModelPermissions,)
    queryset = CountryHealthInsurance.objects.all()  # pylint: disable=E1101
    serializer_class = CountryHealthInsuranceSerializer

#####################
#####################


####################
# Other Insurance
###################


class CountryOtherInsurance(BasicModule):
    country = models.OneToOneField(
        Country, on_delete=models.CASCADE, related_name='country_other_insurances', primary_key=True)


class CountryOtherInsuranceSerializer(BasicModuleSerializer):
    class Meta:
        model = CountryOtherInsurance
        fields = '__all__'


# TODO switch to dict
class CountryOtherInsuranceViewSet(viewsets.ModelViewSet):
    permission_classes = (permissions.DjangoModelPermissions,)
    queryset = CountryOtherInsurance.objects.all()  # pylint: disable=E1101
    serializer_class = CountryOtherInsuranceSerializer

#####################
#####################


####################
# CountryCulture
###################

class CountryCulture(BasicModule):
    country = models.OneToOneField(
        Country, on_delete=models.CASCADE, related_name='country_culture', primary_key=True)


class CountryCultureSerializer(BasicModuleSerializer):
    class Meta:
        model = CountryCulture
        fields = '__all__'


class CountryCultureViewSet(viewsets.ModelViewSet):  # TODO switch to dict
    permission_classes = (permissions.DjangoModelPermissions,)
    queryset = CountryCulture.objects.all()  # pylint: disable=E1101
    serializer_class = CountryCultureSerializer

#####################
#####################


####################
# CountryTransport
###################

class CountryTransport(BasicModule):
    country = models.OneToOneField(
        Country, on_delete=models.CASCADE, related_name='country_transport', primary_key=True)


class CountryTransportSerializer(BasicModuleSerializer):
    class Meta:
        model = CountryTransport
        fields = '__all__'


class CountryTransportViewSet(viewsets.ModelViewSet):  # TODO switch to dict
    permission_classes = (permissions.DjangoModelPermissions,)
    queryset = CountryTransport.objects.all()  # pylint: disable=E1101
    serializer_class = CountryTransportSerializer


#####################
#####################


####################
# CountryOtherStuff
###################

class CountryOtherStuff(BasicModule):
    country = models.OneToOneField(
        Country, on_delete=models.CASCADE, related_name='country_other_stuff', primary_key=True)


class CountryOtherStuffSerializer(BasicModuleSerializer):
    class Meta:
        model = CountryOtherStuff
        fields = '__all__'


# TODO switch to dict
class CountryOtherStuffViewSet(viewsets.ModelViewSet):
    permission_classes = (permissions.DjangoModelPermissions,)
    queryset = CountryOtherStuff.objects.all()  # pylint: disable=E1101
    serializer_class = CountryOtherStuffSerializer