U &c\@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@sleZdZgfddZddZdddZdd Zd d Zdd 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__+szDistribution.__init__cCs||_d|_g|_g|_g|_g|_g|_g|_g|_g|_ d|_ g|_ d|_ d|_ d|_g|_|jjjD]&}||jrj|j|jkrj||_qqj|jdkrtd|j|jfg}g}g}g}|jjD]}|js||jr|jr|jjr||jjr|jdr6|js6|j ||!|j"n|jdrZ|jrZ|j |n|j#|j$kr|js|j ||!|j"| |jnj|j#|j$kr|jr|j |nF|j#|j%kr|js|j |n"|j#|j%kr|jr|j ||js|j|jj&kr|js8|j#|j$kr8|j |q|js\|j#|j%kr\|j |q|j |qt'||_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 startswithZdisabledappendextendcompstyperrZchildrenset get_mirrors)rrr'Zmediar,rrsourcerrr get_sources7s                   zDistribution.get_sourcesNc Cs|jj|_|jD]}|ds|j|qt|jdkrD|j|_ n|jdj |_ i|_ d}t j |rt|d}t|dddkr|j|_|d}|D]v}z|jd}Wntk r|jd }YnXz|jd }Wntk r|jd }YnXtd ||j |<qd|_d|_t jd dd} | d} | d} | dkrRt| } | | d| } |rv|| |_| |j kr|j | |_| |_dS)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_3166ZLANGZen_UK)default_.)rbase_urir rr)r"r*lenrrr( countriesospathexistsrgetattrZ getiteratorr3ZattribKeyErrorrdgettextlowercountry country_codegetenvfindr!) rmirror_templateZmediumfnameZetitZelmZdescrcodeZlocaleazrErrr r/sN            zDistribution.get_mirrorscCsjd}|d}|d}|dkr<|dkr<||td|}||jkrXtd|j|Sd|dSdS)zu Try to get a human readable name for the main mirror of a country Customize for different distributions Nz://z.archive.ubuntu.comr8 Server for %s%s/ )rGr;r<r6rstriprserverrDiZlirrr _get_mirror_names   zDistribution._get_mirror_namecCsdd}g}t|jdks:t|jdkrt||jd|jrt|td|jdg|jrp|||j|jdgn8t|jdkr||jd|js|td|jdg|jd}|jr|||js|||j|jdg|r||||dgnt|jdkr|td|jdg|jrD|||j|jdg|tdd dg|jD]L}|||dg}|||js^|||jrq^n||kr^||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 slashesrP)rematchstriprQ)Zmir1Zmir2rrr compare_mirrorssz5Distribution.get_server_list..compare_mirrorsr9rz Main serverTFzCustom serversN)r;r"r r*r6r!rU)rrYZmirrorsrSZ mirror_entryrrr get_server_listsV       zDistribution.get_server_listrc Cs|dkr|j}|dkr|j}|dkr.t|j}|dkr<|j}|j|||||}|jr||jkr|jj|j|||||j |jj |dddS)z3 Add distribution specific sources Nr9)r2pos) rr r%rrraddrrr2index)rr-r(r&r,ZcommentZ new_sourcerrr add_sources& zDistribution.add_sourcecCsXt|g}|jD].}|jjD] }|j|kr|jr||jqq|D]}||qDdS)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,r0crrr enable_component%s   zDistribution.enable_componentc sZfdd}g}||j||ji}i}|D]B}|j|jkr4|j|krZt||j<|jD]}||j|q`q4|j D]B}|j|j kr~|j|krt||j<|jD]}||j|qq~t |jdkr|j dgdn*|D]}|||q|j D]}|||q|j rVt |j dkr>|j |j dgdn|j D]}|||qDdS)Ncs@|j|krdS||jkr dS|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\)r0comps_per_distr`rr add_component_only_once7s   z?Distribution._enable_component..add_component_only_oncer9rO)r,)r-r,)r+rrr-rr&r.r,r\rrr;r^r) rr`resourcesrcZcomps_per_sdistsrar0rrdr r_5s:                zDistribution._enable_componentcCsg}||j||j||j||jkrBg}||j|D]4}||jkrF|j|t|jdkrF|j|qFdS)zx Disable a component in all main, child and source code sources (excluding cdrom based sources) r9N) r+rrrrr,remover;r)rr`rfr0rrr disable_componentss       zDistribution.disable_componentcsfdd}g}g}|_jD]}||||q jD]*}|jjdksV|jj|jkr8||||q8jD]}||||qjdS)zN Change the server of all distro specific sources to a given host csd||_|jD]8}|j|j|g|kr0|j|q ||j|j|gq t|jdkr`j|dS)Nr9)r(r,r&rhr*r;r)r0r(seenr`rrr change_server_of_sources z;Distribution.change_server..change_server_of_sourceN)rrrr'r:r(r)rr(rlZ seen_binaryZ seen_sourcer0rrkr change_servers      zDistribution.change_servercCs||jkrdSdSdS)z1 Compare a given name with the release codename. TFN)r rr$rrr r#s zDistribution.is_codename)N)NNNNr)rrrrr1r/rUrZr^rbr_rirmr#rrrr r )s  ] 93 >r c@s(eZdZdZddZddZddZdS) DebianDistributionz+ Class to support specific Debian features cCs ||jks|jdkrdSdSdS)z Compare a given name with the release codename and check if if it can be used as a synonym for a development releases )ZtestingunstableTFN)r rrnrrr r#szDebianDistribution.is_codenamecCszd}|d}|d}|dkr<|dkr<||td|}||jkrhtdtd|j|Sd|dSdS) z_ Try to get a human readable name for the main mirror of a country Debian specific Nz://ftp.z .debian.orgr8rNr4rOrP)rGr;r<r6rrBrQrRrrr rUs     z#DebianDistribution._get_mirror_namecCstj|dddS)Nz http://ftp.%s.debian.org/debian/rHr r/rkrrr r/szDebianDistribution.get_mirrorsN)rrr__doc__r#rUr/rrrr rosroc@seZdZdZddZdS)UbuntuDistributionz+ Class to support specific Ubuntu features cCstj|dddS)Nz$http://%s.archive.ubuntu.com/ubuntu/rqrrrkrrr r/szUbuntuDistribution.get_mirrorsNrrrrsr/rrrr rtsrtc@seZdZdZddZdS)UbuntuRTMDistributionz/ Class to support specific Ubuntu RTM features cCs|jj|_dS)N)rr:r rkrrr r/sz!UbuntuRTMDistribution.get_mirrorsNrurrrr rvsrvc Csddlm}m}ddl}ddddd}z@|d d g|d d}|d }|d d|dDWn<tk r}z|j|j krt d|W5d}~XYnX|S)z-Call lsb_release --idrc and return a mapping.rPopenPIPENZsidDebianzDebian GNU/Linux unstable (sid)rp)CodenameDistributor ID DescriptionReleaseZ lsb_releasez-idrc)stdoutzutf-8css |]}d|kr|dVqdS)z: N)split).0lrrr sz_lsb_release.. z#lsb_release failed, using defaults:) subprocessrxryerrno communicatedecodeupdaterOSErrorENOENTloggingwarning)rxryrresultoutexcrrr _lsb_releases   rc Csddlm}m}ddl}zddlm}Wn$tk rLttjtj }YnXzN|ddg||dd d}| D]$}| d rt| d d d WSqtWn<tk r}z|j|jkrtd |W5d}~XYnXdS) z=Get the current channel from system-image-cli -i if possible.rrwN)DEVNULLzsystem-image-cliz-iT)rstderrZuniversal_newlinesz channel: z: r9z+system-image-cli failed, using defaults: %s)rrxryrr ImportErrorr=opendevnullO_RDWRr splitlinesr)rrrrr)rxryrrrrrrrr _system_image_channels0   rc@s>eZdZdZdZdddZddZddZd d Zd d Z d S) _OSReleasez/etc/os-releaseTcCsDi|_d|_tj|_tj|js&dS|d|_|r@| dS)NFT) rvalidrOS_RELEASE_FILEr2r=r>isfileparseinject_lsb_compat)rZ lsb_compatrrr rsz_OSRelease.__init__cCsd|jd|jd<|jd|jd<|jd|jd<|jdsN|jd|jd<|jd|jd <dS) NZIDr|Z PRETTY_NAMEr}ZVERSION_CODENAMEr{ZUBUNTU_CODENAMEZ VERSION_IDr~)rgetrkrrr rs  z_OSRelease.inject_lsb_compatcCsBt|jd}|D]$}|}|s"q|j|ddq|dS)Nr=r9)rr2rX parse_entryrclose)rflinerrr rs z_OSRelease.parsecCs4||}|dkr&t|tr&|d}||j|<dS)NID_LIKE ) parse_value isinstancestrrr)rkeyvaluerrr r&s  z_OSRelease.parse_entrycCs"t|}t|dkr|dS|S)Nr9r)shlexrr;)rrvaluesrrr r.s  z_OSRelease.parse_valueN)T) rrrZDEFAULT_OS_RELEASE_FILErrrrrrrrrr rs   rc Cs|r|r|r|st}g}t}|jr,|j}t}|d}|d}|d}|d}|dg}|dkrt}|dk rd|krd }|d d d d d d }|}|}|dkrt|||||S|d krt |||||S|dkrt |||||St |||||SdS)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 r|r{r}r~rZUbuntuNz ubuntu-rtm/z Ubuntu-RTM/r9-rrz) rrrrrrrsplitrrtrvror ) r r r rrZ os_releaseZ os_resultZ lsb_resultZchannelrrr get_distro5s@   r)rrrVrr=Zxml.etree.ElementTreerapt_pkgr6 Exceptionrobjectr rortrvrrrrrrrr s(  5