o 1iy @sFddlmZddlmZddlmZddlmZGdddejZ dS))Any) force_str) serializers) countriescsJeZdZfddZddZdefddZdd Zd d Zd d Z Z S) CountryFieldcsZ|dd|_|dd|_|dd}|pt|_tj|jg|Ri||dS)N country_dict name_onlyr)poprrrsuper__init___setup_spectacular_annotation)selfargskwargsfield_countries __class__c/var/www/html/humari/django-venv/lib/python3.10/site-packages/django_countries/serializer_fields.pyr s   zCountryField.__init__cCsT|j|}|s|jr dSdS|jrt|j|S|js|S|t|j|dS)Ncodename)ralpha2 allow_nullrrrr)r objrrrrto_representations zCountryField.to_representationdatacCsj|js|dkr|jd|dt|tr|d}|j|}|r3|s3|jt|}|s3|jd|d|S)Nrinvalid_choice)inputr) allow_blankfail isinstancedictgetrrby_namer)r rcountryrrrto_internal_value$s   zCountryField.to_internal_valuecCs<dddddddidddgd }|jrd |dd igi}|S) z.Return schema for country_dict representation.objectstring)type minLength maxLengthr+rrr)r+ propertiesrequiredoneOfnullrr schemarrr_get_country_dict_schema1s z%CountryField._get_country_dict_schemacCs"ddi}|jrd|ddigi}|S)z+Return schema for name_only representation.r+r)r0r1r2r3rrr_get_name_only_schema?sz"CountryField._get_name_only_schemacCs<|jr|}d|i|_dS|jr|}d|i|_dSdS)ab Set up schema annotation for drf-spectacular. This sets the _spectacular_annotation attribute that drf-spectacular checks when generating OpenAPI schemas. When country_dict=True or name_only=True, we override the default ChoiceField enum schema with the appropriate schema for the actual representation. fieldN)rr5_spectacular_annotationrr6r3rrrr Fs z*CountryField._setup_spectacular_annotation) __name__ __module__ __qualname__r rrr'r5r6r __classcell__rrrrr s    rN) typingrdjango.utils.encodingrrest_frameworkrdjango_countriesr ChoiceFieldrrrrrs