o * bŒã@s4ddlZddlmZddlmZGdd„deƒZdS)éN)ÚSoSCleanerParser)ÚSoSHostnameMapcsPeZdZdZdZdgZd‡fdd„ Zdd„Zd d „Zd d „Z ‡fd d„Z ‡Z S)ÚSoSHostnameParserzHostname ParserZ hostname_mapz5(((\b|_)[a-zA-Z0-9-\.]{1,200}\.[a-zA-Z]{1,63}(\b|_)))NcsJtƒ|_tt|ƒ |¡|j ¡|j |¡g|_| ¡|j  ¡dS©N) rÚmappingÚsuperrÚ__init__Zload_domains_from_mapZload_domains_from_optionsÚ short_namesÚload_short_names_from_mappingZset_initial_counts)ÚselfZconfigZ opt_domains©Ú __class__©úE/usr/lib/python3/dist-packages/sos/cleaner/parsers/hostname_parser.pyrs  zSoSHostnameParser.__init__cCs>|jj ¡D]}t| d¡ƒdkr||jvr|j |¡qdS)aWhen we load the mapping file into the hostname map, we have to do some dancing to get those loaded properly into the "intermediate" dicts that the map uses to hold hosts and domains. Similarly, we need to also extract shortnames known to the map here. Ú.éN)rÚdatasetÚkeysÚlenÚsplitr Úappend)r Zhnamerrrr "s   €üz/SoSHostnameParser.load_short_names_from_mappingcCszd|vrdS| d¡}t|ƒdkr|j |d¡n|j |¡t|ƒdkr5d |dd…¡}|j |¡|j |¡dS)zöForce add the domainname found in /sos_commands/host/hostname into the map. We have to do this here since the normal map prep approach from the parser would be ignored since the system's hostname is not guaranteed Ú localhostNrrrééþÿÿÿ)rrr rÚjoinrÚadd)r Zhostname_stringZdomainsZ high_domainrrrÚload_hostname_into_map.s     z(SoSHostnameParser.load_hostname_into_mapcCsp| ¡}|D]/}| d¡sd|vrq| ¡dd…}|D]}t| d¡ƒdkr.|j |¡q|j |¡qqdS)a$Parse an archive's copy of /etc/hosts, which requires handling that is separate from the output of the `hostname` command. Just like load_hostname_into_map(), this has to be done explicitly and we cannot rely upon the more generic methods to do this reliably. ú#rrNr)Ú splitlinesÚ startswithrrr rrr)r ZcontentÚlinesÚlineZhostlnÚhostrrrÚload_hostname_from_etc_hostsBsúüz.SoSHostnameParser.load_hostname_from_etc_hostsc s¬d‡fdd„ }d}ttˆƒ |¡\}}dd„ˆjj ¡Dƒ}t|ddd „d D]}|}d D]}| |d ¡}q-|||||ƒ\}}q'tˆjdd D] }||||ƒ\}}qG||fS)z€Override the default parse_line() method to also check for the shortname of the host derived from the hostname. NcsL|ˆjjvr ||fSt |tj¡}| |¡r"| ˆj |p|¡|¡S||fS)zqPerform a second manual check for substrings that may have been missed by regex matching )rZ skip_keysÚreÚcompileÚIÚsearchÚsubnÚget)ZlnÚcountr'ÚreplZ_reg©r rrÚ _check_lineZs  z1SoSHostnameParser.parse_line.._check_linercSsg|]}d|vr|‘qS)rr)Ú.0ÚhrrrÚ isz0SoSHostnameParser.parse_line..TcSst|ƒSr)r)ÚxrrrÚjsz.SoSHostnameParser.parse_line..)ÚreverseÚkeyz.-Ú_)r3r) rrÚ parse_linerrrÚsortedÚreplacer ) r r!r-r*Zhostsr"ZfqdnÚcZ short_namer r,rr6Us zSoSHostnameParser.parse_liner) Ú__name__Ú __module__Ú __qualname__ÚnameZ map_file_keyZregex_patternsrr rr#r6Ú __classcell__rrr rrsÿ  r)r$Zsos.cleaner.parsersrZ!sos.cleaner.mappings.hostname_maprrrrrrÚs