CasperSecurity
o
4��f�\ � @ 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 �4/usr/lib/python3/dist-packages/DistUpgrade/distro.pyr % s r c @ sp e Zd Zg fdd�Zdd� Zddd�Zdd � Zd
d � Z dd
d�Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS )� Distributionc C s. || _ || _|| _|| _|| _d| _d| _dS )z2 Container for distribution specific informations �debzdeb-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�rH|j| jkrH|| _ nq5| jdu rXtd| j| jf ��g }g }g }g }| j jD ]�}|js�| �|j�r�|jr�|jj r�| �|jj�r�|j � d�r�|j s�| j� |� |�!|j"� n[|j � d�r�|j r�| j� |� nK|j#| j$kr�|j s�| j� |� |�!|j"� |� |j � n/|j#| j$kr�|j r�| j� |� n |j#| j%kr�|j s�| j � |� n|j#| j%kr�|j r�| j� |� |j�s$|j| jj&v �r$|j �s |j#| j$k�r | j� |� qd|j �s |j#| j%k�r | j � |� qd| j� |� qdt'|�| _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_sources�disabled_sources�
cdrom_sources�download_comps�
enabled_comps� cdrom_comps�
used_media�get_source_code�source_code_sources�default_server� main_server�nearest_server� used_servers�matcher� templates� is_codename�name� distributionr r r
�list�invalid�dist�template�official�uri�
startswith�disabled�append�extend�comps�typer r �children�set� get_mirrors)r r r/ �mediar6 r r �sourcer r r � get_sources7 s� �
��
��� � � � � � � � �
zDistribution.get_sourcesNc
C s� | j j| _| jD ]
}|�d�s| j�|� qt| j�dkr"| j| _ n| jd j
| _ i | _ d}t j
�|�r�t|d�}t|dd�du rE|j|_|�d�}|D ]9}z|jd }W n tyc |jd }Y nw z|jd
}W n tyy |jd }Y nw t�d |�| j |�� < qLd| _d| _t jd
dd�} | �d�}
| �d�} | dkr�t| �} | |
d | � �� } |r�|| | _ | | j v r�| j | | _| | _dS 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�iterN�iso_3166_entry� common_namer* � alpha_2_code� alpha_3_code�iso_3166�LANG�en_UK)�default�_�.���� ) r �base_urir$ r r2 r& r4 �lenr r# r1 � countries�os�path�existsr �getattr� getiteratorr? �attrib�KeyErrorr �dgettext�lower�country� country_code�getenv�findr% )
r �mirror_template�medium�fname�et�it�elm�descr�code�locale�a�zrY r r r r: � sV
�
� ��
�zDistribution.get_mirrorsc C sf d}|� d�}|� d�}|dkr |dkr ||td� |� }|| jv r,td�| j| S d|�d� S )zu Try to get a human readable name for the main mirror of a country
Customize for different distributions Nz://z.archive.ubuntu.comrJ �
Server for %s�%s�/ )r[ rM rN rH �rstrip�r �serverrX �i�lir r r �_get_mirror_name� s
z Distribution._get_mirror_namec C s� dd� }g }t | j�dk s t | j�dkr:|| jd | j�r:|�td�| jdg� | jr8|�| �| j�| jdg� |S t | j�dkr|| jd | j�s|�td�| jdg� | jd }| jrp||| j�sp|�| �| j�| jdg� |r}|�| �|�|dg� |S t | j�dkr�|�td�| jdg� | jr�|�| �| j�| jdg� |�td�d dg� | jD ] }| �|�|dg}||| j�s�||| j�r�q�||vr�|�|� 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 slashesri )�re�match�striprj )�mir1�mir2r r r �compare_mirrors� s z5Distribution.get_server_list.<locals>.compare_mirrorsrK r z Main serverTFzCustom serversN)rM r& r$ r4 rH r% ro )r ru �mirrorsrl � mirror_entryr r r �get_server_list� sT ��!��
���
�
�z Distribution.get_server_listr c
C s� |du r| j }|du r| j}|du rt| j�}|du r | j}| j�|||||�}| jrH|| jkrJ| jj| j|||||j | jj�
|�d d� dS dS dS )z3
Add distribution specific sources
NrK )r> �pos) r# r
r, r r r �addr! r r> �index)r r7 r1 r. r6 �comment�
new_sourcer r r �
add_source s
��zDistribution.add_sourcec C sX t |g�}| jD ]}|jjD ]}|j|kr |jr |�|j� qq|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)r9 r r/ �
componentsr* �parent_componentrz �_enable_component)r �compr6 r< �cr r r �enable_component% s
�� �z Distribution.enable_componentc sZ � fdd�}g }|� | j� |� | j� i }i }|D ]!}|j| jkr;|j|vr-t� ||j< |jD ]
}||j �|� q0q| j D ]!}|j| j
kr`|j|vrRt� ||j< |jD ]
}||j �|� qUq?t | j�dk rr| j d� gd� n|D ]}|||� qt| j D ]}|||� q| j
r�t | j �dk r�| j | j
d� gd� d S | j D ] }|||� q�d S d S )Nc s@ | j |vrdS � || j v rdS | 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. r6 r4 rz )r< �comps_per_dist�r� r r �add_component_only_once7 s
z?Distribution._enable_component.<locals>.add_component_only_oncerK rh )r6 )r7 r6 )r5 r r r7 r r. r9 r6 rz r" r rM r~ r! ) r r� r� �sourcesr� �comps_per_sdist�sr� r< r r� r r� 5 sB
�
�
�z Distribution._enable_componentc C s� g }|� | j� |� | j� |� | j� || jv r!g }|� | j� |D ]}||jv r=|j�|� t|j�dk r=| j�|� q#dS )zx
Disable a component in all main, child and source code sources
(excluding cdrom based sources)
rK N) r5 r r r" r r6 �removerM r )r r� r� r< r r r �disable_components s
��z Distribution.disable_componentc s� � fdd�}g }g }|� _ � jD ]}||||� q� jD ]}|jjdu s+|jj|jkr1||||� q � jD ]}||||� q5dS )zN Change the server of all distro specific sources to
a given host c sh || _ | jD ] }| j | j|g|v r| j�|� q|�| j | j|g� qt| j�dk r2� j�| � d S d S )NrK )r1 r6 r. r� r4 rM r )r< r1 �seenr� �r r r �change_server_of_source� s
�z;Distribution.change_server.<locals>.change_server_of_sourceN)r# r r r/ rL r1 r" )r r1 r� � seen_binary� seen_sourcer< r r� r �
change_server� s
�
�zDistribution.change_serverc C s || j krdS dS )z1 Compare a given name with the release codename. TF)r
�r r* r r r r) � s
zDistribution.is_codename�N)NNNNr )r r r r r= r: ro rx r~ r� r� r� r� 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
| jdv r 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 )�testing�unstableTF)r
r r� r r r r) � s z DebianDistribution.is_codenamec C sv d}|� d�}|� d�}|dkr |dkr ||td� |� }|| jv r4td�t�d| j| �� ��� S d|�d� S ) z_ Try to get a human readable name for the main mirror of a country
Debian specific Nz://ftp.z .debian.orgrJ rg rD rh ri )r[ rM rN rH r rV rj rk r r r ro � s
��z#DebianDistribution._get_mirror_namec C � t j| dd� d S )Nz http://ftp.%s.debian.org/debian/�r\ �r
r: r� r r r r: � �
�z DebianDistribution.get_mirrorsN)r r r �__doc__r) ro r: r r r r r� � s
r� c @ � e Zd ZdZdd� ZdS )�UbuntuDistributionz+ Class to support specific Ubuntu features c C r� )Nz$http://%s.archive.ubuntu.com/ubuntu/r� r� r� r r r r: � r� z UbuntuDistribution.get_mirrorsN�r r r r� r: r r r r r� � � r� c @ r� )�UbuntuRTMDistributionz/ Class to support specific Ubuntu RTM features c C s | j j| _d S r� )r rL r$ r� r r r r: � s z!UbuntuRTMDistribution.get_mirrorsNr� r r r r r� � r� r� c
C s� ddl m} m} ddl}ddddd�}z!| d d
g|d ��� d }|�d �}|�d
d� |�d�D �� W |S ty[ } z|j|j krPt
� d| � W Y d}~|S W Y d}~|S d}~ww )z-Call lsb_release --idrc and return a mapping.r ��Popen�PIPEN�sid�Debianz Debian GNU/Linux unstable (sid)r� )�Codename�Distributor ID� Description�Release� lsb_releasez-idrc)�stdoutzutf-8c s s" � | ] }d |v r|� d �V qdS )z: N)�split)�.0�liner r r � <genexpr>� s � ��z _lsb_release.<locals>.<genexpr>�
z#lsb_release failed, using defaults:) �
subprocessr� r� �errno� communicate�decode�updater� �OSError�ENOENT�logging�warning)r� r� r� �result�out�excr r r � _lsb_release� s& �
�
���r� c
C s� ddl m} m} ddl}zddl m} W n ty% t�tjtj �}Y nw z(| ddg||dd��
� d }|� � D ]}|� d �rK|�
d
d �d W S q9W dS tyu } z|j|jkrjt�d | � W Y d}~dS W Y d}~dS d}~ww )
z=Get the current channel from system-image-cli -i if possible.r r� N)�DEVNULLzsystem-image-cliz-iT)r� �stderr�universal_newlinesz channel: z: rK z+system-image-cli failed, using defaults: %s)r� r� r� r� r� � ImportErrorrO �open�devnull�O_RDWRr� �
splitlinesr2 r� r� r� r� r� )r� r� r� r� r� r� r� r r r �_system_image_channel� s< �
��
��� �
���r� c @ s> e Zd ZdZdZddd�Zdd� Zdd� Zd d
� Zd d � Z d
S )�
_OSReleasez/etc/os-releaseTc C sH i | _ d| _tj| _tj�| j�sd S | �� d| _|r"| � � d S d S )NFT)
r� �validr� �OS_RELEASE_FILEr> rO rP �isfile�parse�inject_lsb_compat)r �
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 s'| j �d�| j d< | j �d�| j d < d S )
N�IDr� � PRETTY_NAMEr� �VERSION_CODENAMEr� �UBUNTU_CODENAME�
VERSION_IDr� )r� �getr� r r r r� s
z _OSRelease.inject_lsb_compatc C sB t | jd�}|D ]}|�� }|sq| j|�dd�� q|�� d S )N�r�=rK )r� r> rr � parse_entryr� �close)r �fr� r r r r� s z_OSRelease.parsec C s4 | � |�}|dkrt|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 )NrK r )�shlexr� rM )r r� �valuesr r r r� / s
z_OSRelease.parse_valueN)T)
r r r �DEFAULT_OS_RELEASE_FILEr� r r� r� r� r� r r r r r� s
r� c C s� | r|r|r|sRt � }g }t� }|jr|j}t� }|d } |d }|d }|d }|�dg �}| dkrRt� }|durRd|v rRd } |�d
d �d �d d �d
}|}|}| dkr^t| ||||�S | d krjt | ||||�S | dkrvt
| ||||�S t | ||||�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
r� r� r� r� r� �UbuntuNz ubuntu-rtm/z
Ubuntu-RTM�/rK �-r r� ) r� r� r� r� r� r� �rsplitr� r� r� r� r
) r r
r r r �
os_release� os_result�
lsb_result�channelr r r �
get_distro6 s8
�r� )r r� rp r� rO �xml.etree.ElementTreer �apt_pkgrH � Exceptionr �objectr
r� r� r� r� r� r� r� r r r r �<module> s* 5�