3 `\@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|_x2|jjjD]$}|j|jrl|j|jkrl||_PqlW|jdkrtd|j|jfg}g}g}g}x|jjD]}|j r|j|jr|jr|jjr|j|jjr|jjdr@|j r@|jj ||j!|j"n|jjdrd|jrd|jj |n|j#|j$kr|j r|jj ||j!|j"|j |jnl|j#|j$kr|jr|jj |nH|j#|j%kr|j r|j j |n"|j#|j%kr|jr|jj ||j r|j|jj&kr|j rJ|j#|j$krJ|jj |q|j rp|j#|j%krp|j j |q|jj |qWt'||_t'||_|j!||j!|t'||_t'||_ |j(dS)zf Find the corresponding template, main and child sources for the distribution NFz7Error: could not find a distribution template for %s/%szcdrom:)) 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|_x$|jD]}|jds|jj|qWt|jdkrH|j|_ n|jdj |_ i|_ d}t j j|rt|d}t|dddkr|j|_|jd}x|D]x}y|jd}Wntk r|jd }YnXy|jd }Wn tk r|jd }YnXtjd ||j |j<qWd|_d|_t jd dd} | jd} | jd} | dkr\t| } | | d| j} |r|| |_| |j kr|j | |_| |_dS)zR Provide a set of mirrors where you can get the distribution from zcdrom:rz%/usr/share/xml/iso-codes/iso_3166.xml)fileiterNZiso_3166_entryZ common_namer#Z alpha_2_codeZ alpha_3_codeiso_3166ZLANGZen_UK)default_.)rbase_urirrr(r!r)lenrrr' countriesospathexistsrgetattrZ getiteratorr2ZattribKeyErrorrdgettextlowercountry country_codegetenvfindr ) rmirror_templateZmediumfnameZetitZelmZdescrcodeZlocaleazrDrrr r.sL             zDistribution.get_mirrorscCsjd}|jd}|jd}|dkr<|d kr<||td|}||jkrXtd|j|Sd|jdSdS) zu Try to get a human readable name for the main mirror of a country Customize for different distributions Nz://z.archive.ubuntu.comr7z Server for %sz%sz/ r8r8)rFr:r;r5rstrip)rserverrCilirrr _get_mirror_names   zDistribution._get_mirror_namecCsdd}g}t|jdks:t|jdkrt||jd|jrt|jtd|jdg|jrp|j|j|j|jdgnBt|jdko||jd|j r|jtd|jdg|jd}|jr|||js|j|j|j|jdg|r|j|j||dgnt|jdkr|jtd|jdg|jrJ|j|j|j|jdg|jtdd dgxV|jD]L}|j||dg}|||jsf|||jrqfn||krf|j|qfW|S) z- Return a list of used and suggested servers cSstj|jd|jdS)zl Helper function that handles comaprision of mirror urls that could contain trailing slashesz/ )rematchstriprM)Zmir1Zmir2rrr compare_mirrorssz5Distribution.get_server_list..compare_mirrorsr7rz Main serverTFzCustom serversN)r:r!rr)r5r rQ)rrUZmirrorsrNZ mirror_entryrrr get_server_listsB      zDistribution.get_server_listrc Cs|dkr|j}|dkr|j}|dkr.t|j}|dkr<|j}|jj|||||}|jr||jkr|jj|j|||||j |jjj |dddS)z3 Add distribution specific sources Nr7)r1pos) rr r$rrraddrrr1index)rr,r'r%r+ZcommentZ new_sourcerrr add_sources  zDistribution.add_sourcecCsdt|g}x<|jD]2}x,|jjD] }|j|kr |jr |j|jq WqWx|D]}|j|qNWdS)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_componentrX_enable_component)rcompr+r/crrr enable_component%s   zDistribution.enable_componentc sxfdd}g}|j|j|j|ji}i}xN|D]F}|j|jkr6|j|kr\t||j<x|jD]}||jj|qdWq6WxP|j D]F}|j|j kr|j|krt||j<x|jD]}||jj|qWqWt |jdkr|j dgdn4x|D]}|||qWx|j D]}|||qW|j rtt |j dkrX|j |j dgdnx|j D]}|||q`WdS)Ncs@|j|krdS||jkr dS|jj||jjdS)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)rX)r/comps_per_dist)r\rr add_component_only_once7s   z?Distribution._enable_component..add_component_only_oncer7z%s)r+)r,r+)r*rrr,rr%r-r+rXrrr:rZr) rr\r`sourcesr_Zcomps_per_sdistsr]r/r)r\r r[5s:                zDistribution._enable_componentcCsg}|j|j|j|j|j|j||jkrBg}|j|jx<|D]4}||jkrH|jj|t|jdkrH|jj|qHWdS)zx Disable a component in all main, child and source code sources (excluding cdrom based sources) r7N) r*rrrrr+remover:r)rr\rar/rrr disable_componentss        zDistribution.disable_componentcsfdd}g}g}|_xjD]}||||q"Wx4jD]*}|jjdks\|jj|jkr>||||q>WxjD]}||||qtWdS)zN Change the server of all distro specific sources to a given host csh||_xB|jD]8}|j|j|g|kr2|jj|q|j|j|j|gqWt|jdkrdjj|dS)Nr7)r'r+r%rcr)r:r)r/r'seenr\)rrr change_server_of_sources z;Distribution.change_server..change_server_of_sourceN)rrrr&r9r'r)rr'rfZ seen_binaryZ seen_sourcer/r)rr 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)rrrrr0r.rQrVrZr^r[rdrgr"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 testingunstableTFN)rirj)r r)rr#rrr r"szDebianDistribution.is_codenamecCszd}|jd}|jd}|d kr<|d kr<||td|}||jkrhtdtjd|j|jjSd|jdSdS) z_ Try to get a human readable name for the main mirror of a country Debian specific Nz://ftp.z .debian.orgr7z Server for %sr3z%sz/ r8r8)rFr:r;r5rrArM)rrNrCrOrPrrr rQs    z#DebianDistribution._get_mirror_namecCstj|dddS)Nz http://ftp.%s.debian.org/debian/)rG)r r.)rrrr r.szDebianDistribution.get_mirrorsN)rrr__doc__r"rQr.rrrr rhsrhc@seZdZdZddZdS)UbuntuDistributionz+ Class to support specific Ubuntu features cCstj|dddS)Nz$http://%s.archive.ubuntu.com/ubuntu/)rG)r r.)rrrr r.szUbuntuDistribution.get_mirrorsN)rrrrkr.rrrr rlsrlc@seZdZdZddZdS)UbuntuRTMDistributionz/ Class to support specific Ubuntu RTM features cCs|jj|_dS)N)rr9r)rrrr r.sz!UbuntuRTMDistribution.get_mirrorsN)rrrrkr.rrrr rmsrmcCsddlm}m}ddl}ddddd}y@|d d g|d jd}|jd }|jd d|jdDWn<tk r}z |j|j krt j d|WYdd}~XnX|S)z-Call lsb_release --idrc and return a mapping.r)PopenPIPENZsidDebianzDebian GNU/Linux unstable (sid)rj)CodenamezDistributor ID DescriptionReleaseZ lsb_releasez-idrc)stdoutzutf-8css |]}d|kr|jdVqdS)z: N)split).0lrrr sz_lsb_release.. z#lsb_release failed, using defaults:) subprocessrnroerrno communicatedecodeupdateruOSErrorENOENTloggingwarning)rnror{resultoutexcrrr _lsb_releases   rcCsddlm}m}ddl}yddlm}Wn$tk rLtjtjtj }YnXyL|ddg||ddj d}x*|j D]}|j d rv|j d d d SqvWWn<tk r}z |j|jkrtjd |WYdd}~XnXdS) z=Get the current channel from system-image-cli -i if possible.r)rnroN)DEVNULLzsystem-image-cliz-iT)rtstderrZuniversal_newlinesz channel: z: r7z+system-image-cli failed, using defaults: %s)rzrnror{r ImportErrorr<opendevnullO_RDWRr| splitlinesr(rurrrr)rnror{rrrwrrrr _system_image_channels$   rc@s>eZdZdZdZdddZddZddZd d Zd d Z d S) _OSReleasez/etc/os-releaseTcCsDi|_d|_tj|_tjj|js&dS|jd|_|r@|j dS)NFT) rvalidrOS_RELEASE_FILEr1r<r=isfileparseinject_lsb_compat)rZ lsb_compatrrr rsz_OSRelease.__init__cCsd|jd|jd<|jd|jd<|jjd|jd<|jdsN|jjd|jd<|jjd|jd <dS) NZIDzDistributor IDZ PRETTY_NAMErrZVERSION_CODENAMErqZUBUNTU_CODENAMEZ VERSION_IDrs)rget)rrrr rs  z_OSRelease.inject_lsb_compatcCsFt|jd}x,|D]$}|j}|s$q|j|jddqW|jdS)Nr=r7)rr1rT parse_entryruclose)rflinerrr rs  z_OSRelease.parsecCs4|j|}|dkr&t|tr&|jd}||j|<dS)NID_LIKE ) parse_value isinstancestrrur)rkeyvaluerrr r&s  z_OSRelease.parse_entrycCs"tj|}t|dkr|dS|S)Nr7r)shlexrur:)rrvaluesrrr r.s  z_OSRelease.parse_valueN)T) rrrZDEFAULT_OS_RELEASE_FILErrrrrrrrrr rs   rc Cs|o|o|o|st}g}t}|jr,|j}t}|d}|d}|d}|d}|jdg}|dkrt}|dk rd|krd }|jd d d jd 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 zDistributor IDrqrrrsrZUbuntuNz ubuntu-rtm/z Ubuntu-RTM/r7-rrp) rrrrrrrsplitrurlrmrhr ) r r r rrZ os_releaseZ os_resultZ lsb_resultZchannelrrr get_distro5s6   r)rrrRrr<Zxml.etree.ElementTreerapt_pkgr5 Exceptionrobjectr rhrlrmrrrrrrrr s&  5