File: //lib/python3/dist-packages/DistUpgrade/__pycache__/DistUpgradeAptCdrom.cpython-38.pyc
U
٪zc�1 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m
Z G dd� de �Z
G dd� de�ZdS )� N)�gettextc @ s e Zd ZdZdS )�
AptCdromErrorz% base exception for apt cdrom errors N)�__name__�
__module__� __qualname__�__doc__� r r �A/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeAptCdrom.pyr # s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z d"dd�Z d#dd�Z
dd� Zd$dd�Zdd � Zd%d d �Zd d � Zejd!k r�eZdS )&�AptCdromz represents a apt cdrom object c C s0 || _ || _t� | _t� | _t� | _t�� d S )N)�view� cdrompath�set�packages�
signatures�i18n�apt_pkgZ init_config)�selfr �pathr r r �__init__+ s zAptCdrom.__init__c C sB t j�tj�d�tj�d��}t j�|| �r>t� || |� dS )z\ restore the backup copy of the cdroms.list file
(*not* sources.list)!
�
Dir::State�Dir::State::cdromsN)
�osr �joinr �config�find_dir�find�exists�shutil�copy)r �
backup_ext�
cdromstater r r �restore_backup4 s
�zAptCdrom.restore_backupc C sr | � � }| �|| j�}tj�d�}t|��}|�� }W 5 Q R X |�|d| �}t|d��}|� |� W 5 Q R X dS )z comment out the cdrom entry �Dir::Etc::sourcelistz# %s�wN)
�
_readDiskName�_generateSourcesListLiner r r � find_file�open�read�replace�write)r �diskname�pentry� sourceslist�f�contentr r r �comment_out_cdrom_entry? s
z AptCdrom.comment_out_cdrom_entryc C s< t � }t � }t � }tj| jdd�D �] \}}}|�d�sB|�d�rP|dd�= q"q"d|kr\q"q"d|krz|�tj�|d�� q"d|kr�|�tj�|d�� q"d |ks�d
|kr�t� d | � q"d |kr�|�tj�|d �� q"d
|k�rt�
tj�|d
��D ]}|�tj�|d
|�� q�q"tj� |�d dkr"|dd�= q"|||fS )z|
scan the CD for interessting files and return them as:
(packagesfiles, signaturefiles, i18nfiles)
T)�topdownzdebian-installerz
dist-upgraderNz.aptignrZPackagesz Packages.gzZSourcesz
Sources.gzz+Sources entry found in %s but not supportedz Release.gpgr � Zpool) r
r �walkr �endswith�addr r �logging�error�listdir�split)r r r r �root�dirs�filesr. r r r �_scanCDJ s6
�
�
zAptCdrom._scanCDc C sl t j�d�}t �� }|�tjj�� �}| � � }t
|d��*}|� d||f � |� d||f � W 5 Q R X dS )z update apts cdrom.list r �az
CD::%s "%s";
zCD::%s::Label "%s";
N) r r r& ZCdromZident�aptZprogress�baseZ
CdromProgressr$ r' r* )r ZdbfileZcdrom�idZlabel�outr r r �_writeDatabasek s zAptCdrom._writeDatabasec C s@ t |�}tj�d�}t |�D ] }d|kr ||kr |�|� q |S )z& drop architectures that are not ours zAPT::Architecturez/binary-)r
r r r �remove)r r Zarch�dr r r � _dropArchu s zAptCdrom._dropArchc C sZ | j }tj�| j dd�}tj�|�rVt|��}|�� }W 5 Q R X dD ]}|�|d�}qD|S )Nz.disk�info)�"�]�[�_rK )r r r r r r' r( r) )r r+ rG r. Zspecialr r r r$ � s
zAptCdrom._readDiskNamec C s� d}d}g }|D ]D}t �d|�}|s0ttd���|�d�}|�d�}|�|�d�� q|r^|sbd S |�� d||d�|�f }|S ) N� z (.*)/dists/([^/]*)/(.*)/binary-*z&Could not calculate sources.list entryr2 � � zdeb cdrom:[%s]/ %s %s� )�re�matchr rK �group�append�sortr ) r r+ r r Zdist�compsrE �mr, r r r r% � s" �
z!AptCdrom._generateSourcesListLineNc
C s� |st j�d�}| �� }|D ]�}t �d|||�d�d � f �}tj�|tj� |�d �}|�
d�r�t � |��<}t |d��&}|�
d�} |�| � | dkr|q�q|W 5 Q R X W 5 Q R X q t�||� q d S �
N�Dir::State::lists�
cdrom:[%s]/%s�distsr z.gz�wbi � � T�r r r r$ �uri_to_filenamer r r r �splitextr4 �gzipr' r( r* r r )
r Z translations� targetdirr+ r. �fname�outf�grB �sr r r �_copyTranslations� s �
zAptCdrom._copyTranslationsc
C s� |st j�d�}| �� }|D ]�}t �d|||�d�d � f �}tj�|tj� |�d �}|�
d�r�t � |��<}t |d��&}|�
d�} |�| � | dkr|q�q|W 5 Q R X W 5 Q R X q t�||� q d S rW r] )
r r ra r+ r. rb rc rd rB re r r r �
_copyPackages� s �
zAptCdrom._copyPackagesc C sL |D �]@}t j�|�d }t j�|�d }t j�� }ddd||g}t�� �.}|�t j
� � � |j |d< t
j||d�}W 5 Q R X |dks� dS t|��"} t �| �}
|
�� |
jd } W 5 Q R X | �d �D ]~} | �� \}
}}t j�||�} t j�| ��st�d
| � q�t| ��}t �|�}W 5 Q R X ||
ks�t�d |
|f � dS q�qd S )
z! verify the signatues and hashes r zapt-keyz--quietZverifyZ
APT_CONFIG)�envFZSHA256�
zignoring missing '%s'z(hash sum mismatch expected %s but got %sT)r r r9 r_ �environr �tempfileZNamedTemporaryFiler* r r �dump�name�
subprocess�callr' ZTagFile�stepZsectionr r r6 rG Z sha256sumr7 )r r �sigZbasepath�releasefZ
verify_env�cmd�fpZretr. �tZsha256_section�entry�hash�sizerm �sumr r r �_verifyRelease� s@
�� zAptCdrom._verifyReleasec
C sz |st j�d�}| �� }|D ]X}tj�|�d }||fD ]:}t �d|||�d�d� f �}t �
|tj� ||�� q8q dS )z copy the release file rX r rY rZ NT) r r r r$ r r r_ r^ r r r r )r r ra r+ rq rr r. rb r r r � _copyRelease� s �zAptCdrom._copyReleasec C s� | � � \| _| _| _| �| j�| _t| j�dkrFt�d� tt d���| �
| j�r^| � | j� | � | j� | �
| j� | �� }|s�t�d� dS | �|| j�}tj�d�}t|��}|�� }W 5 Q R X t|d��}|�d||f � W 5 Q R X | �� d S )
z' reimplement pkgCdrom::Add() in python r z+no useable indexes found on CD, wrong ARCH?z`Unable to locate any package files, perhaps this is not a Ubuntu Disc or the wrong architecture?zno .disk/ directory foundFr" r# z%# added by the release upgrader
%s
%sT)r= r r r rF �lenr6 r7 r rK rz r{ rg rf r$ r% r r r& r' r( r* rC )r r+ Zdebliner- r. r/ r r r �_doAdd� s6
�
��zAptCdrom._doAddc
C s� t �d| j� |rLtj�tj�d�tj� d��}tj�
|�rLt � ||| � tj�
d| j� tj�
dd� z | �� }W nT ttfk
r� } z2t �d| � | j�td�td �| � W Y �d
S d }~X Y nX t �d | � |S )
z add a cdrom to apt's database z AptCdrom.add() called with '%s'r r zAcquire::cdrom::mountzAPT::CDROM::NoMount�truezcan't add cdrom: %szFailed to add the CDz�There was a error adding the CD, the upgrade will abort. Please report this as a bug if this is a valid Ubuntu CD.
The error message was:
'%s'FNzAptCdrom.add() returned: %s)r6 �debugr r r r r r r r r r r r
r} � SystemErrorr r7 r rK )r r r �res�er r r r5 # s,
� ��z AptCdrom.addc C s
| j dk S )z# helper to use this as 'if cdrom:' N)r )r r r r �__bool__? s zAptCdrom.__bool__�3)N)N)N)N)r r r r r r! r0 r= rC rF r$ r% rf rg rz r{ r} r5 r� �sys�versionZ __nonzero__r r r r r
( s$ !
"
)
r
)rP r r? r r6 r` r rn r� rk r rK � Exceptionr �objectr
r r r r �<module> s