CasperSecurity
o
S(e�& � @ sP z1d dl mZ e� dkre�ddlmZ ddlmZmZmZ ddl m
Z
ddl m Z m
Z
mZmZ W n+ ey\ ddlmZmZmZ dd lm Z m
Z
mZm
Z
dd
lmZ ddlmZ Y nw dd lZdd lmZ dd
lmZ ddl m Z ddl m Z dd l Z dd l!Z!d dl"m"Z" dd� Z#dd� Z$dd� Z%G dd� de�Z&G dd� de!j'j(j)�Z*d S )� )�get_dist�trusty� )�uic)� QTranslator� PYQT_VERSION�QLocale)�QIcon)�QDialog�QDialogButtonBox� QMessageBox� QApplication)�KIcon� KMessageBox�KStandardGuiItem)r
r r
r )r N)�DistUpgradeFetcherCore)�gettext)�urlopen)� HTTPError)�
QUrlOpenerc C s� t �� sCdtjd< dtjd< dtjd< dtjd< t dg�} t| �}tt�tkr2|�t �
� d d
d � n
|�t �
� d d
d � | � |� | S t �� S )
N�kde�QT_PLATFORM_PLUGIN�TRUE�KDE_FULL_SESSION�5�KDE_SESSION_VERSION�QT_QPA_PLATFORMTHEMEzubuntu-release-upgrader�qt�_z/usr/share/qt5/translationsz/usr/share/qt4/translations) r
�instance�os�environr �typer �int�loadr �system�installTranslator)�app�
translator� r) �C/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeFetcherKDE.py�_ensureQApplicationE s"
��
r+ c C s0 t t�tkrt�d d| � d S t�d | d� d S )N� )r" r r# r �warningr �sorry)�textr) r) r* �_warningd � r0 c C s t t�tkr t�| �S t| �S �N)r" r r# r � fromThemer )�namer) r) r* �_iconk s
r5 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�DistUpgradeFetcherKDEc C sB t �| ||� t� | _| j�td�� || _t� �� t �
� d S )Nzsystem-software-update) r �__init__r+ r' �
setWindowIconr5 �datadirr �setupUrlHandlesr
�
processEvents)�self�new_dist�progress�parentr9 r) r) r* r7 t s
z DistUpgradeFetcherKDE.__init__c C s0 t t�tkrt�d ||� d S t�d ||� d S r2 )r" r r# r �criticalr r. )r< �summary�messager) r) r* �error� r1 zDistUpgradeFetcherKDE.errorc C sD t �� dkrt �dd| jd g� d S t �| j| jdg| j � d S )Nr z/usr/bin/pkexec�pkexecz --frontend=DistUpgradeViewKDEz --frontend=DistUpgradeViewKDE)r �getuid�execv�script� run_options�r< r) r) r* �runDistUpgrader� s ����z%DistUpgradeFetcherKDE.runDistUpgraderc C s^ t � | _t�| jd | j� | jj�tj�}|� t
d�� |� t d�� | jj�tj
�}|� t
d�� |� t d�� | j�t
d�� | j�� | jjd ur�| �| jj�}d }zt|�}|�� �dd�}| jj� |� | j�� }W n5 ty� d t
d
� }t
d �}t|d | � Y n ty� d t
d
� }t
d�}t|d | � Y nw |t jkr�dS dS )Nz/dialog_release_notes.uiz&Upgradez dialog-okz&Cancelz
dialog-cancelz
Release NoteszUTF-8�replacez+<span weight="bold" size="larger">%s</span>z Could not find the release notesz The server may be overloaded. z<br />z$Could not download the release notesz&Please check your internet connection.TF) r
�dialogr �loadUir9 � buttonBox�buttonr �Ok�setTextr �setIconr5 �Cancel�setWindowTitle�showr= �releaseNotesHtmlUri�
_expandUrir �read�decode�scrolled_notes�exec_r r0 �IOError�Accepted) r< �
upgradeButton� cancelButton�uri�result�
release_notes�notes�primary� secondaryr) r) r* �showReleaseNotes� sD
� ��
z&DistUpgradeFetcherKDE.showReleaseNotesN)�__name__�
__module__� __qualname__r7 rC rJ rf r) r) r) r* r6 r s
r6 c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�KDEAcquireProgressAdapterc C sn t � | _t� | _tj�|d�}t�|| j� | j� t
d�� | jj � |� | jj
j�| j� d| _t�� d S )Nzfetch-progress.ui�UpgradeT)r+ r' r
rL r �path�joinr rM rT r �installingLabelrQ rN �rejected�connect�abort� _continuer
r; )r< r? r9 �label�uiFiler) r) r* r7 � s z"KDEAcquireProgressAdapter.__init__c C s
d| _ d S )NF)rr rI r) r) r* rq � s
z KDEAcquireProgressAdapter.abortc C s. | j j�td�� | j j�d� | j �� d S )N�'Downloading additional package files...r )rL rn rQ r �installationProgress�setValuerU rI r) r) r* �start� s
�z KDEAcquireProgressAdapter.startc C s | j �� d S r2 )rL �hiderI r) r) r* �stop� s z KDEAcquireProgressAdapter.stopc C s� t jjj�| |� | jj�| j| j t
| j | j � d � | j d }|| j kr+| j }t
d�}| jdkrF|t
d�| j | j t�| j�f 7 }n |t
d�| j | j f 7 }| jj�|� t�� | jS )N�d r ru r zFile %s of %s at %sB/sz
File %s of %s)�aptr> �base�AcquireProgress�pulserL rv rw �
current_bytes�
current_items�float� total_bytes� total_itemsr � current_cps�apt_pkg� size_to_strrn rQ r
r; rr )r< �owner� current_item�
label_textr) r) r* r � s.
���
��z KDEAcquireProgressAdapter.pulsec C s� t d�||f }tt�tkr$t�d t d�|tjtj�}|tjkr"dS dS t� d t d�t d�d | t
� � t
� � �}|tj
kr@dS dS )Nz&Please insert '%s' into the drive '%s'z Media ChangeTz<br>F)r r" r r# r �questionrP rS r �
questionYesNor �ok�cancel�Yes)r< �medium�drive�msg�changer) r) r* � mediaChange� s$ �
�
��
z%KDEAcquireProgressAdapter.mediaChangeN) rg rh ri r7 rq rx rz r r� r) r) r) r* rj � s rj )+�utilsr � ImportError�PyQt5r � PyQt5.QtCorer r r � PyQt5.QtGuir �PyQt5.QtWidgetsr
r r r
� PyKDE4.kdeuir r r � PyQt4.QtGui� PyQt4.QtCore�PyQt4r� �"DistUpgrade.DistUpgradeFetcherCorer r r �urllib.requestr � urllib.errorr r r| r r+ r0 r5 r6 r> r} r~ rj r) r) r) r* �<module> s6
� F