'j+ddlmcmZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZddlZddlmZddlmZmZmZmZddlmZddlmZmZmZmZm Z m!Z!dd l"m#Z# ddl$Z%d Z&n #e'$rd Z&YnwxYw eGd d Z(ed GddZ) de*ddfdZ+de*de(fdZ,d,de*de-de(fdZ.dee*e e*fdee*efddfdZ/de e*de*fdZ0 edz Z1dZ2Gdd eZ3d-d!Z4d-d"Z5d-d#Z6d$e*de)fd%Z7 dej8fd&Z9d'ej8d(e*de:fd)Z;Gd*d+Z<e<Z=dS).N) dataclass) AnyDictIterableListOptionalSetTupleUnioncast)tags) LOCAL_DATAGitHubDownloaderrprintwebdl) LeakWarning) InvalidLocaleMissingReleaseNotInstalledGeoIPExtraUnknownIPLocationUnknownLanguageUnknownTerritory) validate_ipTFceZdZUdZeed<dZeeed<dZeeed<e defdZ de eeffdZ dS) Localez8 Stores locale, region, and script information. languageNregionscriptreturnc@|jr|jd|jS|jS)N-)rrselfs S/home/ice/.hermes/hermes-agent/venv/lib/python3.11/site-packages/camoufox/locale.py as_stringzLocale.as_string,s, ; 4m33dk33 3}cX|jsJ|j|jd}|jr |j|d<|S)zB Converts the locale to a intl config dictionary. )z locale:regionzlocale:languagez locale:script)rrrr$datas r% as_configzLocale.as_config2sD{{![#}   ; 0$(KD ! r') __name__ __module__ __qualname____doc__str__annotations__rrrpropertyr&rr+r'r%rr"sMMM FHSM    FHSM    3X 4S>      r'r)frozencreZdZUdZeed<eed<eed<eed<dZe eed<de ee ffd Z dS) Geolocationz) Stores geolocation information. locale longitudelatitudetimezoneNaccuracyr c|j|j|jd|j}|jr |j|d<|S)zB Converts the geolocation to a config dictionary. )zgeolocation:longitudezgeolocation:latituder:zgeolocation:accuracy)r8r9r:r7r+r;r)s r%r+zGeolocation.as_configLsU &*^$(M   k##%%   = 9+/=D' ( r') r,r-r.r/rr1floatr0r;rrrr+r3r'r%r6r6@s NNNOOOMMM $Hhuo$$$ 4S>      r'r6locr cVtj|rdStj|)zX Verifies that a locale is valid. Takes either language-region or language. N)r checkr invalid_input)r>s r% verify_localerB`s*  z#  %c * **r'r7c$t|tj|}|jst j||jjd}t|d|jjdd| dS)z1 Normalizes and validates a locale code. recordSubtagzSuppress-Script)rrr) rBr tagrrrArr*rget)r7parserrDs r%normalize_localerIjs&Xf  F =2)&111 _ !( +F !}!(+H5zz+,,   r' ignore_regionct|dkrt|S t|S#t$rYnwxYw|rt |t |S t|}tj d|S#t$rYnwxYwtj |)z> Handles a locale input, normalizing it if necessary. )r no_region) lenrISELECTOR from_regionrrBr from_languagerwarnrrrA)r7rJrs r% handle_localerSs  6{{Q''' ##F+++      'fv&&&&))&11 %%%        %f - --s!> A  A 0B B-,B-localesconfigcHt|trd|dD}t|d}||t |dkrdStd|D|d<dS)z$ Handles a list of locales. c6g|]}|Sr3)strip).0r>s r% z"handle_locales..s ===3399;;===r',rNc3BK|]}t|djVdS)T)rJN)rSr&)rYr7s r% z!handle_locales..sC((@F fD111;((((((r'z locale:all) isinstancer0splitrSupdater+rN _join_unique)rTrU intl_locales r%handle_localesrds'3>=='--*<*<===  ++K MM+''))*** 7||a(((JQ(((F<r'seqcbtdfd|DS)z8 Joins a sequence of strings without duplicates z, c3NK|]}|v||V dSN)add)rYxseens r%r^z_join_unique..s5FF1AII!IQIIIIFFr')setjoin)rerks @r%rbrbs5UUD 99FFFFFFF F FFr'zGeoLite2-City.mmdbzP3TERX/GeoLite.mmdbc6eZdZdZdedeefdZddZdS)MaxMindDownloaderz? MaxMind database downloader from a GitHub repository. assetr cL|ddr|dSdS)Nnamez -City.mmdbbrowser_download_url)endswith)r$rps r% check_assetzMaxMindDownloader.check_assets- = ! !, / / 1/0 0tr'Nc td)Nz+Failed to find GeoIP database release asset)rr#s r%missing_asset_errorz%MaxMindDownloader.missing_asset_errorsJKKKr'r N) r,r-r.r/rrr0rurwr3r'r%roros^(3- LLLLLLr'roc2tstddS)z3 Checks if the geoip2 module is available. zOPlease install the geoip extra to use this feature: pip install camoufox[geoip]N) ALLOW_GEOIPrr3r'r% geoip_allowedr{s+  $ ]     r'cttt}t t d5}t |d|ddddS#1swxYwYdS)z0 Downloads the MaxMind GeoIP2 database. wbzDownloading GeoIP database)descbufferN)r{ro MMDB_REPO get_assetopen MMDB_FILEr) asset_urlfs r% download_mmdbrsOOO!),,6688I i   ! -                      s A**A.1A.ctstddSttddS)z. Removes the MaxMind GeoIP2 database. zGeoIP database not found.NzGeoIP database removed.)rexistsrunlinkr3r'r% remove_mmdbrsS     *+++  $%%%%%r'ipc tstt|tjtt5}||}tt|j j  }|j tfddDrtd|t |}t%|tt&|j jtt&|j jtt|j jcdddS#1swxYwYdS)z1 Gets the geolocation for an IP address. c3:K|]}t| VdSrh)getattr)rYattrlocations r%r^z"get_geolocation.. s0^^t78T***^^^^^^r')r8r9 time_zonezUnknown IP location: )r7r8r9r:N)rrrrgeoip2databaseReaderr0cityr registered_countryiso_codeupperranyrrOrPr6r=r8r9r)rreaderresprr7rs @r%get_geolocationrs     OOO   I / / 6{{2T4=>>DDFF= ^^^^7]^^^ ^ ^ B#$@B$@$@AA A%%h//5$-"9::%!788#t}677                      s)C>E44E8;E8cttdz d5}tj|}dddn #1swxYwY| Jd|S)z Fetches supplemental data from the territoryInfo.xml file. Source: https://raw.githubusercontent.com/unicode-org/cldr/master/common/supplemental/supplementalData.xml zterritoryInfo.xmlrbNz Failed to load territoryInfo.xml)rrETXMLread)rr*s r%get_unicode_infors j.. 5 5 vaffhh                 ?    Ks'A  AAelementrcHt||dS)z+ Converts an attribute to a float. r)r=rG)rrs r% _as_floatr(s  T1%% & &&r'ceZdZdZdZdedeejejffdZ dedeejejffdZ dejd ejdeejejffd Z d ede fd Z dede fd ZdS)StatisticalLocaleSelectorz Selects a random locale based on statistical data. Takes either a territory code or a language code, and generates a Locale object. c,t|_dSrh)rrootr#s r%__init__z"StatisticalLocaleSelector.__init__5s$&& r'rr c\|jd|d}|td||d}|st d|t jd|D}t jd|D}|||S) z Calculates a random language based on the territory code, based on the probability that a person speaks the language in the territory. zterritory[@type='z']NzUnknown territory: languagePopulationz#No language data found for region: c8g|]}|dS)type)rGrYlangs r%rZzBStatisticalLocaleSelector._load_territory_data..Es$LLL4dhhv..LLLr'c.g|]}t|dS)populationPercent)rrs r%rZzBStatisticalLocaleSelector._load_territory_data..Fs#bbb $0C D Dbbbr')rfindrfindall ValueErrornparraynormalize_probabilities)r$r territorylang_populations languages percentagess r%_load_territory_dataz.StatisticalLocaleSelector._load_territory_data8s INN#Cx#C#C#CDD  "#C#C#CDD D$,,-ABB OM8MMNN NHLL;KLLLMM hbbQabbbcc ++I{CCCr'rc6|jd|d}|std|g}g}|D]}|d}||d|d}|6|||t |dt |d zd z t |d z|std ||tj |tj |S) z Calculates a random region for a language based on the total speakers of the language in that region. z'.//territory/languagePopulation[@type="z"]/..z#No region data found for language: rNzlanguagePopulation[@type="z"]rliteracyPercenti' populationz)No valid region data found for language: ) rrrrGrappendrrrrr)r$r territoriesregionsrterrrlang_pops r%_load_language_dataz-StatisticalLocaleSelector._load_language_dataJsV i''(aRZ(a(a(abb  T!"R"R"RSS S   DXXf%%F~yy!Jh!J!J!JKKH NN6 " " "   ($788D"3445D,//0     USSSTT T++BHW,=,=rx ?T?TUUUr'rfreqc8tj|}|||z fS)z* Normalize probabilities. )rsum)r$rrtotals r%rz1StatisticalLocaleSelector.normalize_probabilitiesls t $,&&r'rc||\}}tj||dd}t |d|S)zj Get a random locale based on the territory ISO code. Returns as a Locale object. p_r")rrrandomchoicereplacerI)r$rr probabilitiesrs r%rPz%StatisticalLocaleSelector.from_regionusb $(#<#xml.etree.ElementTreeetree ElementTreer dataclassesrtypingrrrrrr r r r numpyr language_tagsr camoufox.pkgmanrrrrcamoufox.warningsr exceptionsrrrrrrrrgeoip2.databaserrz ImportErrorrr6r0rBrIboolrSrdrbrrror{rrrElementrr=rrrOr3r'r%rs"""""""""!!!!!!OOOOOOOOOOOOOOOOOOOOOOGGGGGGGGGGGG))))))KKKKK    : $4 +s+t++++SV*..#.d.v....@E#tCy.14S>d*GhsmGGGGG - - !  L L L L L( L L L         & & & &      > "*'rz''''''V8V8V8V8V8V8V8V8r % $ & &sA A*)A*