File: //lib/python3/dist-packages/DistUpgrade/__pycache__/DistUpgradeFetcherCore.cpython-310.pyc
o
t�d�1 � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dl
mZ d dlmZ d dlmZ ddlmZmZmZ dd lmZ G d
d � d e�Zed kroedd�Z e d
e � dd� � dS dS )� )�TemplateN)�gettext)� SourcesList)�urlopen)� HTTPError� )�get_dist�url_downloadable�country_mirror)�readlinec @ s� e Zd ZdZdZdZdejv Zdd� 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�main�DEBUG_UPDATE_MANAGERc C s || _ t� | _|| _g | _d S �N)�new_distr �current_dist_name� _progress� run_options)�selfr �progress� r �D/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeFetcherCore.py�__init__1 s
z DistUpgradeFetcherCore.__init__c C s | j r
tj�|d � dS dS )z" helper to show debug information �
N)�DEBUG�sys�stderr�write)r �msgr r r �_debug8 s �z DistUpgradeFetcherCore._debugc C s� d| j v rdS | jjd urX| �| jj�}t�� }zt�d� t|�}|�� � dd�}W n' t
y? | � t d�t d�� Y dS t
yR | � t d �t d
�� Y dS w t�|� t� t|� tt d �d d
� t� }|�� �� �t d��rwdS dS )Nz(--frontend=DistUpgradeViewNonInteractiveT� zUTF-8�replacez'Could not find the release announcementz The server may be overloaded.Fz+Could not download the release announcementz&Please check your internet connection.zContinue [yN] � )�end�y)r r �releaseNotesURI�
_expandUri�socket�getdefaulttimeout�setdefaulttimeoutr �read�decoder �error�_�IOError�printr �strip�lower�
startswith)r �uri�timeout�
release_notes�notes�resr r r �showReleaseNotes= s8
�
��
z'DistUpgradeFetcherCore.showReleaseNotesc C s t |� t |� dS )z� minimal implementation for error display, should be overwriten
by subclasses that want to more fancy method
F)r0 )r �summary�messager r r r- X s z DistUpgradeFetcherCore.errorc C sv | j jr9| jd tj�| j j� }| jd tj�| j j� }ttd�tj�|�tj�|�d� � | � ||�r9dS dS )N�/z0authenticate '%(file)s' against '%(signature)s' )�file� signatureTF)
r �upgradeToolSig�tmpdir�os�path�basename� upgradeToolr0 r. �gpgauthenticate)r �f�sigr r r � authenticate` 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 --keyring�verify)r r )�
subprocess�call�PIPE)r r= r> �keyring�gpg�retr r r rE l s z&DistUpgradeFetcherCore.gpgauthenticatec
C s� t j�| jt j�| j��}ttd�t j�|� � t j�|�s!dS z!t �
| jd t j�| j� d�}|D ]}|� |� q4|� � W dS t j
y] } zt�d| � W Y d }~dS d }~ww )Nzextracting '%s'Fr< �rzfailed to open tarfile (%s)T)rA rB �joinr@ rC r4 r0 r. �exists�tarfile�open�extract�close� ReadError�loggingr- )r �fname�tar�tarinfo�er r r �extractDistUpgrader| s& ��
���z*DistUpgradeFetcherCore.extractDistUpgraderc C sH d| j | jjf | _}tj�|�s"| �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�scriptrA rB rR r- r. )r r_ r r r �verifyDistUprader� s
��z(DistUpgradeFetcherCore.verifyDistUpraderc C s� | � d| j � tdd�}t� }|jD ]P}|js |js |jdks!q|j|v r'q|j� |�r=|j
| jkr=| j |j v r=| S |j
| jkrcd|j v rc|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)� withMatcher�debr Nzno mirror foundr# )r r r �set�list�disabled�invalid�typer4 r3 �dist�DEFAULT_COMPONENT�comps�lenr �add)r r4 � default_uri�sources�seenr\ �
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�}zt|| j�s0td| ��W |S tyR } z| �d| � |jdd�}W Y d}~|S d}~ww )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)�
countrymirrorzfailed to download %sz url '%s' could not be downloadedr# N) r3 �DEFAULT_MIRRORr rq r r
�safe_substituter � Exception)r r4 �new_uri� uri_template�mr\ r r r r' � s&
����z!DistUpgradeFetcherCore._expandUric C sF t jdd�}|| _t�|� t�d| � | jdkr(tj �
dd� tj �
dd� t� | j �}| j
jdurG| �| j
j�}tj||td �d
�}| | j
jdur�| �| j
j�| _tj|| jtd �d
�}| |�� }||jkrtt�d | � d
S tj�| j
j�tj�| j
j�fD ]}tj�|�r�tj�|�dks�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)�descrz Upgrade toolz fetch result != continue (%s)Fzfile '%s' missingT)�tempfile�mkdtempr@ rA �chdirrX �debugr �apt_pkg�configrc �Acquirer r r? r' � AcquireFiler. rD r4 �run�RESULT_CONTINUE�warningrB rC rR �getsize)r r@ �fetcherr4 �af1�af2�resultrF r r r �fetchDistUpgrader� sF
� �
� �z(DistUpgradeFetcherCore.fetchDistUpgraderc C sD | j g| j }t�� dkrt�dddg| � d S t�| j |� d S )Nr z
/usr/bin/sudo�sudoz-E)r_ r rA �getuid�execv)r �argsr r r �runDistUpgrader� s z&DistUpgradeFetcherCore.runDistUpgraderc C s t �d� t�| j� d S )Nz..)rA r~ �shutil�rmtreer@ )r r r r �cleanup s
z DistUpgradeFetcherCore.cleanupc
C s@ | � � sd S | �� s| �td�td�� d S | �� s*| �td�td�� | �� d S | �� s:| �td�td�� d S | �� sN| �td�td�� | �� d S zt� | j
tj �sct d | j
�}d
|_
|�| �� W dS t y� } z*|j
d
kr�| �td �td �� W Y d }~d
S | �td �td�|j � W Y d }~dS d }~ww )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)r9 r� r- r. rH r� r] r` rA �accessr_ �X_OK�OSError�errnor� �strerror)r �exr\ r r r r� sV
�
�
�
�
�
�
���zDistUpgradeFetcherCore.runr )�__name__�
__module__� __qualname__�__doc__rs ri rA �environr r r r9 r- rH rE r] r` rq r' r� r� r� r� 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 rA r� rX rS r| r� r( r rJ r r. �aptsources.sourceslistr �urllib.requestr � urllib.errorr �utilsr r r
�DistUpgradeViewTextr �objectr r� �dr0 rq r r r r �<module> s8
�
��