File: //opt/alt/python37/lib/python3.7/site-packages/clwizard/__pycache__/wizard.cpython-37.opt-1.pyc
B
�S�;����� @ s0 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 d dlmZ d d lmZ d
d lmZ d
d lmZ d
d
l m Z m Z m Z d
dl m!Z!m"Z"m#Z#m$Z$m%Z% d
dl&m'Z'm(Z( d
dl)m*Z* d
dl+m,Z,m-Z-m.Z. G dd� de/�Z0dS )� )�print_function)�absolute_importN)�Any�Optional�Dict�NoReturn)�FormattedException)� run_command�ExternalProgramFailed�get_cl_version)�get_package_db_errors)� is_ubuntu)� 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_loggerc @ 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 � Zd d � Zed&d#d$��Z d
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% �py/clwizard/wizard.py�__init__C s zCloudlinuxWizard.__init__c
C s� t |�| _�y
| jjdkrj| �� | �� r@| jjs@| jddd� | �� | jjrXt � q�| j
| jj d� nf| jjdkr�| �� | jj r�| �
� q�| �� n6| jjdkr�| �| jj� n| jjdkr�| �� nt�| jjd kr�| �� s�| jjdk�r | �� �s | �� | �� | �� W n� tk
�rd } z.| j�d
t|�� | j|j|j|j dd � W d d }~X Y nj t k
�r� | 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;Got an error while running cloudlinux-wizard, message: '%s')r) �context�detailsr* Nz<Module installation failed, see the log for more informationz&Unknown error in cloudlinux-wizard, %szVUnknown error occured, please, try again or contact CloudLinux support if it persists.)r) r/ )"r r! Z subparser�_validate_system�is_installation_finishedZforce�_print_result_and_exit�_prepare_for_installationZno_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�errr% r% r& �runH sR
$ zCloudlinuxWizard.runc C s t j�t�S )N)�os�path�isfiler r% r% r% r& r1 � 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 )NzWizard execution completez:Wizard 'finish' command called more than once, error: '%s'z-Wizard 'finish' command called more than oncer )r) r* )
rH �mknodr r# �info�OSError�IOError�warningr@ r2 )r$ rF r% r% r& r: � 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-asynczMUnable to start a new installation because a background task is still workingr )r) r* �success)r) �pid)
�sysrE �appendr r r2 Z set_modulesr rQ �
worker_pid)r$ r+ �cmd�configrT r% r% r& r4 � s
z,CloudlinuxWizard.run_background_installationc C s t � dkr| jdd� dS )z;
Check that wizard supports current system
Nz�Could not identify the CloudLinux version. Restart your system. If you have the same problem again - contact CloudLinux support.)r) )r r2 )r$ r% r% r& r0 � s
z!CloudlinuxWizard._validate_systemc
C s� t � rbdddg}yt|�}| j�d|� W q� tk
r^ } z| jjd||d� W dd}~X Y q�X n\dd d
d g}yt|�}| j�d |� W n4 tk
r� } z| jjd
||d� W dd}~X Y nX dS )z�
Prepare the enviroment before performing the installation.
This function updates the package lists if run on Ubuntu, expires Yum
cache when running on RHEL-based systems.
zapt-getz-q�updatezapt-get update output:
%sz!Error during apt-get update: '%s')�exc_infoNZyumz-qyZcleanz expire-cachez!yum clean expire-cache output:
%sz)Error during yum clean expire-cache: '%s')r
r r# rL r
r? )r$ rU �outrF r% r% r& r3 � s
& z*CloudlinuxWizard._prepare_for_installationc 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_SKIPPEDr[ rS r �CRASHEDr2 r )r$ r_ rV �stater\ r, r] Z
module_statusr% r% r& r7 � s&
z&CloudlinuxWizard._get_modules_statusesc C sL t � }|r| j|d� n0| 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)�.0rZ �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& r6 � 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 the error if it's not possiblez0Trying to cancel the installation of module '%s')rZ Z new_statez.Module '%s' installation successfully canceledzGUnable to cancel module '%s' installation, because it is in status '%s'N)
r# rL r ra �CANCELLABLE_MODULE_STATUSESZset_module_statusr � CANCELLEDrO r )r$ r9 rV r, r% r% r& r8 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--asynczCollecting statistics...z*Statistics collection command output: '%s'z(Error during statistics collection: '%s'N) rH �environ�getrS r# rL r r
r? )r$ rU rY rF r% r% r& r= s
z*CloudlinuxWizard.run_collecting_statisticsc s6 t � �}t|j�� �}W dQ R X t� fdd�|D ��S )zy
Check that all modules were either:
-- installed
-- canceled
-- or auto-skipped
Nc 3 s | ]}|� j kV qd S )N)�DONE_MODULES_STATUSES)rf r, )r$ r% r& � <genexpr>9 s z<CloudlinuxWizard.is_all_modules_installed.<locals>.<genexpr>)r rj �statuses�values�all)r$ rV rr 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-lockz2Starting cagefs force-update in the background: %sZCL_TEST_SYSTEM)
rH rI rJ r# rL r �boolrn ro 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)rp )rf r, )r$ r% r& rq \ s z5CloudlinuxWizard._get_wizard_state.<locals>.<genexpr>c s s | ]}|t jt jfkV qd S )N)r rb rm )rf r, r% r% r& rq a s c s s | ]}|t jfkV qd S )N)r Z
INSTALLING)rf r, r% r% r& rq c s )rT r ZIDLE�psutilZProcessZ
NoSuchProcessrj rr rs rt ZDONE�anyrb rd Z IN_PROGRESS)r$ rV rr r% )r$ r& r` K 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 for the response,
usually we expect 'context' here
)r) Z timestamp� T)�indentZ sort_keysN)�timerW �print�json�dumpsrR �exit)r) r* ZextrarA r% r% r& r2 i s
z'CloudlinuxWizard._print_result_and_exit)N)rP r ) �__name__�
__module__� __qualname__�__doc__r ZPENDINGrb rm rl Z INSTALLEDrc rp r' rG � staticmethodr1 r: r4 r0 r3 r[ r7 r6 r8 r= r< r> r` r2 r% r% r% r& r 2 s0 H
r )1Z
__future__r r r| rH rR rz rC �typingr r r r rv Zclcommonr Zclcommon.utilsr r
r r Zclcommon.lib.cleditionr
Zclwizard.configr rV r r r_ r r r Z constantsr r r r r �
exceptionsr r �parserr Zutilsr r r �objectr r% r% r% r& �<module> s*