File: //lib/python3/dist-packages/DistUpgrade/__pycache__/DistUpgradeFetcherCore.cpython-38.pyc
U
٪zc- � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m
Z d dl m
Z
ddlmZmZmZ G dd� de�Zed kr�edd�Zed
e�d d � � dS )
� )�TemplateN)�gettext)� SourcesList� )�get_dist�url_downloadable�country_mirrorc @ s� e Zd ZdZdZdZdejkZdd� Z dd� Z
d d
� Z d d � Z d
d� Z
d"dd�Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� ZdS )#�DistUpgradeFetcherCorez2 base class (without GUI) for the upgrade fetcher � http://archive.ubuntu.com/ubuntu�mainZDEBUG_UPDATE_MANAGERc C s || _ t� | _|| _g | _d S )N)�new_distr �current_dist_name� _progress� run_options)�selfr Zprogress� r �D/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeFetcherCore.py�__init__- s z DistUpgradeFetcherCore.__init__c C s | j rtj�|d � dS )z" helper to show debug information �
N)�DEBUG�sys�stderr�write)r �msgr r r �_debug4 s z DistUpgradeFetcherCore._debugc C s dS )NTr �r r r r �showReleaseNotes9 s z'DistUpgradeFetcherCore.showReleaseNotesc C s t |� t |� dS )z dummy implementation for error display, should be overwriten
by subclasses that want to more fancy method
F)�print)r Zsummary�messager r r �error< s z DistUpgradeFetcherCore.errorc C sv | j jrr| jd tj�| j j� }| jd tj�| j j� }ttd�tj�|�tj�|�d� � | � ||�rrdS dS )N�/z0authenticate '%(file)s' against '%(signature)s' )�file� signatureTF)
r �upgradeToolSig�tmpdir�os�path�basename� upgradeToolr �_�gpgauthenticate)r �fZsigr r r � authenticateD s �
�
z#DistUpgradeFetcherCore.authenticateNc C s@ dg}|sd}|d|g7 }|d||g7 }t j|t jd�}|dkS )zx authenticated a file against a given signature, if no keyring
is given use the apt default keyring
zapt-keyz./usr/share/keyrings/ubuntu-archive-keyring.gpgz --keyringZverify)r r )�
subprocess�call�PIPE)r r! r" ZkeyringZgpgZretr r r r* P s z&DistUpgradeFetcherCore.gpgauthenticatec
C s� t j�| jt j�| j��}ttd�t j�|� � t j�|�sBdS z@t �
| jd t j�| j� d�}|D ]}|� |� qh|� � W n8 t j
k
r� } zt�d| � W Y �dS d }~X Y nX dS )Nzextracting '%s'Fr �rzfailed to open tarfile (%s)T)r% r& �joinr$ r' �urir r) �exists�tarfile�open�extract�closeZ ReadError�loggingr )r �fnameZtarZtarinfo�er r r �extractDistUpgrader` s" �� z*DistUpgradeFetcherCore.extractDistUpgraderc C sH d| j | jjf | _}tj�|�sD| �td�td�d td� �S dS )Nz%s/%sz Could not run the upgrade toolz. z�This is most likely a bug in the upgrade tool. Please report it as a bug using the command 'ubuntu-bug ubuntu-release-upgrader-core'.T) r$ r �name�scriptr% r& r3 r r) )r r= r r r �verifyDistUpraderq s
��z(DistUpgradeFetcherCore.verifyDistUpraderc C s� | � d| j � tdd�}t� }|jD ]�}|js&|js&|jdksBq&|j|krNq&|j� |�rz|j
| jkrz| j |j krz| S |j
| jkr&d|j kr&|j|t
|�d� }t|| j �r�| S |�|j� q&| � d� dS ) a
try to figure what the mirror is from current sources.list
do this by looing for matching DEFAULT_COMPONENT, current dist
in sources.list and then doing a http HEAD/ftp size request
to see if the uri is available on this server
z mirror_from_sources_list: %sF)Z withMatcherZdebr Nzno mirror found� )r r
r �set�listZdisabledZinvalid�typer2 �
startswithZdist�DEFAULT_COMPONENT�comps�lenr �add)r r2 Z default_uriZsources�seenr: Z
mirror_urir r r �mirror_from_sources_list� s*
�
�
z/DistUpgradeFetcherCore.mirror_from_sources_listc
C s� |� | j�r,| �d� | �|| j�}|r,|S t|�}t� }|j|d�}z t|| j�s`td| ��W n< tk
r� } z | �d| � |jdd�}W 5 d}~X Y nX |S )z�
expand the uri so that it uses a mirror if the url starts
with a well known string (like archive.ubuntu.com)
z trying to find suitable mirror)Z
countrymirrorzfailed to download %sz url '%s' could not be downloadedr? N) rC �DEFAULT_MIRRORr rI r r Zsafe_substituter � Exception)r r2 Znew_uriZ uri_template�mr: r r r �
_expandUri� s
z!DistUpgradeFetcherCore._expandUric C sN t jdd�}|| _t�|� t�d| � | jdkrPtj �
dd� tj �
dd� t� | j �}| j
jdk r�| �| j
j�}tj||td �d
�}| | j
jdk �rJ| �| j
j�| _tj|| jtd �d
�}| |�� }||jkr�t�d | � d
S tj�| j
j�tj�| j
j�fD ]:}tj�|��r.tj�|�dk�s
t�d| � d
S �q
dS d
S )z. download the tarball with the upgrade script zubuntu-release-upgrader-)�prefixzusing tmpdir: '%s'r zDebug::Acquire::http�1zDebug::Acquire::ftpNzUpgrade tool signature)Zdescrz Upgrade toolz fetch result != continue (%s)Fzfile '%s' missingT)�tempfileZmkdtempr$ r% �chdirr8 �debugr �apt_pkgZconfigr@ ZAcquirer r r# rM Z AcquireFiler) r( r2 �runZRESULT_CONTINUEZwarningr& r' r3 �getsize)r r$ Zfetcherr2 Zaf1Zaf2�resultr+ r r r �fetchDistUpgrader� sD
��
�
z(DistUpgradeFetcherCore.fetchDistUpgraderc C sB | j g| j }t�� dkr0t�dddg| � nt�| j |� d S )Nr z
/usr/bin/sudoZsudoz-E)r= r r% �getuid�execv)r �argsr r r �runDistUpgrader� s z&DistUpgradeFetcherCore.runDistUpgraderc C s t �d� t�| j� d S )Nz..)r% rQ �shutilZrmtreer$ r r r r �cleanup� s
z DistUpgradeFetcherCore.cleanupc
C s> | � � s d S | �� s,| �td�td�� d S | �� sT| �td�td�� | �� d S | �� st| �td�td�� d S | �� s�| �td�td�� | �� d S z4t� | j
tj �s�t d | j
�}d
|_
|�| �� W nh t k
�r8 } zH|j
d
k�r| �td �td �� W Y �"d
S | �td �td�|j � W 5 d }~X Y nX dS )NzFailed to fetchz=Fetching the upgrade failed. There may be a network problem. zAuthentication failedz_Authenticating the upgrade failed. There may be a problem with the network or with the server. zFailed to extractz[Extracting the upgrade failed. There may be a problem with the network or with the server. zVerification failedz[Verifying the upgrade failed. There may be a problem with the network or with the server. zCan not execute '%s'�
zCan not run the upgradezyThis usually is caused by a system where /tmp is mounted noexec. Please remount without noexec and run the upgrade again.FzThe error message is '%s'.T)r rW r r) r, r] r; r> r% �accessr= �X_OK�OSError�errnor[ �strerror)r �exr: r r r rT � sP
�
�
�
�
�
�zDistUpgradeFetcherCore.run)N)�__name__�
__module__� __qualname__�__doc__rJ rD r% �environr r r r r r, r* r; r> rI rM rW r[ r] rT r r r r r & s&
�
'r �__main__zgot mirror: '%s'zghttp://archive.ubuntu.com/ubuntu/dists/intrepid-proposed/main/dist-upgrader-all/0.93.34/intrepid.tar.gzr
)�stringr r% rS r8 r4 rP r\ r r- r r) Zaptsources.sourceslistr Zutilsr r r �objectr re �dr rI r r r r �<module> s* v
��