U dKN@sddlZddlZddlmZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZddlmZddlmZddlmZmZddlmZmZdd lmZd d l m!Z!erdd l"m#Z#z,ddl$Z$e$%Z&ee'e'fed ddZ(Wn6e)k r&ddl*Z*ee'e'fed ddZ(YnXeZ+GdddeZ,eddZ-GdddeZ.ede.fZ/ee'e0dfZ1Gddde Z2Gddde Z3Gddde!Z4e4Z5dS)N)contextmanager)NullTranslations) TYPE_CHECKINGAnyCallableDictIterableList NamedTupleOptionalSetTupleUnioncastoverload)Local) force_str)override trans_real)Literal TypedDict)settings) CountriesBase) StrPromise)itemreturncCst|dSNr)collatorsort_keyrr!=/tmp/pip-unpacked-wheel-dnjn05s9/django_countries/__init__.pyr(srcCstd|ddddS)NZNFKDrasciiignore) unicodedata normalizeencodedecoder r!r!r"r0scs$eZdZeedfdd ZZS)EmptyFallbackTranslator)messagercsttddsdSt|S)NfallbackT)getattr_translation_statesupergettext)selfr* __class__r!r"r0<s zEmptyFallbackTranslator.gettext)__name__ __module__ __qualname__strr0 __classcell__r!r!r2r"r);sr)ccsXtjsdVdSt}|j}t|ts:t}||_||_dt_z dVW5dt_XdS)NFT) rZUSE_I18Nrcatalog _fallback isinstancer)r.r+)r9Zoriginal_fallbackZempty_fallback_translatorr!r!r"no_translation_fallbackCs  r<c@s6eZdZUded<ded<eed<eed<eed<dS) ComplexCountryNamernamezList[StrPromise]namesalpha3numericioc_codeN)r4r5r6__annotations__r7intr!r!r!r"r=Ws r=rc@s"eZdZUeed<eeed<dS)AltCodesr@rAN)r4r5r6r7rCr rDr!r!r!r"rEcs rEc@s,eZdZUeed<eed<edddZdS) CountryTuplecoder>rcCsd|jd|jdS)z Display the repr as a standard tuple for better backwards compatibility with outputting this in a template. (z, )rGr>r1r!r!r"__repr__lszCountryTuple.__repr__N)r4r5r6r7rCrMr!r!r!r"rFhs rFc @s,eZdZUdZeeefed<eeefed<edddZ e eeefddd Z e j d d Z e eeefdd d Z e eeefdd dZe ddZd;eeeedddZdeeedfed/d0d-Z eedd1d2Z!d3d4Z"d5d6Z#d7d8Z$d9d:Z%dS)? Countriesz An object containing a list of ISO3166-1 countries. Iterating this object will return the countries as namedtuples (of the country ``code`` and ``name``), sorted by name. _countries _alt_codes)optioncCs,t||d}|dk r|Sttd|S)z Get a configuration option, trying the options attribute first and falling back to a Django project setting. NZ COUNTRIES_)r-rupper)r1rQvaluer!r!r" get_options zCountries.get_optionrHcCst|ds|d}d}|rDt|tsD|D]}t|tr,d}qDq,i|_|r`|r`t||_ndddlm}t|}|ri|_|D]0}t|tr|||j|<q|\}}||j|<qn | |_|dr|j D]\}} ||jkr| |j|<q|d} | r@t t tttd ff|j } | | d d | D|_|dr|j D],}||jkrR|| krR||g|j|<qR|j D]8\}} ||jkr|| kr|j|g} | | qg|_|d pg}|D](}||}||jkr|j|q|jS) z Return the a dictionary of countries, modified by any overriding options. The result is cached so future lookups are less work intensive. rOonlyTFr) COUNTRIESZ common_namesrNcSsi|]\}}|dk r||qSNr!).0rGr>r!r!r" sz'Countries.countries..first)hasattrrTr;dictr7_shadowed_namesrOdjango_countries.datarVcopyZ COMMON_NAMESitemsrrr CountryNameupdateZ OLD_NAMES setdefaultextendcountries_firstalpha2append)r1rUZ only_choicesrrVZcountries_dictkeyrSrGr>rrOr?Zcountry_shadowedrZr!r!r" countriessn              zCountries.countriescCs<t|dr|`t|dr|`t|dr*|`t|dr8|`dS)z Reset the countries cache in case for some crazy reason the settings or internal options change. But surely no one is crazy enough to do that, right? rOrP _ioc_codesr]N)r[rOrPrjr]rLr!r!r"ris    cCst|dsddlm}||_d}|jD]x\}}t|tr*d|ksLd|kr*|s`|j|_d}|j |d\}}d|kr|d}d|kr|d}t |||j|<q*|jS) NrPr) ALT_CODESFr@rAT)r,N) r[r^rkrPrir`r;r\r_getrE)r1rkalteredrGcountryr@rAr!r!r" alt_codess(    zCountries.alt_codescCsnt|dshddlm}||_d}|jD]<\}}t|tr*d|kr*|sX|j|_d}|d|j|<q*|jS)Nrjr) ISO_TO_IOCFrBT) r[Zdjango_countries.ioc_datarprjrir`r;r\r_)r1rprmrGrnr!r!r" ioc_codess   zCountries.ioc_codescCst|dds|j|jS)Nr]F)r-rir]rLr!r!r"shadowed_names s zCountries.shadowed_namesN)rG ignore_firstccsl|j|}t|tr2d|kr&|d}q8|dg}n|g}|rP||krP|dd}|D]}|||VqTdS)zA Return translated countries for a country code. r?r>rN)rir;r\translate_pair)r1rGrsrnr?r>r!r!r"translate_codes      zCountries.translate_coderKc Cs|dkr|j|}t|trPd|krB|ddd}|dd}q^g}|d}n|j|g}|rt0t|}|s|D]}t|}|rz|}qqzW5QRX|st|}nt|}t||S)z Force a country to the current activated translation. :returns: ``CountryTuple(code, translated_country_name)`` namedtuple Nr?rrr>)rir;r\rrrlr<rrF)r1rGr>Zfallback_namesZ country_nameZ fallback_nameZfallback_name_strr!r!r"rt!s,    zCountries.translate_pairc#sj}fddjD}dr0t|td}|EdHjr^d}|r^tdt|Vdrldnjttj fd d|D}t|tdEdHdS) a Iterate through countries, sorted by name. Each country record consists of a namedtuple of the two letter ISO3166-1 country ``code`` and short ``name``. The sorting happens based on the thread's current translation. Countries that are in ``settings.COUNTRIES_FIRST`` will be displayed before any sorted countries (in the order provided), and are only repeated in the sorted list if ``settings.COUNTRIES_FIRST_REPEAT`` is ``True``. The first countries can be separated from the sorted list by the value provided in ``settings.COUNTRIES_FIRST_BREAK``. c3s|]}|VqdSrW)rtrXrGrLr!r" Ysz%Countries.__iter__..Z first_sort)rhN first_breakr,Z first_repeatc3s|]}|VqdSrW)rurvrsr1r!r"rwhs) rirerTsortedrrFrtuple itertoolschain from_iterable)r1rirerxr!ryr"__iter__Ds     zCountries.__iter__)rGrcst|}|r*t|fdd}n(t|dkrN|tdfdd }nd}|r~d}|jD]\}}||rd|}q~qd||jkr|SdS)af Return the normalized country code when passed any type of ISO 3166-1 country code. Overridden countries objects may actually have country codes that are not two characters (for example, "GB-WLS"), so the returned length of the code is not guaranteed. If no match is found, returns an empty string. cs |dkSrr!ro)lookup_numericr!r"findszCountries.alpha2..findrHcs |dkS)Nrr!r) lookup_alpha3r!r"rsNr,) rrRisdigitrDlenboolror`ri)r1rGZcode_strrrfror!)rrr"rfps"   zCountries.alpha2cCs&||}||jkrdS||dS)zz Return the name of a country, based on the code. If no match is found, returns an empty string. r,r)rfrirtr1rGrfr!r!r"r>s  zCountries.nameFenT)regexlanguage insensitive)rnrrrrcCsdSrWr!r1rnrrrr!r!r"by_names zCountries.by_name)rrcCsdSrWr!rr!r!r"rs c Csrt}|rt||otj}n |r*|}t|*|jD]\}}t|t rrd|krf|d} qx|dg} n|g} | D]l} |r| t | r| |q||r|| kr|W5QRSq||| kr||W5QRSq|||j kr@|j |D]X} |r$| t | rT| |n0|r2| } || kr|W5QRSqq@W5QRX|rn|SdS)a  Fetch a country's ISO3166-1 two letter country code from its name. An optional language parameter is also available. Warning: This depends on the quality of the available translations. If no match is found, returns an empty string. If ``regex`` is set to True, then rather than returning a string containing the matching country code or an empty string, a set of matching country codes is returned. If ``insensitive`` is set to False (True by default), then the search will be case sensitive. ..warning:: Be cautious about relying on this returning a country code (especially with any hard-coded string) since the ISO names of countries may change over time. r?r>r,)setrecompile IGNORECASElowerrrir`r;r\searchr7addrr) r1rnrrrZ code_listre_matchrGZ check_countryZ check_namesr>Z shadowed_namer!r!r"rsB         (cCs>||}z|j|d}Wntk r4d}YnX|prrrrr rr@rDrArBrrrrr!r!r!r"rNts  I  #,%   B    rN)6r|r contextlibrr0rtypingrrrrrr r r r r rrrZ asgiref.localrZdjango.utils.encodingrZdjango.utils.translationrrZtyping_extensionsrrZdjango_countries.confrbaserZdjango_stubs_extrZpyucaZCollatorrr7r ImportErrorr%r.r)r<r=rarDrrErFrNrir!r!r!r"sB  <        b