File: //lib/python3/dist-packages/DistUpgrade/__pycache__/DistUpgradeView.cpython-38.pyc
U
٪zc�A � @ s� d dl mZ d dl mZ d dlmZ ddlZddlm Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlZd dlmZ z e
�e
jd� e
�� \ZZW n e�d � d
ZY nX es�d
Ze �d d � ze
�e
jd � W n e
jk
r� Y nX e� d
e
� � � dd� Z G dd� dej j j!�Z!G dd� dej j j"�Z"G dd� de#�Z$G dd� de#�Z%G dd� de �Z&ed�ed�ed �ed �ed �f G d d � d e#�Z'e(d!k�r�e!� Z)e)�*� dS )"� )�gettext)�ngettext)�get� N)�Enum)�apport_pkgfailure� zgetting the encoding failedzutf-8�LC_CTYPEzC.UTF-8zlocale: '%s' '%s'c C s0 t | �} | d }| d d }| d d }| d }|dkr<d}ddddd�}|dkrftd d
|�| |d < |dkr�td d
|�| |d< |dkr�tdd|�| |d< tdd|�| |d< |dkr�|d dkr�|d S td�| S |dkr�|d S |dk�r|d dk�r
|d S td�| S |dk�r(|d S |d S )z; return the time a bit fuzzy (no seconds if time > 60 secs i�Q i � �< r r r )�str_days� str_hours� str_minutes� str_secondsz%li dayz%li daysr z%li hourz %li hoursr
z
%li minutez %li minutesr z
%li secondz %li secondsr z%(str_days)s %(str_hours)s� z %(str_hours)s %(str_minutes)s)�intr �_)ZsecZdaysZhoursZminutesZseconds�map� r �=/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeView.py�FuzzyTimeToStr; s@ �
r c sZ e Zd Z� fdd�Z� fdd�Z� fdd�Zdd� Zd� fd
d � Zd d
� Zdd� Z � Z
S )�AcquireProgressc s t t| ��� d| _d S )N� )�superr �__init__� est_speed��self�� __class__r r r � s zAcquireProgress.__init__c s* t t| ��� d| _d| _d| _d| _d S )Nr F)r r �startr �eta�percent�release_file_download_errorr r r r r � s
zAcquireProgress.startc sd t t| ��||||� |tjkr`t�d| � |�d�sB|�d�r`dD ]}|�|�rFd| _ q`qFd S )Nz update_status: dlFailed on '%s' z Release.gpgZRelease)ZhttpZftpZmirrorT)
r r �
update_status�apt_pkgZ STAT_FAILED�loggingZwarning�endswith�
startswithr# )r ZuriZdescrZ
shortDescrZstatusZnetr r r r$ � s
z AcquireProgress.update_statusc C s dS )NTr )r �argr r r � pulse_items� s zAcquireProgress.pulse_itemsNc sv t t| ��|� | j| j d t| j| j � | _| j | j
krP| j
| j d | _
| j dkrr| j| j t| j � | _ dS )Ng Y@g @r T) r r �pulseZ
current_bytesZ
current_items�floatZ total_bytesZ total_itemsr" Z current_cpsr r! )r �ownerr r r r+ � s �
�zAcquireProgress.pulsec C s
| j dkS )Nr )r r r r r �isDownloadSpeedEstimated� s z(AcquireProgress.isDownloadSpeedEstimatedc C sP | j dkr6|d }|d }td�t|�t|�f }|S td�t|| j � }|S )z! get the estimated download time r g �@g Az[This download will take about %s with a 1Mbit DSL connection and about %s with a 56k modem.z7This download will take about %s with your connection. )r r r )r Zrequired_downloadZ timeModemZtimeDSL�sr r r �estimatedDownloadTime� s
�z%AcquireProgress.estimatedDownloadTime)N) �__name__�
__module__� __qualname__r r r$ r* r+ r. r0 �
__classcell__r r r r r s
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �InstallProgressz_ Base class for InstallProgress that supports some fancy
stuff like apport integration
c C s t jjj�| � d | _d S �N)�apt�progress�baser5 r Z master_fdr r r r r � s zInstallProgress.__init__c
C s� zt � | jgg g | j� W n6 t jk
rP } z|jd tjkr@� W 5 d}~X Y nX | �� z&t� | j
tj �\}}|| j
kr~W q�W q t k
r� } z&|jtjkr�� |jtj
kr�W Y �q�W 5 d}~X Y q X q |S )z�Wait for child progress to exit.
The return values is the full status returned from os.waitpid()
(not only the return code).
r N)�selectZstatusfdZselect_timeout�error�args�errnoZEINTRZupdate_interface�os�waitpid� child_pid�WNOHANG�OSErrorZECHILD)r �e�pid�resr r r �
wait_child� s
zInstallProgress.wait_childc C s� | � � }|dkr�t�tjtj� z|�| j�}W nf tk
r� } zHtd|� t� d� t
dd��}|� t |�� W 5 Q R X t
�|j� W 5 d }~X Y nX t
�|� || _t
�| �� �}|S )Nr z!Exception during pm.DoInstall(): z Exception during pm.DoInstall()z./var/run/ubuntu-release-upgrader-apt-exception�w)�fork�signal�SIGPIPE�SIG_IGNZ
do_installZwritefd� Exception�printr& � exception�open�write�strr> �_exitZ
RESULT_FAILEDr@ � WEXITSTATUSrF )r ZpmrD rE rC �fr r r �run� s
zInstallProgress.runc C s� t jjj�| ||� t�d||f � d|kr:tj�|�}|� d�d �
� rdd� |� d�dd� �}d|krz|� d�d }t ||� dS )z install error from a package z*got an error from dpkg for pkg: '%s': '%s'�/�-r r Nr )
r7 r8 r9 r5 r; r& r> �path�basename�split�isdigit�joinr )r �pkgZerrormsgr r r r; � s zInstallProgress.errorN)r1 r2 r3 �__doc__r rF rU r; r r r r r5 � s
r5 c @ s e Zd Zddd�ZdS )� DumbTerminalFc C s ddl }|�|� dS )z7 expects a command in the subprocess style (as a list) r N)�
subprocess�call)r �cmdZhiddenr` r r r ra � s zDumbTerminal.callN)F)r1 r2 r3 ra r r r r r_ � s r_ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
DummyHtmlViewc C s d S r6 r )r Zurlr r r rO � s zDummyHtmlView.openc C s d S r6 r r r r r �show� s zDummyHtmlView.showc C s d S r6 r r r r r �hide s zDummyHtmlView.hideN)r1 r2 r3 rO rd re r r r r rc � s rc c @ s( e Zd ZdZdZdZdZdZdZdZ dS ) �Stepr � r � � � � N)
r1 r2 r3 ZPREPAREZMODIFY_SOURCESZFETCHZINSTALLZCLEANUPZREBOOT�Nr r r r rf s rf zPreparing to upgradez Getting new software channelszGetting new packageszInstalling the upgradesz Cleaning upc @ s� e Zd ZdZdd� Zdd� Zdd� Zd/d d
�Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zd0dd�Zd1d d �Zd2d d!�Zd"d#� Zd3d$d%�Zd4d&d'�Zd(d)� Zd5d+d,�Zd-d.� ZdS )6�DistUpgradeViewz" abstraction for the upgrade view c C s
d| _ d S )NF)Z needs_screenr r r r r s zDistUpgradeView.__init__c C s t jj�� S )z5 return a OpProgress() subclass for the given graphic)r7 r8 r9 Z
OpProgressr r r r �getOpCacheProgress s z"DistUpgradeView.getOpCacheProgressc C s t � S )z# return an acquire progress object )r r r r r �getAcquireProgress s z"DistUpgradeView.getAcquireProgressNc C s t � S )z" return a install progress object )r5 )r �cacher r r �getInstallProgress s z"DistUpgradeView.getInstallProgressc C s t � S r6 )r_ r r r r � getTerminal# s zDistUpgradeView.getTerminalc C s t � S r6 )rc r r r r � getHtmlView% s zDistUpgradeView.getHtmlViewc C s dS )z\ update the current status of the distUpgrade based
on the current view
Nr )r �msgr r r � updateStatus' s z DistUpgradeView.updateStatusc C s dS )z8 provide a visual feedback that the upgrade was aborted Nr r r r r �abort, s zDistUpgradeView.abortc C s t � �|j� dS )z� we have 6 steps current for a upgrade:
1. Analyzing the system
2. Updating repository information
3. fetch packages
3. Performing the upgrade
4. Post upgrade stuff
5. Complete
N)�
get_telemetryZ add_stage�name�r �stepr r r �setStep/ s zDistUpgradeView.setStepc C s dS )z" hide a certain step from the GUI Nr ry r r r �hideStep: s zDistUpgradeView.hideStepc C s dS )z" show a certain step from the GUI Nr ry r r r �showStep= s zDistUpgradeView.showStepTc
C s� d| _ || _g | _g | _g | _g | _g | _g | _|D ]�}|jrL| j� |� q4|j
r`| j� |� q4|j rt| j� |� q4|j r�|j
j�|j�r�| j� |� q�| j� |� q4|jr4| j� |� q4| j| j | _| j�� | j�� | j�� | j�� | j�� d}t| j�t| j� } t| j�t| j� }
t| j�} t| j�dk�rl|tddt| j��dt| j�i 7 }|d7 }| dk�r�|tdd | �| 7 }|d
7 }|
dk�r�|td d |
�|
7 }|d
7 }| dk�r�|td
d| �| 7 }|d
7 }|dk�r*t�|�} t| t��r| �t�} |td�| 7 }|| � � � |�7 }| |
dk�r|| |
| dk�r|| � � � � �rj|dtd� 7 }n|dtd� 7 }n| dk�r�|dtd� 7 }| |
| dk �r�td�}td�}| � ||� dS || _ dS )zm display the list of changed packages (apt.Package) and
return if the user confirms them
r �
r zo%(amount)d installed package is no longer supported by Canonical. You can still get support from the community.zq%(amount)d installed packages are no longer supported by Canonical. You can still get support from the community.Zamountz
z"%d package is going to be removed.z$%d packages are going to be removed.� z(%d new package is going to be installed.z*%d new packages are going to be installed.z#%d package is going to be upgraded.z%%d packages are going to be upgraded.z&
You have to download a total of %s. �d z
%sznInstalling the upgrade can take several hours. Once the download has finished, the process cannot be canceled.z{Fetching and installing the upgrade can take several hours. Once the download has finished, the process cannot be canceled.z.Removing the packages can take several hours. r z,The software on this computer is up to date.zRThere are no upgrades available for your system. The upgrade will now be canceled.FT) ZconfirmChangesMessage� demotionsZ toInstallZ toReinstallZ toUpgradeZtoRemoveZ toRemoveAutoZ toDowngradeZmarked_install�appendZmarked_upgradeZmarked_reinstallZ
marked_deleteZ_pcacheZ _depcacheZis_auto_installedZ_pkgZmarked_downgrade�sort�lenr r% Z size_to_str�
isinstance�bytes�decode�ENCODINGr ro r0 r. r; )
r �summaryZchangesr� Z downloadSizeZactionsZ removal_boldr] rt Z pkgs_removeZ pkgs_instZ pkgs_upgradeZdownloadSizeStrr r r �confirmChanges@ s�
� �
��
��
��
�
z DistUpgradeView.confirmChanges�Noc C s dS )z0 ask a Yes/No question and return True on 'Yes' Nr �r r� rt �defaultr r r �askYesNoQuestion� s z DistUpgradeView.askYesNoQuestion�Cancelc C s dS )z= ask a Cancel/Continue question and return True on 'Continue'Nr r� r r r �askCancelContinueQuestion� s z)DistUpgradeView.askCancelContinueQuestionc C s t d�}t d�}| �||�S )z2 generic ask about the restart, can be overridden zReboot requiredzMThe upgrade is finished and a reboot is required. Do you want to do this now?)r r� )r r� rt r r r �confirmRestart� s z DistUpgradeView.confirmRestartc C s dS )z display a error Nr �r r� rt Z extended_msgr r r r; � s zDistUpgradeView.errorc C s dS )z display a information msgNr r� r r r � information� s zDistUpgradeView.informationc C s dS )zQ process gui events (to keep the gui alive during a long
computation Nr r r r r �
processEvents� s z DistUpgradeView.processEventsFc C s dS )zX do a progress pulse (e.g. bounce a bar back and forth, show
a spinner)
Nr )r Zfinishedr r r �
pulseProgress� s z DistUpgradeView.pulseProgressc C s | � ||d�|�� dS )zr
show demoted packages to the user, default implementation
is to just show a information dialog
r~ N)r� r\ )r r� rt r� r r r �
showDemotions� s z DistUpgradeView.showDemotions)N)NT)r� )r� )N)N)F)r1 r2 r3 r^ r rn ro rq rr rs ru rv r{ r| r} r� r� r� r� r; r� r� r� r� r r r r rm s.
�
c
rm �__main__)+ZDistUpgradeGettextr r r Z telemetryr rw r7 �enumr r= r> r% Zlocaler& rI r: ZDistUpgradeApportr � setlocale�LC_ALLZgetdefaultlocale�coder� rN �putenvr r; �infoZ getlocaler r8 r9 r r5 �objectr_ rc rf rm r1 �fpr+ r r r r �<module> sV
D6@ � 3