Edit File: wizard.cpython-37.pyc
B 5/9e�8 � @ 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 � 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-getz-q�updatezapt-get update output: %sz!Error during apt-get update: '%s'N)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_SKIPPEDrZ 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)�.0rY �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')rY Z new_statez.Module '%s' installation successfully canceledzGUnable to cancel module '%s' installation, because it is in status '%s'N) r# rL r r` �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 rX 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)re r, )r$ r% r& � <genexpr>1 s z<CloudlinuxWizard.is_all_modules_installed.<locals>.<genexpr>)r ri �statuses�values�all)r$ rV rq 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 �boolrm rn Z communicate)r$ Z cagefsctl_binrU 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)ro )re r, )r$ r% r& rp T s z5CloudlinuxWizard._get_wizard_state.<locals>.<genexpr>c s s | ]}|t jt jfkV qd S )N)r ra rl )re r, r% r% r& rp Y s c s s | ]}|t jfkV qd S )N)r Z INSTALLING)re r, r% r% r& rp [ s )rT r ZIDLE�psutilZProcessZ NoSuchProcessri rq rr rs ZDONE�anyra rc Z IN_PROGRESS)r$ rV rq r% )r$ r& r_ C 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 a s z'CloudlinuxWizard._print_result_and_exit)N)rP r ) �__name__� __module__� __qualname__�__doc__r ZPENDINGra rl rk Z INSTALLEDrb ro r' rG � staticmethodr1 r: r4 r0 r3 rZ r7 r6 r8 r= r< r> r_ r2 r% r% r% r& r 2 s0 H r )1Z __future__r r r{ rH rR ry rC �typingr r r r ru 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*