o ra\@sddlZddlZddlZddlZddlZddlmZddlmZGddde Z Gddde Z Gdd d e Z Gd d d e ZGd d d eZddZddZGdddZddddgfddZdS)N) ElementTree)gettextc@s eZdZdS)NoDistroTemplateExceptionN)__name__ __module__ __qualname__rr3/usr/lib/python3/dist-packages/aptsources/distro.pyr%src@speZdZgfddZddZdddZdd Zd d Z  dd dZddZ ddZ ddZ ddZ ddZ dS) DistributioncCs.||_||_||_||_||_d|_d|_dS)z2 Container for distribution specific informations Zdebzdeb-srcN)idcodename descriptionreleaseis_like binary_type source_type)selfr r r rrrrr __init__+s zDistribution.__init__cCs||_d|_g|_g|_g|_g|_g|_g|_g|_g|_ d|_ g|_ d|_ d|_ d|_g|_|jjjD]}||jrH|j|jkrH||_nq5|jdurXtd|j|jfg}g}g}g}|jjD]}|js||jr|jr|jjr||jjr|jdr|js|j ||!|j"n[|jdr|jr|j |nK|j#|j$kr|js|j ||!|j"| |jn/|j#|j$kr|jr|j |n|j#|j%kr|js|j |n|j#|j%kr|jr|j ||js$|j|jj&vr$|js |j#|j$kr |j |qd|js|j#|j%kr|j |qd|j |qdt'||_t'||_|!||!|t'||_t'||_ |(dS)zf Find the corresponding template, main and child sources for the distribution NFz7Error: could not find a distribution template for %s/%scdrom:)) sourceslistsource_template child_sources main_sourcesZdisabled_sourcesZ cdrom_sourcesZdownload_comps enabled_comps cdrom_comps used_mediaget_source_codesource_code_sourcesdefault_server main_servernearest_server used_serversZmatcherZ templates is_codenamenameZ distributionr rr listZinvaliddisttemplateZofficialuri startswithZdisabledappendextendcompstyperrchildrenset get_mirrors)rrr'Zmediar,rrsourcerrr get_sources7s                            zDistribution.get_sourcesNc Cs|jj|_|jD] }|ds|j|qt|jdkr"|j|_ n|jdj |_ i|_ d}t j |rt|d}t|dddurE|j|_|d}|D]9}z|jd}Wntyc|jd }Ynwz|jd }Wntyy|jd }Ynwtd ||j |<qLd|_d|_t jd dd} | d} | d} | dkrt| } | | d| } |r|| |_| |j vr|j | |_| |_dSdS)zR Provide a set of mirrors where you can get the distribution from rrz%/usr/share/xml/iso-codes/iso_3166.xml)fileiterNZiso_3166_entryZ common_namer$Z alpha_2_codeZ alpha_3_codeiso_3166LANGZen_UK)default_.)rbase_urir rr)r"r*lenrrr( countriesospathexistsrgetattrZ getiteratorr4ZattribKeyErrorrdgettextlowercountry country_codegetenvfindr!) rmirror_templateZmediumfnameetitZelmZdescrcodelocaleazrGrrr r0sV                zDistribution.get_mirrorscCsfd}|d}|d}|dkr|dkr||td|}||jvr,td|j|Sd|dS)zu Try to get a human readable name for the main mirror of a country Customize for different distributions Nz://z.archive.ubuntu.comr: Server for %s%s/ )rIr=r>r8rstriprserverrFiZlirrr _get_mirror_names   zDistribution._get_mirror_namecCsdd}g}t|jdkst|jdkr:||jd|jr:|td|jdg|jr8|||j|jdg|St|jdkr||jd|js|td|jdg|jd}|jrp|||jsp|||j|jdg|r}||||dg|St|jdkr|td|jdg|jr|||j|jdg|tdd dg|jD] }|||dg}|||js|||jrq||vr||q|S) z- Return a list of used and suggested servers cSst|d|dS)zl Helper function that handles comaprision of mirror urls that could contain trailing slashesrT)rematchstriprU)Zmir1Zmir2rrr compare_mirrorssz5Distribution.get_server_list..compare_mirrorsr;rz Main serverTFzCustom serversN)r=r"r r*r8r!rY)rr]ZmirrorsrWZ mirror_entryrrr get_server_listsV !      zDistribution.get_server_listrc Cs|dur|j}|dur|j}|durt|j}|dur|j}|j|||||}|jrH||jkrJ|jj|j|||||j |jj |dddSdSdS)z3 Add distribution specific sources Nr;)r3pos) rr r%rrraddrrr3index)rr-r(r&r,ZcommentZ new_sourcerrr add_sources    zDistribution.add_sourcecCsXt|g}|jD]}|jjD]}|j|kr|jr||jqq|D]}||q"dS)z Enable a component in all main, child and source code sources (excluding cdrom based sources) comp: the component that should be enabled N)r/rr'Z componentsr$Zparent_componentr`_enable_component)rcompr,r1crrr enable_component%s     zDistribution.enable_componentc sZfdd}g}||j||ji}i}|D]!}|j|jkr;|j|vr-t||j<|jD] }||j|q0q|j D]!}|j|j kr`|j|vrRt||j<|jD] }||j|qUq?t |jdkrr|j dgdn|D]}|||qt|j D]}|||q|j rt |j dkr|j |j dgddS|j D] }|||qdSdS)Ncs@|j|vrdS||jvrdS|j||jdS)z Check if we already added the component to the repository, since a repository could be splitted into different apt lines. If not add the component N)r&r,r*r`)r1comps_per_distrdrr add_component_only_once7s  z?Distribution._enable_component..add_component_only_oncer;rS)r,)r-r,)r+rrr-rr&r/r,r`rrr=rbr) rrdrisourcesrgZcomps_per_sdistsrer1rrhr rc5sB                 zDistribution._enable_componentcCsg}||j||j||j||jvr!g}||j|D]}||jvr=|j|t|jdkr=|j|q#dS)zx Disable a component in all main, child and source code sources (excluding cdrom based sources) r;N) r+rrrrr,remover=r)rrdrjr1rrr disable_componentss        zDistribution.disable_componentcsfdd}g}g}|_jD]}||||qjD]}|jjdus+|jj|jkr1||||qjD]}||||q5dS)zN Change the server of all distro specific sources to a given host csh||_|jD]}|j|j|g|vr|j|q||j|j|gqt|jdkr2j|dSdS)Nr;)r(r,r&rlr*r=r)r1r(seenrdrrr change_server_of_sources z;Distribution.change_server..change_server_of_sourceN)rrrr'r<r(r)rr(rpZ seen_binaryZ seen_sourcer1rror change_servers        zDistribution.change_servercCs||jkrdSdS)z1 Compare a given name with the release codename. TF)r rr$rrr r#s zDistribution.is_codenameN)NNNNr)rrrrr2r0rYr^rbrfrcrmrqr#rrrr r )s  ]93 > r c@s(eZdZdZddZddZddZdS) DebianDistributionz+ Class to support specific Debian features cCs||jks |jdvr dSdS)z Compare a given name with the release codename and check if if it can be used as a synonym for a development releases )ZtestingunstableTF)r rrrrrr r#szDebianDistribution.is_codenamecCsvd}|d}|d}|dkr|dkr||td|}||jvr4tdtd|j|Sd|dS) z_ Try to get a human readable name for the main mirror of a country Debian specific Nz://ftp.z .debian.orgr:rRr5rSrT)rIr=r>r8rrDrUrVrrr rYs    z#DebianDistribution._get_mirror_namecCtj|dddS)Nz http://ftp.%s.debian.org/debian/rJr r0rorrr r0 zDebianDistribution.get_mirrorsN)rrr__doc__r#rYr0rrrr rts  rtc@eZdZdZddZdS)UbuntuDistributionz+ Class to support specific Ubuntu features cCrv)Nz$http://%s.archive.ubuntu.com/ubuntu/rwrxrorrr r0ryzUbuntuDistribution.get_mirrorsNrrrrzr0rrrr r| r|c@r{)UbuntuRTMDistributionz/ Class to support specific Ubuntu RTM features cCs|jj|_dSrs)rr<r rorrr r0sz!UbuntuRTMDistribution.get_mirrorsNr}rrrr rr~rc Csddlm}m}ddl}ddddd}z!|d d g|d d}|d }|d d|dDW|Sty[}z|j|j krPt d|WYd}~|SWYd}~|Sd}~ww)z-Call lsb_release --idrc and return a mapping.rPopenPIPENZsidDebianzDebian GNU/Linux unstable (sid)ru)CodenameDistributor ID DescriptionReleaseZ lsb_releasez-idrc)stdoutzutf-8css"|] }d|vr|dVqdS)z: N)split).0linerrr sz_lsb_release.. z#lsb_release failed, using defaults:) subprocessrrerrno communicatedecodeupdaterOSErrorENOENTloggingwarning)rrrresultoutexcrrr _lsb_releases&   rc Csddlm}m}ddl}zddlm}Wnty%ttjtj }Ynwz(|ddg||dd d}| D]}| d rK| d d d WSq9WdStyu}z|j|jkrjtd |WYd}~dSWYd}~dSd}~ww) z=Get the current channel from system-image-cli -i if possible.rrN)DEVNULLzsystem-image-cliz-iT)rstderrZuniversal_newlinesz channel: z: r;z+system-image-cli failed, using defaults: %s)rrrrr ImportErrorr?opendevnullO_RDWRr splitlinesr)rrrrr)rrrrrrrrrr _system_image_channels<      rc@s>eZdZdZdZdddZddZddZd d Zd d Z d S) _OSReleasez/etc/os-releaseTcCsHi|_d|_tj|_tj|jsdS|d|_|r"| dSdS)NFT) rvalidrOS_RELEASE_FILEr3r?r@isfileparseinject_lsb_compat)rZ lsb_compatrrr rs z_OSRelease.__init__cCsd|jd|jd<|jd|jd<|jd|jd<|jds'|jd|jd<|jd|jd <dS) NZIDrZ PRETTY_NAMErZVERSION_CODENAMErZUBUNTU_CODENAMEZ VERSION_IDr)rgetrorrr rs  z_OSRelease.inject_lsb_compatcCsBt|jd}|D]}|}|sq|j|ddq|dS)Nr=r;)rr3r\ parse_entryrclose)rfrrrr rs  z_OSRelease.parsecCs4||}|dkrt|tr|d}||j|<dS)NID_LIKE ) parse_value isinstancestrrr)rkeyvaluerrr r's  z_OSRelease.parse_entrycCs"t|}t|dkr|dS|S)Nr;r)shlexrr=)rrvaluesrrr r/s  z_OSRelease.parse_valueN)T) rrrZDEFAULT_OS_RELEASE_FILErrrrrrrrrr rs   rc Cs|r|r|r|sRt}g}t}|jr|j}t}|d}|d}|d}|d}|dg}|dkrRt}|durRd|vrRd }|d d d d d d }|}|}|dkr^t|||||S|d krjt |||||S|dkrvt |||||St |||||S)z Check the currently used distribution and return the corresponding distriubtion class that supports distro specific features. If no paramter are given the distro will be auto detected via a call to lsb-release rrrrrZUbuntuNz ubuntu-rtm/z Ubuntu-RTM/r;-rr) rrrrrrrsplitrr|rrtr ) r r r rrZ os_releaseZ os_resultZ lsb_resultZchannelrrr get_distro6s8    r)rrrZrr?Zxml.etree.ElementTreerapt_pkgr8 Exceptionrobjectr rtr|rrrrrrrrr s*  5