class CompanyProfile(models.Model):
    name = models.CharField(_('nombre'), max_length=255, db_index=True)
    email = models.EmailField(_('email de contacto'), max_length=255, blank=True)
    description = HTMLField(_('descripción'), blank=True)
    countries = models.ManyToManyField('countries.Country', blank=True, related_name='countries')

class Country(models.Model):
    name = models.CharField(_('nombre'), max_length=255)
    description = HTMLField(_('descripción'), blank=True)

class CompanyWriteSerializer(serializers.ModelSerializer):
    countries_id = serializers.PrimaryKeyRelatedField(queryset=Country.objects.all(), write_only=True, many=True)
    def create(self, validated_data):
        countries = validated_data.pop('countries_id')
        company = CompanyProfile.objects.register_company(**validated_data,
                                                      country=self.context['request'].user.country.id)
        for c in countries:
            print(c)
            company.countries.add(c)
        return company

    def update(self, instance, validated_data):
        instance.countries = validated_data.get('countries_id', instance.countries)
        instance.save()
        return instance

    class Meta:
        model = CompanyProfile
        fields = ('countries_id', 'countries')