Current File : //opt/alt/python37/lib/python3.7/site-packages/clwizard/__pycache__/wizard.cpython-37.pyc
B
�2Lb�2 � @ s$ d dl mZ d dl mZ d dlZd dlZd dlZd dlZd dlZd dlm Z m
Z
m Z m Z d dl
Z
d dlmZ d dlmZmZmZ d dlmZ dd lmZ dd
lmZ dd lmZmZm Z dd l m Z m Z m Z m!Z!m"Z" dd
l#m$Z$m%Z% ddl&m'Z' ddl(m)Z)m*Z*m+Z+ d dlm,Z, G dd� de-�Z.dS )� )�print_function)�absolute_importN)�Union�Any�Optional�Dict)�FormattedException)� run_command�ExternalProgramFailed�get_cl_version)� NoSuchModule� )�acquire_config_access)�Config)�run_installation� ALL_MODULES�get_supported_modules)� WizardStatus� ModuleStatus�CRASH_LOG_PATH�FILE_MARKER_PATH�
MAIN_LOG_PATH)�CancelModuleException�InstallationFailedException)� parse_cloudlinux_wizard_opts)� is_background_process_running�run_background� setup_logger)�get_rpm_db_errorsc @ s� e Zd ZdZejejejgZej ejej
gZ dd� Z dd� Z
edd� �Zdd � Zd#d d �Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zed$d!d"��Zd
S )%�CloudlinuxWizardzAMain class for working with Wizard that exposes high level logic.c C s d | _ t� | _tdt�| _d S )Nz wizard.main)�_optsr �_supported_modulesr r �log)�self� r$ �@/opt/alt/python37/lib/python3.7/site-packages/clwizard/wizard.py�__init__@ s zCloudlinuxWizard.__init__c
C s� t |�| _y�| jjdkr`| �� | �� r>| jjs>| jddd� | jjrNt� q�| j | jj
d� nf| jjdkr�| �� | jj r�| � � q�| �
� n6| jjdkr�| �| jj� n| jjdkr�| �� nt�| jjd kr�| �� s�| jjdkr�| �� s�| �� | �� | �� W n� tk
�rV } z.| j�d
t|�� | j|j|j|jdd � W d d }~X Y nj t k
�rx | jd
dd� Y nH t k
�r� } z(| j� dt|�� | jdt � � d� W d d }~X Y nX d S )zv
CL Wizard main function
:param argv: command line arguments for wizard
:return: None
�installz Installation already finishedr
)�result� exit_code)�options�status�cancelZfinish)r' r, z>We got an error while running cloudlinux-wizard, message: '%s')r( �context�detailsr) Nz8Module installation failed, see log for more informationz&Unknown error in cloudlinux-wizard, %sz\Unknown error happened, please, try again or contact CloudLinux support if it happens again.)r( r. )!r r Z subparser�_validate_system�is_installation_finished�force�_print_result_and_exitZno_asyncr �run_background_installationZ json_data�initial�_get_initial_status�_get_modules_statuses�_cancel_module_installation�module�create_completion_marker�NotImplementedError�is_all_modules_installed�run_collecting_statistics�run_cagefs_force_updater r" �error�str�messager- r. r � ExceptionZ exception� traceback�
format_exc)r# �argv�er$ r$ r% �runE sH
$zCloudlinuxWizard.runc C s t j�t�S )N)�os�path�isfiler r$ r$ r$ r% r0 ~ s z)CloudlinuxWizard.is_installation_finishedc
C sf yt �t� | j�d� W nF ttfk
r` } z$| j�dt|�� | j ddd� W d d }~X Y nX d S )Nz Wizard was successfully finishedz1Finish command called more than once, error: '%s'z$Finish command called more than oncer
)r( r) )
rG �mknodr r" �info�OSError�IOError�warningr? r2 )r# �errr$ r$ r% r9 � s
z)CloudlinuxWizard.create_completion_markerNc C sr t jd d � }|�d� t� �<}t� r4| jddd� |d k rF|�|� t|�j}||_ W d Q R X | jd|d� d S )Nz
--no-asynczGUnable to run new installation because background task is still workingr
)r( r) �success)r( �pid)
�sysrD �appendr r r2 Z set_modulesr rQ �
worker_pid)r# r* �cmd�configrT r$ r$ r% r3 � s
z,CloudlinuxWizard.run_background_installationc C s t � dkr| jdd� dS )z;
Check that wizard supports current system
Nz|Could not identify CloudLinux version. Restart your system. If you have the same problem again - contact CloudLinux support.)r( )r r2 )r# r$ r$ r% r/ � s
z!CloudlinuxWizard._validate_systemc C s | j | jS )zGet path to module log file)r! ZLOG_FILE)r# � module_namer$ r$ r% �_get_module_log_path� s z%CloudlinuxWizard._get_module_log_pathc
C s� g }t � ��}| �|�}xr| jD ]h}y|�|�}|�|�}W n tk
rP w Y nX |||d�}|tjtjgkr|| � |�|d< |�
|� q W W dQ R X |t j kr�| j
||td� | j
||d� dS )z?
Get information about background worker state
)r+ �name� status_timeZlog_fileN)�
wizard_status�modulesZ crash_log)r[ r\ )r �_get_wizard_stater! �get_module_statusZget_module_status_timer r �FAILED� AUTO_SKIPPEDrX rS r �CRASHEDr2 r )r# r\ rV �staterY r+ rZ Z
module_statusr$ r$ r% r6 � s
z&CloudlinuxWizard._get_modules_statusesc C sJ t � }|r| j|d� | jdd� | j�� D �ttt�t| j� �d� dS )zg
Get initial modules status that is used
by lvemanager to display wizard pages
)r( c S s i | ]\}}|� � � |�qS r$ )Zinitial_status)�.0rW �clsr$ r$ r% �
<dictcomp>� s z8CloudlinuxWizard._get_initial_status.<locals>.<dictcomp>)r\ Zunsuppored_by_cpN)r r2 r! �items�list�setr )r# Z
error_messager$ r$ r% r5 � s
z$CloudlinuxWizard._get_initial_statusc C sr | j �d|� t� �T}|�|�}|| jkrJ|j|tjd� | j �d|� n| j �d||� t ||��W dQ R X dS )z=Remove module from queue or print error if it is not possiblez Trying to cancel module '%s')rW Z new_statez!Module '%s' successfully canceledz<Not able to cancel module '%s', because it is in status '%s'N)
r" rK r r^ �CANCELABLE_MODULE_STATUSESZset_module_statusr � CANCELLEDrN r )r# r8 rV r+ r$ r$ r% r7 � s
z,CloudlinuxWizard._cancel_module_installationc
C sz ddg}t j�d�s |�d� | j�d� yt|�}| j�d|� W n0 tk
rt } z| j�d|� W dd}~X Y nX dS ) z,
Collects user`s statistics
z /usr/sbin/cloudlinux-summaryz--sendZSYNCHRONOUS_SUMMARYz--asynczTry to collect statistics..z-Output of statistics collection command: '%s'z>Error during collecting and sending statistics. Reason is '%s'N) rG �environ�getrS r" rK r r
r> )r# rU �outrE r$ r$ r% r< � s
z*CloudlinuxWizard.run_collecting_statisticsc s6 t � �}t|j�� �}W dQ R X t� fdd�|D ��S )z�
Check that all modules were successfully installed:
-- installed
-- canceled
-- auto-skipped
Nc 3 s | ]}|� j kV qd S )N)�DONE_MODULES_STATUSES)rc r+ )r# r$ r% � <genexpr>� s z<CloudlinuxWizard.is_all_modules_installed.<locals>.<genexpr>)r rg �statuses�values�all)r# rV rp r$ )r# r% r; � s z)CloudlinuxWizard.is_all_modules_installedc C sT d}t j�|�sdS |ddg}| j�d|� t|�}tt j�d��}|rP|� � dS )z=
Runs cagefsctl --force-update in background
z/usr/sbin/cagefsctlNz--force-updatez --wait-lockz0Starting cagefs force-update in background..: %sZCL_TEST_SYSTEM)
rG rH rI r" rK r �boolrk rl Z communicate)r# Z
cagefsctl_binrU Zcagefsctl_procZis_test_environmentr$ r$ r% r= s
z(CloudlinuxWizard.run_cagefs_force_updatec s� |j d krtjS yt�|j � W nl tjk
r� t|j�� �}t � fdd�|D ��r\tj
S t dd� |D ��r�t dd� |D ��s�tj S tj
S X tjS d S )Nc 3 s | ]}|� j kV qd S )N)rn )rc r+ )r# r$ r% ro s z5CloudlinuxWizard._get_wizard_state.<locals>.<genexpr>c s s | ]}|t jt jfkV qd S )N)r r_ rj )rc r+ r$ r$ r% ro $ s c s s | ]}|t jfkV qd S )N)r Z
INSTALLING)rc r+ r$ r$ r% ro % s )rT r ZIDLE�psutilZProcessZ
NoSuchProcessrg rp rq rr ZDONE�anyr_ ra Z IN_PROGRESS)r# rV rp r$ )r# r% r] s
z"CloudlinuxWizard._get_wizard_staterP r c K s: | t � � d�}|�|� ttj|ddd�� t�|� dS )z�
Print data in default format for web and exit
:param dict extra: extra fields in response,
usually we expect 'context' here
)r( Z timestamp� T)�indentZ sort_keysN)�time�update�print�json�dumpsrR �exit)r( r) Zextrar@ r$ r$ r% r2 + s
z'CloudlinuxWizard._print_result_and_exit)N)rP r ) �__name__�
__module__� __qualname__�__doc__r ZPENDINGr_ rj ri Z INSTALLEDr` rn r& rF � staticmethodr0 r9 r3 r/ rX r6 r5 r7 r< r; r= r] r2 r$ r$ r$ r% r 0 s. 9
r )/Z
__future__r r r{ rG rR rx rB �typingr r r r rt Zclcommonr Zclcommon.utilsr r
r Zclwizard.configr rV r r r\ r r r Z constantsr r r r r �
exceptionsr r �parserr Zutilsr r r r �objectr r$ r$ r$ r% �<module> s(
Mini Shell