CasperSecurity
o
>I�c�$ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m Z d dl
m Z d dlmZ g d�Z
ddlmZ dd � Zd
d � Zd d
� Zdd� Zdd� Zdd� ZdS )� N)�datetime)� OptionParser)�gettext)
z/binz/bootz/etcz/initrdz/libz/lib32z/lib64z/sbinz/usrz/var� )�DistUpgradeConfigc C s� t � } | jddddd� | jdddd� | jd dd
d� | jd d d
td�d� | jdddtd�d� | jddddtd�d� | jdddtd�d� | jdd d td �d� | jd dd dtd!�d"� | �� S )#z/ setup option parser and parse the commandline z--have-prerequists�havePrerequists�
store_trueF)�dest�action�defaultz--with-network� withNetwork)r r
z--without-network� store_falsez
--frontend�frontendNz_Use frontend. Currently available:
DistUpgradeViewText, DistUpgradeViewGtk, DistUpgradeViewKDE)r r �helpz--mode�mode�desktopz(*DEPRECATED* this option will be ignoredz --partial�partialz:Perform a partial upgrade only (no sources.list rewriting))r r r
r z--disable-gnu-screenzDisable GNU screen support)r
r r z --datadir�datadir�.z Set datadirz--devel-release�
devel_releasez"Upgrade to the development release)r
r r r )r �
add_option�_�
parse_args)�parser� r �=/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeMain.py�do_commandline6 s8 �
�
�
��
��r c
C s^ |� ddd�}tj�|�st�|� t�|d �rQt�� }|d|j|j |j
|j |j f }tj�|�s8t�|� t�|d �D ]}t
�|tj�|tj�|��� q?tj�|d�}| jr_|d7 }t|d�� W d � n1 sow Y tjtj|d
d d � t�d
|j � t�dd�t�� � � tj�d �}|d j j }t�d| � t�dt j � |S )z setup the logging �Files�LogDirz/var/log/dist-upgrade/z/*.logz/%04i%02i%02i-%02i%02izmain.logz.partial�aNz%%(asctime)s %(levelname)s %(message)s�w)�level�filename�format�filemodezUsing config files '%s'zuname information: '%s'� �aptzapt version: '%s'zpython version: '%s') �getWithDefault�os�path�exists�mkdir�globr �now�year�month�day�hour�minute�shutil�move�join�basenamer �open�logging� basicConfig�DEBUG�info� config_files�unamer& �apt_pkg�Cache� current_ver�ver_str�sys�version) �options�config�logdirr- �
backup_dir�f�fname�cache� apt_versionr r r �
setup_loggingO s8
�� rL c
C s zddl m} W n ty t�d� Y d S w tj�| d�}t�d| � t� d�}|� }|j
d|ddd � t� |� z3t j d
d gt j
dd ��� d
}ttj�| d�d��}|�|� W d � W d S 1 shw Y W d S ty� } zt�d| � W Y d }~d S d }~ww )Nr )�AptClonezfailed to import AptClonez apt-clone_system_state.tar.gzzcreating statefile: '%s'�6 �/T)� sourcedir�target�with_dpkg_status�
scrub_sources�lspciz-nn��stdout�universal_newlinesr z lspci.txtr zlspci failed: %s)� apt_clonerM � ImportErrorr8 �errorr( r) r5 �debug�umask�
save_state�
subprocess�Popen�PIPE� communicater7 �write�OSError)rF rM rQ � old_umask�clone�srH �er r r �save_system_statep s6
�
�
�� &� ��rh c C s� | j g|�dd� D ]E}|sq
z ddlm} |d| d�}t||�}||| jd�}W |S tyO } zt�d||f � t d||f � W Y d }~q
d }~ww t�
d
� t d
� t � d � |S ) z0 setup view based on the config and commandline �Viewr )�
import_moduler � DistUpgrade)rF r zcan't import view '%s' (%s)zcan't load %s (%s)Nz!No view can be imported, abortingr )
r �getlist� importlibrj �getattrr � Exceptionr8 �warning�printrZ rB �exit) rD rE rF �requested_viewrj �
view_modul�
view_class�instancerg r r r �
setup_view� s&
� ��
rw c C s� d} zt jddgt jdd��� d }t�d| � W n ty) t�d� Y d S w | |v r>t�d
� t� d dd d
d| g� dtj
d< dddddd| gt j }t�d| � t� d |� d S )z� check if there is a upgrade already running inside gnu screen,
if so, reattach
if not, create new screen window
z$ubuntu-release-upgrade-screen-window�screenz-lsTrU r zscreen returned: '%s'zscreen could not be runNz)found active screen session, re-attachingz/usr/bin/screenz-dz-rz-p�1�RELEASE_UPGRADER_NO_SCREENz-ez\0\0z-c�screenrcz-Szre-exec inside screen: '%s')
r^ r_ r` ra r8 r[ rc r; r( �execv�environrB �argv)�
SCREENNAME�out�cmdr r r �%run_new_gnu_screen_window_or_reattach� s4
��
�
��r� c C s t � \} }t| j�}t| |�}ddlm} t�d| � t� d� | jdu s+| jdkr:t
j � t
�
� d�}t�d|� t| ||�}|jrNdt
jvrN| jsNt� t
�d �sXd
t
jd < dd lm} ||| | jd �}t�|j� | jr}|�� sxt� d� t� d
� t |� |� � r�d
S dS )z
main method r )�VERSIONz%release-upgrader version '%s' startedzubuntu-release-upgraderNr �morz �NEEDRESTART_SUSPEND�y)�DistUpgradeController)r r ) r r r rL �DistUpgradeVersionr� r8 r; r �
textdomainr( r) r5 �getcwd�bindtextdomainrw � needs_screenr} �disable_gnu_screenr� �getenvr� �atexit�register�_enableAptCronJobr �doPartialUpgraderB rr rh �run) rD �argsrE rF r� � localedir�viewr� �appr r r �main� s8
�
r� )r& r� r r, r8 r( r3 r^ rB r �optparser r � SYSTEM_DIRS�DistUpgradeConfigParserr r rL rh rw r� r� r r r r �<module> s(
! !