File: //proc/thread-self/root/lib/python3/dist-packages/aptsources/__pycache__/distro.cpython-38.pyc
U
&��c�\ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlm Z G dd� de �Z
G dd� de �Z G dd � d e �Z
G d
d � d e �ZG d d
� d
e�Zdd� Zdd� ZG dd� d�Zddddg fdd�ZdS )� N)� ElementTree)�gettextc @ s e Zd ZdS )�NoDistroTemplateExceptionN)�__name__�
__module__� __qualname__� r r �3/usr/lib/python3/dist-packages/aptsources/distro.pyr % s r c @ sl e Zd Zg fdd�Zdd� Zddd�Zdd � Zd
d � Zdd
d�Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS )� Distributionc C s. || _ || _|| _|| _|| _d| _d| _dS )z2 Container for distribution specific informations Zdebzdeb-srcN)�id�codename� description�release�is_like� binary_type� source_type)�selfr r r
r r r r r �__init__+ s zDistribution.__init__c C s� || _ d| _g | _g | _g | _g | _g | _g | _g | _g | _ d| _
g | _ d| _ d| _
d| _g | _| j jjD ]&}| �|j�rj|j| jkrj|| _ q�qj| jdkr�td| j| jf ��g }g }g }g }| j jD �]�}|j�s | �|j��r |j�r |jj �r | �|jj��r |j � d��r6|j �s6| j� |� |�!|j"� n�|j � d��rZ|j �rZ| j� |� n�|j#| j$k�r�|j �s�| j� |� |�!|j"� |� |j � nj|j#| j$k�r�|j �r�| j� |� nF|j#| j%k�r�|j �s�| j � |� n"|j#| j%k�r |j �r | j� |� |js�|j| jj&kr�|j �s8|j#| j$k�r8| j� |� q�|j �s\|j#| j%k�r\| j � |� q�| j� |� q�t'|�| _t'|�| _|�!|� |�!|� t'|�| _t'|�| _ | �(� dS )zf
Find the corresponding template, main and child sources
for the distribution
NF� z7Error: could not find a distribution template for %s/%s�cdrom:))� sourceslist�source_template�
child_sources� main_sourcesZdisabled_sourcesZ
cdrom_sourcesZdownload_comps�
enabled_comps� cdrom_comps�
used_media�get_source_code�source_code_sources�default_server� main_server�nearest_server� used_serversZmatcherZ templates� is_codename�nameZ distributionr r r �listZinvalid�dist�templateZofficial�uri�
startswithZdisabled�append�extend�comps�typer r Zchildren�set� get_mirrors)r r r' Zmediar, r r �sourcer r r � get_sources7 s�
�
��
��� �� �� ��� �
�
�
zDistribution.get_sourcesNc
C s� | j j| _| jD ]}|�d�s| j�|� qt| j�dkrD| j| _ n| jd j
| _ i | _ d}t j
�|��rt|d�}t|dd�dkr�|j|_|�d�}|D ]v}z|jd }W n tk
r� |jd }Y nX z|jd
}W n tk
r� |jd }Y nX t�d |�| j |�� < q�d| _d| _t jd
dd�} | �d�}
| �d�} | dk�rRt| �} | |
d | � �� } |�rv|| | _ | | j k�r�| j | | _| | _dS )zR
Provide a set of mirrors where you can get the distribution from
r r z%/usr/share/xml/iso-codes/iso_3166.xml)�file�iterNZiso_3166_entryZ common_namer$ Z alpha_2_codeZ alpha_3_code�iso_3166ZLANGZen_UK)�default�_�.���� ) r �base_urir r r) r" r* �lenr r r( � countries�os�path�existsr �getattrZ getiteratorr3 Zattrib�KeyErrorr �dgettext�lower�country� country_code�getenv�findr! )
r �mirror_templateZmedium�fnameZet�itZelmZdescr�codeZlocale�a�zrE r r r r/ � sN
�
zDistribution.get_mirrorsc C sj d}|� d�}|� d�}|dkr<|dkr<||td� |� }|| jkrXtd�| j| S d|�d� S dS )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�/ )rG r; r< r6 �rstrip�r �serverrD �iZlir r r �_get_mirror_name� s
z Distribution._get_mirror_namec C s� dd� }g }t | j�dk s:t | j�dkrt|| jd | j�rt|�td�| jdg� | jrp|�| �| j�| jdg� �n8t | j�dkr�|| jd | j�s�|�td�| jdg� | jd }| jr�||| j�s�|�| �| j�| jdg� |r�|�| �|�|dg� n�t | j�dk�r�|�td�| jdg� | j�rD|�| �| j�| jdg� |�td�d dg� | jD ]L}| �|�|dg}||| j��s^||| j��r��q^n||k�r^|�|� �q^|S )
z- Return a list of used and suggested servers c S s t �| �d�|�d��S )zl Helper function that handles comaprision of mirror urls
that could contain trailing slashesrP )�re�match�striprQ )Zmir1Zmir2r r r �compare_mirrors� s z5Distribution.get_server_list.<locals>.compare_mirrorsr9 r z Main serverTFzCustom serversN)r; r" r r* r6 r! rU )r rY ZmirrorsrS Z mirror_entryr r r �get_server_list� sV �� �
�
� �
�
z Distribution.get_server_listr c
C s� |dkr| j }|dkr | j}|dkr.t| j�}|dkr<| j}| j�|||||�}| jr�|| jkr�| jj| j|||||j | jj�
|�d d� dS )z3
Add distribution specific sources
Nr9 )r2 �pos) r r r% r r r �addr r r2 �index)r r- r( r& r, ZcommentZ
new_sourcer r r �
add_source s&
�zDistribution.add_sourcec C sX t |g�}| jD ].}|jjD ] }|j|kr |jr |�|j� q q|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. r r' Z
componentsr$ Zparent_componentr\ �_enable_component)r �compr, r0 �cr r r �enable_component% s
z Distribution.enable_componentc sZ � fdd�}g }|� | j� |� | j� i }i }|D ]B}|j| jkr4|j|krZt� ||j< |jD ]}||j �|� q`q4| j D ]B}|j| j
kr~|j|kr�t� ||j< |jD ]}||j �|� q�q~t | j�dk r�| j d� gd� n*|D ]}|||� q�| j D ]}|||� q�| j
�rVt | j �dk �r>| j | j
d� gd� n| j D ]}|||� �qDd S )Nc s@ | j |krdS � || j kr dS | j�� � || j �� � dS )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\ )r0 �comps_per_dist�r` r r �add_component_only_once7 s
z?Distribution._enable_component.<locals>.add_component_only_oncer9 rO )r, )r- r, )r+ r r r- r r& r. r, r\ r r r; r^ r ) r r` re �sourcesrc Zcomps_per_sdist�sra r0 r rd r r_ 5 s:
z Distribution._enable_componentc C s� g }|� | j� |� | j� |� | j� || jkrBg }|� | j� |D ]4}||jkrF|j�|� t|j�dk rF| j�|� qFdS )zx
Disable a component in all main, child and source code sources
(excluding cdrom based sources)
r9 N) r+ r r r r r, �remover; r )r r` rf r0 r r r �disable_components s
z Distribution.disable_componentc s� � fdd�}g }g }|� _ � jD ]}||||� q � jD ]*}|jjdksV|jj|jkr8||||� q8� jD ]}||||� qjdS )zN Change the server of all distro specific sources to
a given host c sd || _ | jD ]8}| j | j|g|kr0| j�|� q |�| j | j|g� q t| j�dk r`� j�| � d S )Nr9 )r( r, r& rh r* r; r )r0 r( �seenr` �r r r �change_server_of_source� s
z;Distribution.change_server.<locals>.change_server_of_sourceN)r r r r' r: r( r )r r( rl Z seen_binaryZ seen_sourcer0 r rk r �
change_server� s
�
zDistribution.change_serverc C s || j krdS dS dS )z1 Compare a given name with the release codename. TFN)r �r r$ r r r r# � s
zDistribution.is_codename)N)NNNNr )r r r r r1 r/ rU rZ r^ rb r_ ri rm r# r r r r r
) s ]
93 �
> r
c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �DebianDistributionz+ Class to support specific Debian features c C s || j ks| jdkrdS dS dS )z� Compare a given name with the release codename and check if
if it can be used as a synonym for a development releases )Ztesting�unstableTFN)r r rn r r r r# � s z DebianDistribution.is_codenamec C sz d}|� d�}|� d�}|dkr<|dkr<||td� |� }|| jkrhtd�t�d| j| �� ��� S d|�d� S dS ) z_ Try to get a human readable name for the main mirror of a country
Debian specific Nz://ftp.z .debian.orgr8 rN r4 rO rP )rG r; r< r6 r rB rQ rR r r r rU � s
�
z#DebianDistribution._get_mirror_namec C s t j| dd� d S )Nz http://ftp.%s.debian.org/debian/�rH �r
r/ rk r r r r/ � s �z DebianDistribution.get_mirrorsN)r r r �__doc__r# rU r/ r r r r ro � s ro c @ s e Zd ZdZdd� ZdS )�UbuntuDistributionz+ Class to support specific Ubuntu features c C s t j| dd� d S )Nz$http://%s.archive.ubuntu.com/ubuntu/rq rr rk r r r r/ � s �z UbuntuDistribution.get_mirrorsN�r r r rs r/ r r r r rt � s rt c @ s e Zd ZdZdd� ZdS )�UbuntuRTMDistributionz/ Class to support specific Ubuntu RTM features c C s | j j| _d S )N)r r: r rk r r r r/ � s z!UbuntuRTMDistribution.get_mirrorsNru r r r r rv � s rv c
C s� ddl m} m} ddl}ddddd�}z@| d d
g|d ��� d }|�d �}|�d
d� |�d�D �� W n< tk
r� } z |j|j kr�t
� d| � W 5 d}~X Y nX |S )z-Call lsb_release --idrc and return a mapping.r ��Popen�PIPENZsid�Debianz Debian GNU/Linux unstable (sid)rp )�Codename�Distributor ID� Description�ReleaseZ lsb_releasez-idrc)�stdoutzutf-8c s s | ]}d |kr|� d �V qdS )z: N)�split)�.0�lr r r � <genexpr>� s z _lsb_release.<locals>.<genexpr>�
z#lsb_release failed, using defaults:) �
subprocessrx ry �errno� communicate�decode�updater� �OSError�ENOENT�logging�warning)rx ry r� �result�out�excr r r � _lsb_release� s �
r� c
C s� ddl m} m} ddl}zddl m} W n$ tk
rL t�tjtj �}Y nX zN| ddg||dd��
� d }|� � D ]$}|� d �rt|�
d
d �d W S qtW n< tk
r� } z |j|jkr�t�d | � W 5 d}~X Y nX dS )
z=Get the current channel from system-image-cli -i if possible.r rw N)�DEVNULLzsystem-image-cliz-iT)r �stderrZuniversal_newlinesz channel: z: r9 z+system-image-cli failed, using defaults: %s)r� rx ry r� r� � ImportErrorr= �open�devnull�O_RDWRr� �
splitlinesr) r� r� r� r� r� )rx ry r� r� r� r� r� r r r �_system_image_channel� s0 ��
�r� c @ s> e Zd ZdZdZddd�Zdd� Zdd� Zd d
� Zd d � Z d
S )�
_OSReleasez/etc/os-releaseTc C sD i | _ d| _tj| _tj�| j�s&d S | �� d| _|r@| � � d S )NFT)
r� �validr� �OS_RELEASE_FILEr2 r= r> �isfile�parse�inject_lsb_compat)r Z
lsb_compatr r r r s z_OSRelease.__init__c C sd | j d | j d< | j d | j d< | j �d�| j d< | j d sN| j �d�| j d< | j �d�| j d < d S )
NZIDr| Z PRETTY_NAMEr} ZVERSION_CODENAMEr{ ZUBUNTU_CODENAMEZ
VERSION_IDr~ )r� �getrk r r r r� s
z _OSRelease.inject_lsb_compatc C sB t | jd�}|D ]$}|�� }|s"q| j|�dd�� q|�� d S )N�r�=r9 )r� r2 rX � parse_entryr� �close)r �f�liner r r r� s z_OSRelease.parsec C s4 | � |�}|dkr&t|t�r&|�d�}|| j|< d S )N�ID_LIKE� )� parse_value�
isinstance�strr� r� )r �key�valuer r r r� & s
z_OSRelease.parse_entryc C s" t �|�}t|�dkr |d S |S )Nr9 r )�shlexr� r; )r r� �valuesr r r r� . s
z_OSRelease.parse_valueN)T)
r r r ZDEFAULT_OS_RELEASE_FILEr� r r� r� r� r� r r r r r� s
r� c C s | r|r|r|s�t � }g }t� }|jr,|j}t� }|d } |d }|d }|d }|�dg �}| dkr�t� }|dk r�d|kr�d } |�d
d �d �d d �d
}|}|}| dkr�t| ||||�S | d kr�t | ||||�S | dkr�t
| ||||�S t | ||||�S dS )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~ r� ZUbuntuNz ubuntu-rtm/z
Ubuntu-RTM�/r9 �-r rz ) r� r� r� r� r� r� �rsplitr� rt rv ro r
) r r r
r r Z
os_releaseZ os_resultZ
lsb_resultZchannelr r r �
get_distro5 s@
�r� )r r� rV r� r= Zxml.etree.ElementTreer Zapt_pkgr6 � Exceptionr �objectr
ro rt rv r� r� r� r� r r r r �<module> s( 5�