Edit File: system_upgrade.cpython-36.pyc
3 �*�d�h � @ s� d Z ddlmZmZmZmZ ddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZmZ ddlZddlZddlmZ ddlmZ ddlZddlmZmZ ddlZed � e j d �Z e j d �Z e j d �Z e j d �Z e Z!dZ"ed�Z#ed�Z$ed�Z%dZ&dd� Z'dd� Z(g fdd�Z)d7dd�Z*dd � Z+G d d � d e,�Z-G d d � d e,�Z.e.� Z/G d!d"� d"ej0j1�Z2d#d$� Z3d%d&� Z4d'd(� Z5d)d*� Z6d+d,dd-d.gZ7G d/d0� d0ej8�Z9G d1d2� d2ej:j;�Z<G d3d4� d4e<�Z=G d5d6� d6e<�Z>dS )8zGsystem_upgrade.py - DNF plugin to handle major-version system upgrades.� )�call�Popen� check_output�CalledProcessErrorN)�journal)�_�logger)�CliError)�ucd)�serialize_transaction�TransactionReplayzthe color of the skyZ 9348174c5cc74001a71ef26bd79d302eZ fef1cc509d5047268b83a3a553f54b43Z 3e0a5636d16b4ca4bbe5321d06c6aa62Z 8cec00a1566f4d3594f116450395f06cz/usr/bin/plymouthz<Need a --releasever greater than the current system version.z�Download complete! Use 'dnf {command} reboot' to start the upgrade. To remove cached metadata and transaction use 'dnf {command} clean'zESorry, you need to use 'download --releasever' instead of '--network'� c C s. t jddd�r tjtd�� n tddg� d S )NZ DNF_SYSTEM_UPGRADE_NO_REBOOTF)�defaultz!Reboot turned off, not rebooting.Z systemctl�reboot)�os�getenvr �infor r � r r �$/usr/lib/python3.6/system_upgrade.pyr E s r c C s| d} xrdD ]j}yNt |��<}x4|D ],}|j� }|j| �r |t| �d � jd�S q W W d Q R X W q tk rr w Y q X q W d S )NzUPGRADE_GUIDE_URL=�/etc/os-release�/usr/lib/os-release�")r r )�open�strip� startswith�len�IOError)�key�pathZ release_file�liner r r �get_url_from_os_releaseL s ( r c C s~ t jj| �sd S xht j| �D ]Z}t jj| |�}||kr8q y(t jj|�rTtjj|� n t j|� W q t k rt Y q X q W d S )N) r r �isdir�listdir�join�dnf�utilZrm_rf�unlink�OSError)r �ignore�entryZfullpathr r r � clear_dir[ s r* c C s6 t jj| j�| jkr tt��|r2|| jkr2tt��d S )N)r$ �rpm�detect_releasever� installroot� releaseverr �RELEASEVER_MSG�CANT_RESET_RELEASEVER)�conf�targetr r r �check_release_verl s r3 c C sP yt dd�} | jd� W n2 tk rJ } zttd�| � W Y d d }~X nX d S )Nz /dev/tty0�wbs [9;0]z%Screen blanking can't be disabled: %s)r �write� Exception�printr )Ztty�er r r �disable_blankingu s r9 c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e d�Z e d�Z e d�Z e d�Z e d�Ze d�Ze d�Ze d�Ze d�Ze d�Ze d�Ze d�Ze d�Zd S ) �Statec C s || _ i | _| j� d S )N)� statefile�_data�_read)�selfr; r r r �__init__� s zState.__init__c C sp y&t | j��}tj|�| _W d Q R X W nD tk r@ i | _Y n, tk rj i | _tjt d�| j� Y nX d S )Nz;Failed loading state file: %s, continuing with empty state.) r r; �json�loadr< r � ValueErrorr �warningr )r> �fpr r r r= � s z State._readc C sF t jjtjj| j�� t| jd��}tj | j |ddd� W d Q R X d S )N�w� T)�indent� sort_keys) r$ r% � ensure_dirr r �dirnamer; r r@ �dumpr< )r> Zoutfr r r r5 � s z State.writec C s& t jj| j�rt j| j� | j� d S )N)r r �existsr; r& r= )r> r r r �clear� s z State.clearc C s | S )Nr )r> r r r � __enter__� s zState.__enter__c C s |d kr| j � d S )N)r5 )r> �exc_type� exc_value� tracebackr r r �__exit__� s zState.__exit__c s" � fdd�}� fdd�}t ||�S )Nc s || j � < d S )N)r< )r> �value)�optionr r �setprop� s z State._prop.<locals>.setpropc s | j j� �S )N)r< �get)r> )rT r r �getprop� s z State._prop.<locals>.getprop)�property)rT rU rW r )rT r �_prop� s z State._prop� state_version�download_status�destdir�target_releasever�system_releasever�gpgcheck�gpgcheck_repos�repo_gpgcheck_repos�upgrade_status�upgrade_command� distro_sync�enable_disable_repos�module_platform_idN)�__name__� __module__� __qualname__r? r= r5 rM rN rR rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf r r r r r: � s( r: c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�PlymouthOutputz�A plymouth output helper class. Filters duplicate calls, and stops calling the plymouth binary if we fail to contact it. c C s d| _ t� | _d | _d S )NT)�alive�dict� _last_args� _last_msg)r> r r r r? � s zPlymouthOutput.__init__c G sj || j j|�k}| jr | s$|dkrdytt|f| �dk| _W n tk rX d| _Y nX || j |< | jS )Nz--pingr F)rm rV rk r �PLYMOUTHr' )r> �cmd�argsZdupe_cmdr r r � _plymouth� s zPlymouthOutput._plymouthc C s | j d�S )Nz--ping)rr )r> r r r �ping� s zPlymouthOutput.pingc C s4 | j r | j |kr | jdd| j � || _ | jdd|�S )Nz hide-messagez--textzdisplay-message)rn rr )r> �msgr r r �message� s zPlymouthOutput.messagec C sR d}y$t tdg�}tjdt|��r&d}W n ttfk r@ Y nX | jdd| �S )NZupdatesz--helpz--system-upgradezsystem-upgradez change-modez--)r ro �re�searchr r r' rr )r> �mode�sr r r �set_mode� s zPlymouthOutput.set_modec C s | j ddt|��S )Nz system-updatez --progress)rr �str)r> Zpercentr r r �progress� s zPlymouthOutput.progressN) rg rh ri �__doc__r? rr rs ru rz r| r r r r rj � s rj c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�PlymouthTransactionProgressc C s | j ||||� d S )N)�_update_plymouth)r> �package�actionZti_doneZti_totalZts_doneZts_totalr r r r| � s z$PlymouthTransactionProgress.progressc C sd |dkr d S |t jjkr0tjtd| | �� ntjdtd| | � � tj| j||||�� d S )N� g �V@�Z g $@)r$ �callbackZ PKG_VERIFY�Plymouthr| �intru � _fmt_event)r> r� r� �current�totalr r r r � s z,PlymouthTransactionProgress._update_plymouthc C s t jjj||�}d||||f S )Nz[%d/%d] %s %s...)r$ � transactionZACTIONSrV )r> r� r� r� r� r r r r� s z&PlymouthTransactionProgress._fmt_eventN)rg rh ri r| r r� r r r r r~ � s r~ c c sJ t j� }|j| jdd� d}x(|D ] }|d }||kr8q"|}|V q"W dS )zVFind all boots with this message id. Returns the entries of all found boots. r )� MESSAGE_IDZ_UIDN�_BOOT_ID)r �ReaderZ add_match�hex)� message_id�jZoldbootr) Zbootr r r � find_boots s r� c C st t td�� d } xJttt��D ]:\} }t dj| d |d |d |jdd�|jdd��� q W | d krpt td �� d S ) Nz3The following boots appear to contain upgrade logs:r� u) {} / {.hex}: {:%Y-%m-%d %H:%M:%S} {}→{}r� Z__REALTIME_TIMESTAMP�SYSTEM_RELEASEVERz??�TARGET_RELEASEVERz-- no logs were found --���r� )r7 r � enumerater� �ID_TO_IDENTIFY_BOOTS�formatrV )�nr) r r r � list_logs s r� c C sZ t t| ��}y(|dkrt�|dkr*|d8 }|| d S tk rT ttd���Y nX d S )Nr r� r� z!Cannot find logs with this index.)�listr� � IndexErrorr r )r� r� Zbootsr r r � pick_boot. s r� c C sD t t| �}tdd|jg�}|j� |j}|dkr@tjjt d���d S )NZ journalctlz--bootr� z%Unable to match systemd journal entry) r� r� r r� �wait� returncoder$ � exceptions�Errorr )r� Zboot_idZprocessZrcr r r �show_log= s r� ZdownloadZclean�upgrade�logc s e Zd ZdZ� fdd�Z� ZS )�SystemUpgradePluginzsystem-upgradec s8 t t| �j||� |r4|jt� |jt� |jt� d S )N)�superr� r? Zregister_command�SystemUpgradeCommand�OfflineUpgradeCommand�OfflineDistrosyncCommand)r> �base�cli)� __class__r r r? N s z SystemUpgradePlugin.__init__)rg rh ri �namer? � __classcell__r r )r� r r� K s r� c s( e Zd ZdEZed�ZdZ� fdd�Zedd� �Z d d � Z d d � Z d d� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#d=d>� Z$d?d@� Z%dAdB� Z&dCdD� Z'� Z(S )Fr� �system-upgrade�fedupz+Prepare system for upgrade to a new releasezvar/lib/dnf/system-upgradec sj t t| �j|� tjj|jjj| j �| _ tjj| j d�| _ tjj|jjjd�| _ t tjj| j d��| _d S )Nz system-upgrade-transaction.jsonz system-updatezsystem-upgrade-state.json)r� r� r? r r r# r� r1 r- �DATADIR�datadir�transaction_file� magic_symlinkr: �state)r> r� )r� r r r? \ s z SystemUpgradeCommand.__init__c C sJ | j dddtd�d� | j ddtdd jt� d � | j d ttd �d � d S )Nz--no-downgraderd Z store_falsez=keep installed packages if the new release's version is older)�destr� �help�tidr� z[%s]�|)�nargs�choices�metavarz--numberzwhich logs to show)�typer� )� add_argumentr �CMDSr# r� )�parserr r r � set_argparserd s z"SystemUpgradeCommand.set_argparserc C s( t j||t j| jj| jjtjjd� dS )z Log directly to the journal.)r� ZPRIORITYr� r� Z DNF_VERSIONN) r �sendZ LOG_NOTICEr� r^ r] r$ �const�VERSION)r> ru r� r r r � log_statusn s z SystemUpgradeCommand.log_statusc C s | j d� | j d� d S )NZcheck� pre_configure)� _call_sub)r> r r r r� w s z"SystemUpgradeCommand.pre_configurec C s | j d� d S )N� configure)r� )r> r r r r� { s z SystemUpgradeCommand.configurec C s | j d� d S )N�run)r� )r> r r r r� ~ s zSystemUpgradeCommand.runc C s | j d� d S )Nr� )r� )r> r r r �run_transaction� s z$SystemUpgradeCommand.run_transactionc C s | j d� d S )NZresolved)r� )r> r r r � run_resolved� s z!SystemUpgradeCommand.run_resolvedc C s. t | |d | jjd d �}t|�r*|� d S )Nr r )�getattr�optsr� �callable)r> r� Zsubfuncr r r r� � s z SystemUpgradeCommand._call_subc C s( | j jtkr$td�j|d�}t|��d S )NzFIncompatible version of data. Rerun 'dnf {command} download [OPTIONS]')�command)r� rZ � STATE_VERSIONr r� r )r> r� rt r r r �_check_state_version� s z)SystemUpgradeCommand._check_state_versionc C s* | j | jj_| jjr | jjnd | jj_d S )N)r� r� r1 �cachedirr� r\ )r> r r r � _set_cachedir� s z"SystemUpgradeCommand._set_cachedirc C s� t tjjtjjg �}t tjj�}i }i }xl| jjjD ]^}|j |krp|j }|j |j |j ji �j t|�i �|j < q6|j |kr6|j |j t|j �i �|j < q6W ||fS )z� forward = {repoid:{pkg_nevra: {tsi.action: tsi.reason}} reverse = {pkg_nevra: {tsi.action: tsi.reason}} :return: forward, reverse )�setr$ r� ZBACKWARD_ACTIONS�libdnfZ!TransactionItemAction_REINSTALLEDZFORWARD_ACTIONSr� r� r� �pkg�reason� setdefault�repo�idr{ )r> Zbackward_actionZforward_actionsZforward�reverseZtsir� r r r �%_get_forward_reverse_pkg_reason_pairs� s & z:SystemUpgradeCommand._get_forward_reverse_pkg_reason_pairsc C sb | j | jj_| jjr | jjnd | jj_d| jjkrJ| jj rJtt d���nd| jjkr^d| j_d S )Nzoffline-distrosynczFCommand 'offline-distrosync' cannot be used with --no-downgrade optionzoffline-upgradeF) r� r� r1 r� r� r\ r� rd r r )r> r r r �pre_configure_download� s z+SystemUpgradeCommand.pre_configure_downloadc C s | j � d S )N)r� )r> r r r �pre_configure_reboot� s z)SystemUpgradeCommand.pre_configure_rebootc C s. | j � | jjr | jj| j_| jj| jj_d S )N) r� r� re r� �repos_edr] r� r1 r. )r> r r r �pre_configure_upgrade� s z*SystemUpgradeCommand.pre_configure_upgradec C s | j � d S )N)r� )r> r r r �pre_configure_clean� s z(SystemUpgradeCommand.pre_configure_cleanc C s d| j jksd| j jkr�tjtd�� t� }|rLtd�}tj|jt|��� | j j � r�td�}| j j j s�| j j jdj|�dj|�d� r�tjtd �� tjd � t| j j | j jd � nd | j jkr�| jj| j � d | jj_d | jj_d | jj_d | jj_d | jj_| j j j dg7 _ d S )Nzsystem-upgrader� z\WARNING: this operation is not supported on the RHEL distribution. Proceed at your own risk.z-Additional information for System Upgrade: {}zyBefore you continue ensure that your system is fully upgraded by running "dnf --refresh upgrade". Do you want to continuez {} [y/N]: z {} [Y/n]: )rt Zdefaultyes_msgzOperation aborted.r� )r2 zoffline-upgradeTZtest) r� r� r rC r r r r� r r� Z _promptWantedr1 Zassumeno�outputZ userconfirm�error�sys�exitr3 r. r� Z _populate_update_security_filter�demands� root_user� resolving�available_repos�sack_activationZfreshest_metadataZtsflags)r> Zhelp_urlrt r r r �configure_download� s* z'SystemUpgradeCommand.configure_downloadc C s d| j j_d S )NT)r� r� r� )r> r r r �configure_reboot� s z%SystemUpgradeCommand.configure_rebootc C s� d| j j_d| j j_d| j j_d| j j_| jj| j_| jj d k rN| jj | j j _ | jj d k r�x$| j j j� D ]}|j| jj k|_ qhW | jjd k r�x$| j j j� D ]}|j| jjk|_q�W | jj| j j _d| j j_d| j j _t� | j j_d| j j _d| j j _d S )NTF)r� r� r� r� r� r� r� rd r� r_ r� r1 r` �repos�valuesr� ra � repo_gpgcheckrf Z cacheonlyZ assumeyesr~ Ztransaction_displayZ clean_requirements_on_removeZinstall_weak_deps)r> r� r r r �configure_upgrade� s& z&SystemUpgradeCommand.configure_upgradec C s d| j j_d S )NT)r� r� r� )r> r r r �configure_clean s z$SystemUpgradeCommand.configure_cleanc C s d S )Nr )r> r r r � configure_log s z"SystemUpgradeCommand.configure_logc C s~ | j jdksttd���| j| jj� | j j| jjkrRtd�j| jjd�}t|��t j j | j �rlttd���t jj| j� d S )N�completez system is not ready for upgradezZthe transaction was not prepared for '{command}'. Rerun 'dnf {command} download [OPTIONS]')r� z upgrade is already scheduled)r� r[ r r r� r� r� rc r� r r �lexistsr� r$ r% rI r� )r> rt r r r � check_reboot s z!SystemUpgradeCommand.check_rebootc C s� t jj| j�s$tjtd�� td��t j| j�| j krLtjtd�� td��t j j j | j� | jj}|sp| jj}| j|� | jjdks�td�j|d�}t|��d S )Nz-trigger file does not exist. exiting quietly.r z1another upgrade tool is running. exiting quietly.�readyz/use 'dnf {command} reboot' to begin the upgrade)r� )r r r� r� r r r � SystemExit�readlinkr� r$ ZyumZmiscZunlink_fr� rc r� r� r� rb r� r )r> r� rt r r r � check_upgrade s z"SystemUpgradeCommand.check_upgradec C s, t j| j| j� | j� }d|_W d Q R X d S )Nr� )r �symlinkr� r� r� rb )r> r� r r r � run_prepare, s z SystemUpgradeCommand.run_preparec C s6 | j � | jjd dks d S | jtd�t� t� d S )Nr r z Rebooting to perform upgrade.)r� r� r� r� r �REBOOT_REQUESTED_IDr )r> r r r � run_reboot3 s z SystemUpgradeCommand.run_rebootc s� � j jr� jj� n � jj� � j jdkr�� jj� � fdd�� jjjD �}|r\� jj|� � fdd�� jjj D �}|r�� jj|� � j �$}d|_ � jj j |_� jj j|_W d Q R X d S )N�offline-upgrade�offline-distrosyncc s$ g | ] }� j jjj|j�r|j�qS r )r� �history�grouprV r� )�.0�g)r> r r � <listcomp>G s z5SystemUpgradeCommand.run_download.<locals>.<listcomp>c s$ g | ] }� j jjj|j�r|j�qS r )r� r� �envrV r� )r r )r> r r r J s Z downloading)r� r� )r� rd r� Z upgrade_allr� Z read_comps�comps�groupsZenv_group_upgradeZ environmentsr� r[ r1 r. r] r\ )r> Zinstalled_groupsZinstalled_environmentsr� r )r> r � run_download= s z!SystemUpgradeCommand.run_downloadc C s� d}| j �}d|_|j}W d Q R X |dkr4td�}n|dkrFtd�}ntd�}| j|t� tj� tjd� tj |� t � t | j | j �| _| jj� d S ) N� Z incompletezoffline-upgradez1Starting offline upgrade. This will take a while.zoffline-distrosyncz4Starting offline distrosync. This will take a while.z0Starting system upgrade. This will take a while.r )r� rb rc r r� �UPGRADE_STARTED_IDr� rz r| ru r9 r r� r� �replayr� )r> r� r� rt r r r � run_upgradeS s z SystemUpgradeCommand.run_upgradec C sd t jtd�� t| jjjtjj | jjj�j g� | j �$}d |_ d |_ d |_d |_d |_W d Q R X d S )Nz Cleaning up downloaded data...)r r r r* r� r1 r� r$ Z persistorZTempfilePersistorZdb_pathr� r[ rZ rb rc r\ )r> r� r r r � run_cleanm s z SystemUpgradeCommand.run_cleanc C s | j jrt| j j� nt� d S )N)r� Znumberr� r� )r> r r r �run_logz s z SystemUpgradeCommand.run_logc C s | j j� dS )z5Adjust transaction reasons according to stored valuesN)r Zpost_transaction)r> r r r �resolved_upgrade� s z%SystemUpgradeCommand.resolved_upgradec C s� | j jj� }|j� s&tjtd�� d S t|�}yLt| j d��"}t j ||ddd� |j d� W d Q R X t td�j| j �� W n< tk r� } z tjjtd�jt|����W Y d d }~X nX tjj| j jj�}| j��}d |_t|_| j j |_ | j jj |_ d d � | j j j � D �|_!d d � | j j j � D �|_"||_#| j jj$|_%| j jj&|_&| j j'|_(| j jj)|_)| j j*|_+W d Q R X t,j| j j*d �}tj|� | j-td�t.� d S )NzKThe system-upgrade transaction is empty, your system is already up-to-date.rE rF T)rG rH � zTransaction saved to {}.z Error storing transaction: {}r� c S s g | ]}|j r|j�qS r )r_ r� )r r� r r r r � s z=SystemUpgradeCommand.transaction_download.<locals>.<listcomp>c S s g | ]}|j r|j�qS r )r� r� )r r� r r r r � s )r� zDownload finished.)/r� r� Z get_currentZpackagesr r r r r r� r@ rK r5 r7 r� r' r$ r� r r{ r+ r, r1 r- r� r[ r� rZ r� rd r_ r� r� r` ra r^ r. r] rf r� re r\ r� rc �DOWNLOAD_FINISHED_MSGr� �DOWNLOAD_FINISHED_ID)r> r� �data�fr8 Z system_verr� rt r r r �transaction_download� s: , z)SystemUpgradeCommand.transaction_downloadc C s@ t jtd�� | jtd�t� | j� | jjd dkr<t� d S )Nz.Upgrade complete! Cleaning up and rebooting...r r� ) r� ru r r� �UPGRADE_FINISHED_IDr r� r� r )r> r r r �transaction_upgrade� s z(SystemUpgradeCommand.transaction_upgrade)r� r� ))rg rh ri �aliasesr �summaryr� r? � staticmethodr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r� r r )r� r r� V sF (r� c @ s e Zd ZdZed�ZdS )r� �offline-upgradez%Prepare offline upgrade of the systemN)r )rg rh ri r r r r r r r r� � s r� c @ s e Zd ZdZed�ZdS )r� �offline-distrosyncz(Prepare offline distrosync of the systemN)r )rg rh ri r r r r r r r r� � s r� )N)?r} � subprocessr r r r r@ r Zos.pathrv r� ZuuidZsystemdr Zdnfpluginscorer r r$ Zdnf.clir Zdnf.i18nr Zdnf.transactionZdnf.transaction_srr r Z libdnf.confr� ZUUIDr r� r r r� ro r/ r r0 r� r r r* r3 r9 �objectr: rj r� r� ZTransactionProgressr~ r� r� r� r� r� ZPluginr� r� ZCommandr� r� r� r r r r �<module> sd @. e