0x1949 Team - FAZEMRX - MANAGER
Edit File: distro.cpython-310.pyc
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�