|
Server : Apache System : Linux cvar2.toservers.com 3.10.0-962.3.2.lve1.5.73.el7.x86_64 #1 SMP Wed Aug 24 21:31:23 UTC 2022 x86_64 User : njnconst ( 1116) PHP Version : 8.4.18 Disable Function : NONE Directory : /opt/alt/python37/lib/python3.7/site-packages/clwizard/__pycache__/ |
Upload File : |
B
�cl8 � @ 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% �@/opt/alt/python37/lib/python3.7/site-packages/clwizard/wizard.py�__init__C s zCloudlinuxWizard.__init__c
C s� t |�| _�y
| jjdkrj| �� | �� | �� rH| jjsH| 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�_prepare_for_installation�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& �runH sR
$ zCloudlinuxWizard.runc C s t j�t�S )N)�os�path�isfiler r% r% r% r& r2 � 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* )
rI �mknodr r# �info�OSError�IOError�warningrA r4 )r$ �errr% 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)
�sysrF �appendr r r4 Z set_modulesr rS �
worker_pid)r$ r+ �cmd�configrV r% r% r& r5 � 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 r4 )r$ r% r% r& r0 � s
z!CloudlinuxWizard._validate_systemc
C s` t � r\dddg}yt|�}| j�d|� W n0 tk
rZ } z| j�d|� W dd}~X Y nX dS )z�
Prepare the enviroment before performing the installation.
In its current form, this function only updates the package lists
if run on Ubuntu, does nothing on other OS variants.
zapt-getZupgradez-qzapt-get upgrade output:
%sz"Error during apt-get upgrade: '%s'N)r
r r# rM r
r@ )r$ rW �outrG r% r% r& r1 � 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[ rU r �CRASHEDr4 r )r$ r_ rX �stater\ r, r] Z
module_statusr% r% r& r8 � 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)�.0rZ �clsr% r% r& �
<dictcomp>� s z8CloudlinuxWizard._get_initial_status.<locals>.<dictcomp>)r_ Zunsuppored_by_cpN)r r4 r" �items�list�setr )r$ Z
error_messager% r% r& r7 � 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# rM r ra �CANCELLABLE_MODULE_STATUSESZset_module_statusr � CANCELLEDrP r )r$ r: rX r, r% r% r& r9 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) rI �environ�getrU r# rM r r
r@ )r$ rW rY rG 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>1 s z<CloudlinuxWizard.is_all_modules_installed.<locals>.<genexpr>)r rj �statuses�values�all)r$ rX 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)
rI rJ rK r# rM r �boolrn ro Z communicate)r$ Z
cagefsctl_binrW Zcagefsctl_procZis_test_environmentr% r% r& r? 3 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 T 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 Y s c s s | ]}|t jfkV qd S )N)r Z
INSTALLING)rf r, r% r% r& rq [ s )rV r ZIDLE�psutilZProcessZ
NoSuchProcessrj rr rs rt ZDONE�anyrb rd Z IN_PROGRESS)r$ rX rr r% )r$ r& r` C s
z"CloudlinuxWizard._get_wizard_staterR 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)�time�update�print�json�dumpsrT �exit)r) r* ZextrarB r% r% r& r4 a s
z'CloudlinuxWizard._print_result_and_exit)N)rR r ) �__name__�
__module__� __qualname__�__doc__r ZPENDINGrb rm rl Z INSTALLEDrc rp r' rH � staticmethodr2 r; r5 r0 r1 r[ r8 r7 r9 r> r= r? r` r4 r% r% r% r& r 2 s0 I
r )1Z
__future__r r r} rI rT rz rD �typingr r r r rv Zclcommonr Zclcommon.utilsr r
r r Zclcommon.lib.cleditionr
Zclwizard.configr rX 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*