!۰i~-zdZddlZddlZddlZddlZddlZddlZejjejje Z ejje ddZ ejje ddZejdZddddd d d d d d d d d d ZGddZeZy)zAirport code resolver with Chinese/English/IATA input support. Loads cache/airports.json and builds reverse lookup indices so that user input in any form (IATA, Chinese city, English city, alias) resolves to the correct IATA code. Ncachez airports.jsonzinactive_airports.jsonzflyclaw.airportPVGPEKCTUJFKLHRCDG) u上海shanghaiu北京beijingu成都chengduu纽约newyorkznew yorku伦敦londonu巴黎parisc eZdZdZeefdedefdZdeddfdZdeddfd Z d ede fd Z d d dede de efdZ de ede de efdZdededzfdZdede fdZedede fdZdede fdZdededzfdZdedefdZy)AirportManagerz?Resolve user input (Chinese/English/IATA) to IATA airport code. data_path inactive_pathc||_i|_i|_i|_i|_i|_i|_t|_|j||j|y)N) _data_path _airports _iata_index_city_cn_index_city_en_index _alias_index_name_cn_indexset _inactive_set_load_load_inactive)selfrrs ;/root/.openclaw/workspace/skills/flyclaw/airport_manager.py__init__zAirportManager.__init__+s\#*,+-4646,..0'*u 9 M*pathreturnNcTt|dd5}tj||_ddd|jj D]L\}}||j |j <|jdd}|r+|jj|gj||jdd}|r}|j }|jj|gj||jdd}||k7r+|jj|gj||jdgD]} ||j| j <!|jd d} | s>||j| <Oy#1swYuxYw) Nrutf-8encodingcity_cncity_en aliasesname_cn)openjsonloadritemsrlowergetr setdefaultappendrreplacerr) r r$fiatainfor+r-keyno_spacealiasr0s r!rzAirportManager._load9sr $g . *!!YYq\DN *....0 4JD$-1D  TZZ\ *hhy"-G##..w;BB4Hhhy"-Gmmo##..sB7>>tD;;sB/s?''228R@GGM)R0 837!!%++-0 8hhy"-G/3##G,5 4 * *s FF'c t|dd5}tj|}dddjdi}|Dchc]}|j c}|_t jdt|j y#1swYhxYwcc}w#ttjf$r/}t jd|t|_Yd}~yd}~wwxYw)z+Load inactive airport codes from JSON file.r'r(r)NairportszLoaded %d inactive airportsz%Inactive airports file not loaded: %s) r1r2r3r6upperrloggerdebuglenOSErrorJSONDecodeErrorr)r r$r:datarAcodees r!rzAirportManager._load_inactiveYs 'dC'2 $ayy| $xx B/H;C!D4$**,!DD  LL6Dr+r-r0name_enT) isinstancedictrEr4strisupperissubsetkeys)rHrequired_fieldsr=vals r!_validate_airport_dataz%AirportManager._validate_airport_datas$%TRF  HCsC(SX]s{{}c4("++CHHJ7  r#urlc|sy tj|d}|j|j}|j|st j dy|jj}|jD]G\}}||vr9|jD]%\}} ||j|r| s| |||<'C|||<I |jdz} t| dd 5} tj|| dd dddtj| |ji|_i|_i|_i|_i|_i|_|j+|jt j-dt/|jy#t$r!}t j d||Yd}~yd}~wwxYw#1swYxYw#t$r }t j d |Yd}~yd}~wwxYw)uDownload airport data from *url* and merge into existing data. - Empty URL → return False - Validates downloaded data structure - Merges: keeps existing entries, adds new entries, fills missing fields - Writes via atomic replace - Rebuilds indices on success - Any failure → logs warning, returns False, existing data untouched F)timeoutz+Failed to download airport data from %s: %sNz)Downloaded airport data failed validationz.tmpwr(r)) ensure_asciiindentz Failed to write airport data: %sz Airport data updated: %d entriesT)requestsr6raise_for_statusr2 ExceptionrCwarningrxrcopyr4rr1dumprgr9rrrrrrr<rE) r ryrespnew_datarJmergedr;r<fieldrwtmp_pathr:s r!update_from_urlzAirportManager.update_from_urls <<R0D  ! ! #yy{H **84 NNF G$$&"..* $JD$v~"&**,2JE3!$<++E2s.1t U+2 $t  $ /Hhg6 C! &!%B C JJx 1     4??# 6DNN8KLO  NNH#q Q , C C  NN=q A sG7F&G5G(G& G/G  GGG H(HHr;cT|jj|jS)z'Return full airport info dict, or None.)rr6rB)r r;s r!get_infozAirportManager.get_infos~~!!$**,//r#c |j|}|s|S|jddxs|jdd}|jdg}d}|D]}|jr|}n|r ||k7r||n|}|s|S|d|dS)ulReturn display name like '上海浦东(PVG)'. Falls back to city_en when city_cn is empty. r+r,r-r/())rr6isascii)r r;r<cityr/shorta name_parts r!get_display_namezAirportManager.get_display_names }}T"Kxx 2&A$((9b*A((9b) A99; ).%4-tfUG$T KAdV1%%r#)__name__ __module__ __qualname____doc___AIRPORTS_FILE_INACTIVE_FILErrr"rrboolrLr]rYrTrdintrk staticmethodrqrxrrrr#r!rr(s+I(6&4 +# + # +4#4$4@ '3 '4 '6c6d6BF00$0$s)0d d3i D T#Y SS4Z* <3 <4 <Td*73747r0S0TD[0&S&S&r#r)rr2loggingrgrRrirr$dirnameabspath__file__ _PROJECT_DIRjoinrr getLoggerrCrarairport_managerrr#r!rs  wwrwwx89 lG_ElG5MN   , -  "J&J&\!"r#