U T{)_<"@s ddlmZddlZddlZddlZddlZddlZddddddd d d d d ddddddddddddgZddddddgZd d!d"d#d$d%d&d'gZ d(Z d)d*Z ej d+krd,d-Z nd.d-Z ed/fd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Zdd?Zd@dAZdBdCZdS)D)deepcopyNZaddressZnetmaskZ broadcastZnetworkZmetricZgatewayZ pointtopointZmediaZmtuZhostnameZ leasehoursZ leasetimeZvendorZclientZbootfileserverZhwaddrZproviderframeZnetnumZendpointZlocalZttlzpre-upZupzpost-upZdownzpre-downz post-downZ bridge_ageingZbridge_bridgeprioZ bridge_fdZ bridge_gcinitZ bridge_helloZ bridge_maxageZbridge_maxwaitZ bridge_stpicCst|}t|tr*t|tr*|||S|D]v\}}||krt||trnt|trnt|||||<qt||trt|trtt|||||<q2|||<q2|SN)r isinstancelistextenditemsdict dict_mergeset)Zontosourcetargetkeyvaluer//usr/lib/python3/dist-packages/probert/utils.pyr s  r )rcCs>i}|jD].}|j|}t|tr0|dd}|||<q |SNzutf-8replace) attributesgetrbytesdecodedevicerrvalrrrudev_get_attributes1s     rcCs@i}|jjD].}|j|}t|tr2|dd}|||<q |Sr)rZavailable_attributesrrrrrrrrr:s     cs(ddttfdd|D|DS)Ncss |]\}}dd||VqdS)csfdd|DS)Nc3s|]\}}|kr|VqdSrr).0Zpreditemirr Fsz:partitionn2.....r)r#teerr"rFz'partitionn2...Nr)r xtrrrr$Fszpartitionn2..c3s|]}||fVqdSrr)r r! predicaterrr$Gs) enumerate itertoolsr%r r+nrr*r partitionn2Esr0cs>|tfdddd}dd|D}|\\}}||fS)z{Unpack a partition into a tuple of (first partition, second partition) param: partition iterator from partitionn2 cs|dkS)Nr)index)r(r rrr&Ur'z#partition_to_pair..rr.cSsg|] }t|qSr)r)r prrr Wsz%partition_to_pair..)splitr0)inputZ partitionsdatarrrr2rpartition_to_pairNs  r8cCsdd|dDS)NcSs&g|]}t|r|ddqS);r)lenr5strip)r r(rrrr4^sz3disentagle_data_from_whitespace.. )r5)r7rrrdisentagle_data_from_whitespace\sr=cCszi}i}t|D]V}t|dkr"qt|\}}|dkrN||d|diqd|}|||iq|d|i|S)zuTransform lease string into dictionary of attributes params: lease: string if a dhcp lease structure { to } roption options)r=r:r8updatejoin)leaseZ lease_dictrAlinerrrrr dictify_leasebs    rFcCsddtd|ddDS)zParses dhclient leases file data, returning dictionary of leases :param leasesdata: string of lease data read from leases file cSsg|] }t|qSr)rF)r rDrrrr4}sz.parse_dhclient_leases_file..z {([^{}]*)}")refindallr) leasedatarrrparse_dhclient_leases_filexsrLcCsNi}|dD]:}|dst|dkr*q|d}|d||d<q|S)z6Parses systemd/networkd/netif lease data, returns dictr<#r?=r)r5 startswithr:lower)rKrDrEZkeyvaluerrrparse_networkd_lease_files rQcCs*ddg}|D]}tj|r |Sq dS)Nz /var/lib/dhcpz/var/lib/dhclient)ospathexists)Zsupported_dirsdrrrget_dhclient_ds   rVc Csd}|}|D]}|}|dr*q|d}|d}|dkrXtj||d}q|dkrtj||d}t|D]B} t| d} | } W5QRXt || tj tj | q|q|d kr|ddD] } | |kri|| <d || d <qq|d kr`|dd \} } }| |kr,i|| <nd || krBt d| || d <||| d<| }q|dkr||d||d<q|tkr|d|||<q|tkr|||krg|||<|||d|ddq|drd||kri||d<|dkrJg||dd<|ddD]}||dd|q*nD|dkrg||dd<|ddD]}||dd|qpq|drd||kri||d<|tkr|dd}|d||d|<q|dkr&g||dd<|ddD]} ||dd| qq|dkrX|ddkrX|d||dd<q|dkrd ||dkri||dd <|d||dd |d<q|d!krd"||dkri||dd"<|d||dd"|d<q|D] } d || krd#|| d <qdS)$zParses the file contents, placing result into ifaces. :param ifaces: interface dictionary :param contents: contents of interfaces file :param path: directory interfaces file was located NrMr@rzsource-directoryr?r rautoTifacefamilyz!Cannot define %s interface again.methodZ hwaddresszdns-Zdnsz dns-searchsearchzdns-nameserversZ nameserversZbridge_ZbridgerHZ bridge_portsZportsZ bridge_hwZmacrZbridge_pathcostZpathcostZbridge_portprioZportprioF) splitlinesr;rOr5rRrSrCglobopenreadparse_etc_network_interfacesdirnameabspath ExceptionNET_CONFIG_OPTIONSNET_CONFIG_COMMANDSappendNET_CONFIG_BRIDGE_OPTIONSrrPkeys)ZifacescontentsrSZcurrifZsrc_dirrEr5r>Zsrc_pathZsrc_filefpZsrc_datarXrZr[ZdomainrZ bridge_optionrrrras             "           rac CsPtjdtj|}tj|d}t|}t|t}W5QRX|S)zA /sys/class/block//size and return integer value in bytes/sys/class/blocksize) rRrSrCbasenamer_intr`r;SECTOR_SIZE_BYTES)r device_dirZ blockdev_sizerUrmrrrread_sys_block_size_bytess  rrcCs*tjdtj|}ttj|dS)NrlZslaves)rRrSrCrnlistdir)rrqrrrread_sys_block_slavessrt)copyrr^r-rRrIZpyudevrerfrhrpr Z__version_info__rror0r8r=rFrLrQrVrarrrtrrrrsx        S