관리-도구
편집 파일: apps_manager.cpython-37.pyc
B �v6d�d � @ s& d dl mZ d dl mZ d dl mZ d dlZd dlZd dlZd dlmZm Z d dl m Z d dl m Z d dl Z d dlmZ d dlmZ d d lmZ d d lmZmZ d d lmZmZmZ d d lm Z d d l m Z m Z m Z m!Z! ddl"m#Z# d dlm$Z$m%Z% d dlm&Z& d dlm'Z' G dd� de'ee(��Z)dS )� )�absolute_import)�print_function)�divisionN)�ABCMeta�abstractmethod)�datetime)� iteritems)�unicode)�ClPwd)�get_file_lines�write_file_lines)� pretty_json�delete_using_realpath_keys� get_abs_rel)�ClSelectExcept)�BaseSelectorError�AbsentFileError�MissingVirtualenvError�MissingAppRootError� )�BaseSelectorManager)�file_readlines�file_writelines)�get_using_realpath_keys)�with_metaclassc @ sP e Zd ZdZdZdZdZdZdZdd� Z e dd� �Z dd � Z d d � Z d d � ZdHdd�ZedIdd��Zdd� Zeedd� ��ZdJdd�ZdKdd�Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Z d'd(� Z d)d*� Z d+d,� Z d-d.� Z d/d0� Z!d1d2� Z"ed3d4� �Z#dLd5d6�Z$dMd7d8�Z%d9d:� Z&d;d<� Z'd=d>� Z(dNd?d@�Z)dOdBdC�Z*dDdE� Z+dFdG� Z,dS )P�BaseApplicationsManagerz[ Base class that responsible for gathering and set information about applications. Nz/var/log/selectorctl.logc C s || _ t� | _d S )N)�_managerr �_pwd)�self�manager� r �S/opt/alt/python37/lib/python3.7/site-packages/clselect/baseclselect/apps_manager.py�__init__2 s z BaseApplicationsManager.__init__c C sH y*t �� �d�}t| jd||f gd� W n ttfk rB Y nX dS )zi Writes string to log file :param log_str: String to write :return: None z%Y-%m-%d %H:%M:%Sz%s: %s �aN)r �now�strftimer �_LOG_FILE_NAME�IOError�OSError)�clsZlog_strZ dt_stringr r r! �write_string_to_log6 s z+BaseApplicationsManager.write_string_to_logc C s" | � |�}|||< | �||� dS )a Add data to user's config (update info for an app with app_directory specified) :param str user: name of unix user :param str app_directory: Application root directory :param dict app_data: Application data :return: None N)�get_user_config_data�write_full_user_config_data)r �user� app_directory�app_dataZ current_dictr r r! �add_app_to_configC s z)BaseApplicationsManager.add_app_to_configc C s d}d}| � |||d�}|dkr$dS t|ddd���}dd� |�� �d �D �} || kr~|| kr~| �|�} | �|�} | | | d �= |dk r�| �|� | �d � x&|�� D ]\} } | �d �| | �� q�W | �d � | �|� W dQ R X t|ddd��}|�d � | �� W dQ R X dS )a? Add enviroment variables to .htaccess file for LVEMAN-1623 :param str user_name: Name of unix user :param str app_directory: Application root directory :param dict env_vars: Dict of enviroment variables :param str doc_root: doc root of application :return: None zB# DO NOT REMOVE OR MODIFY. CLOUDLINUX ENV VARS CONFIGURATION BEGINz@# DO NOT REMOVE OR MODIFY. CLOUDLINUX ENV VARS CONFIGURATION ENDN�r�surrogateescape)�errorsc S s g | ] }|r|�qS r r )�.0�liner r r! � <listcomp>_ s zEBaseApplicationsManager.add_env_vars_for_htaccess.<locals>.<listcomp>� r z<IfModule Litespeed>zSetEnv {0} {1}z </IfModule>�w) �get_htaccess_by_appdir�open�read�split�index�append�items�format�write�join)r � user_namer. �env_vars�doc_rootZenv_section_beginZenv_section_end� htaccess_file�htaccess�lines�start�endZenv_var�valuer r r! �add_env_vars_for_htaccessP s( z1BaseApplicationsManager.add_env_vars_for_htaccessc C sD | � |�}y t|||� | �||� W n tk r: dS X dS dS )z� Removes application from config :param user: User name :param app_dir: Application directory :return: True if app is exists in config, False - if app is not exists in config FTN)r+ r r, �KeyError)r r- �app_dir� config_datar r r! �remove_app_from_confign s z.BaseApplicationsManager.remove_app_from_configFc C s� | � |�}|sdS xpt|�D ]d\}}|r>t�d| |d �}nt�d| |d �}|dk r td�|d dt|� � |��|d< q W | �||� dS )a Replace domain in config files when it is renamed. :param username: domain owner :param domain: previous name :param new_domain: name after rename :param include_subdomains: whether we should also rename subdomains :return: Nz (\.|^)%s$�domainz^%s$z{}{})r+ r �re�searchr r@ �lenr, ) r �usernamerQ Z new_domainZinclude_subdomains� full_config�app�config�matchr r r! �replace_domain_in_configs� s z1BaseApplicationsManager.replace_domain_in_configsc s t | |�\� }� fdd�|D �S )z6 Return list of detected config files c s& g | ] }t j�t j�� |��r|�qS r )�os�path�existsrB )r4 rX )�abs_pathr r! r6 � s z>BaseApplicationsManager._find_config_files.<locals>.<listcomp>)r )rC r. �patternsZrel_pathr )r^ r! �_find_config_files� s z*BaseApplicationsManager._find_config_filesc C s� | � ||�}| j�|�}| �||||�}d�tj�d�tj� ||��d�|d �g}|� | � | � |||�|�� |� dd�}|r�|�d| � |�tjd � |� t|d d �� t�|�}t||d d d � dS ) z� Creates .htaccess file for application based on application config data :param str user_name: User's name :param str app_root: Application root :param doc_root: Document root for the domain :return: None z{} zPassengerAppRoot "{}" zPassengerBaseURI "/{}" �app_uri�passenger_log_fileNzPassengerAppLogFile "%s" r7 r2 )r3 r8 )�get_app_configr � get_homedirr9 r@ � clpassenger�HTACCESS_BEGINr[ r\ rB �extend�(get_interpreter_specific_passenger_lines�get_binary_path�getr> � HTACCESS_ENDr �rm_double_empty_linesr ) r rC �app_rootrE � app_config�user_dirrF � new_linesrb r r r! �update_htaccess_file� s z,BaseApplicationsManager.update_htaccess_filec C s t �dS )zu Return list of lines that needs to be added to htaccess and are specific to the interpreter N)�NotImplementedError)� binary_pathrn r r r! rh � s z@BaseApplicationsManager.get_interpreter_specific_passenger_linesc C s2 | � ||�}|dkr| j}tj�|| j||d|�S )a� Return a path to the environment's interpreter binary Get interpreter path for application :param user: owner of the application :param app_root: app path relative to user home (app-root) :param user_dir: User's home directory :param binary_name: name of binary in virtual environemnt (python, npm, node) :return: path to interpreter binary in virtual environment N�bin)� get_interpreter_version_for_app� BINARY_NAMEr[ r\ rB �VENV_DIR)r r- rm ro � binary_name�versionr r r! ri � s z'BaseApplicationsManager.get_binary_pathc C s" |dkr| j �� S || j �|�iS )z� Retrives info about user(s). :param str | None username: Username to retrive information. :return: Dictionary with user info. Example: {'user1': pw_struct} N)r � get_user_dict�get_pw_by_name)r rU r r r! �get_users_dict� s z&BaseApplicationsManager.get_users_dictc C sj i }| � |�}tj�|�rft|�}d�|�}yt�|�}W n* tt fk rd t � d� | j ���Y nX |S )z� Get all data from user's config :param user: name of unix user :return: json data from user's config as dictionary � zUser config "{}" is broken)�_get_path_to_user_configr[ r\ �isfiler rB �json�loads� ValueError� TypeErrorr � WrongDatar@ � _USER_CONFIG)r r- Z_user_config_data� user_config�dataZ joined_datar r r! r+ � s z,BaseApplicationsManager.get_user_config_datac C s0 | � |�}y t|||�S tk r* dS X dS )z� Retrieves full application config :param user: User name :param app_dir: Application directory :return: Application data as dictionary If None - No application data found in config N)r+ r rM )r r- rN �user_config_datar r r! rc � s z&BaseApplicationsManager.get_app_configc C s | � ||�}|d S )z� Retrieves domain for provided user's application :param username: user name :param app_directory: application root directory :return str: application domain rQ )rc )r rU r. rn r r r! �get_app_domain s z&BaseApplicationsManager.get_app_domainc C s | � ||�}|d S )z� Retrieves uri for provided user's application :param username: user name :param app_directory: application root directory :return str: application uri ra )rc )r rU r. r/ r r r! � get_app_uri s z#BaseApplicationsManager.get_app_uric C s | � ||�}|d S )z� Retrieves name of startup file for provided user's application :param username: user name :param app_directory: application root directory :return str: name of startup file of application � startup_file)rc )r rU r. r/ r r r! �get_app_startup_file s z,BaseApplicationsManager.get_app_startup_filec C s | � ||�}|d S )z� Retrieves status for provided user's app_directory :param username: user name :param app_directory: application root directory :return str: status of application � app_status)rc )r rU r. r/ r r r! �get_app_status! s z&BaseApplicationsManager.get_app_statusc C s2 | � ||�}|dkr$t�d�|���|d| j S )z� Retrieves interpreter version for provided user and application :param username: user name :param app_directory: application root directory :return str: major interpreter version Nz)Application config is missed for user: {}z %s_version)rc r �ConfigMissingErrorr@ � INTERPRETER)r rU r. r/ r r r! ru + s z7BaseApplicationsManager.get_interpreter_version_for_appc C s | j �|�}tj�|| j�}|S )z� Get full path to user config ~/${_USER_CONFIG} :param username: name of unix user :return: full path to config )r rd r[ r\ rB r� )r rU � user_homeZapplication_configr r r! r~ 8 s z0BaseApplicationsManager._get_path_to_user_configc C s | j j�|�}||kr|S dS )z� Retrieves full version for supplied major version. :return: str - Full version or None if full version not found N)r �pkg�get_full_version)r � major_version� full_versionr r r! �_get_full_version_for_shortC s z3BaseApplicationsManager._get_full_version_for_shortc C s< |j |kr |||j d |< n|j||id�}|||j < dS )a@ Add single application data to user_data_dest_dict :param users_data_dest_dict: Destination dictionary with application data :param user_pw_entry: User's passwd entry :param app_root_dir: Application root directory :param app_data: Application data :return: None � applications)�homedirr� N)�pw_name�pw_dir)r Zusers_data_dest_dict� user_pw_entry� app_root_dirr/ � user_datar r r! �_add_single_user_appM s z,BaseApplicationsManager._add_single_user_appc C s. | � |j|�}t|d | �}t|�|d< dS )z<Add automatically-detected config files to user-defined list� config_filesN)r` r� �set�list)r r/ r� r� Zconfig_files_detectedZ merged_filesr r r! �)_add_detected_config_files_to_applicationa s zABaseApplicationsManager._add_detected_config_files_to_applicationc C s� x�t |�D ]�\}}| �|d| j �}|dks ||kr8q d|| krPi || d< d| j|j||jdd�i|d< |d| j = || d }| �|||� | �||||� q W dS )aV Add all user's apps information to user_data_dest_dict :param user_pw_entry: User's passwd entry :param user_data_dest_dict: Destination dictionary with application data :param user_app_data: User's application data ([node|python|ruby]-selector.json file content as dictionary). :return: None z %s_versionN�usersZ activate_path�activate)rx � virtualenv)r r� r� ri r� r� r� r� )r r� Zuser_data_dest_dictZ user_app_datar� r/ Zfull_interpreter_version� users_dictr r r! �_add_all_user_appsg s z*BaseApplicationsManager._add_all_user_appsc C s2 x,t | �D ] \}}d|kr"d|d< || |< q W | S )z� Append absent 'passenger_log_file' key with None value to each application :param user_config_dict: Sourse dictionary to modify :return: Modified dict with 'passenger_log_file' keys rb N)r )Zuser_config_dictrm r/ r r r! � _add_absent_passenger_log_file� s z6BaseApplicationsManager._add_absent_passenger_log_filec C s� |dkrt �� }t j�|| j�}t j�|�s4t|��yB|dkrZdt_tj |||ddd�}nt |�}| � t � d�|���S ttttfk r� } z td|t|�d�d ���W dd}~X Y nX dS ) a] Read [python|ruby|node]-selector.json file from user's directory. :param euid: current effective uid :param user_homedir: user's home directory :param uid: uid for drop rights :param gid: gid for drop rights :return: Cortege contents_dict. None - there is no [python|ruby|node]-selector.json file in user's directory :raises BaseSelectorError if error. Exception contents: {'message': "File %(file)s read/parse error: %(error)s", 'context': {'file': node_json_path, 'error': 'some message'}} Nr TF)� exit_on_error� write_logr} z)File %(file)s read/parse error: %(error)s)�file�error)�message�context)r[ �geteuidr\ rB r� r] r �secureio� SILENT_FLAG�read_file_securer r� r� r� r' r( r� r� r �str)r Z user_homedir�uid�gid�euid�json_config_path� file_lines�er r r! � read_user_selector_config_json� s z6BaseApplicationsManager.read_user_selector_config_jsonc C sB y(|dkr| � ||�}tj�||d d�S tk r< dS X dS )a Retrieve .htaccess for user and app :param user: Username :param app_dir: App dir :param doc_root: Document root for selected domain :param app_config: Optional app configuration :return: .htaccess full path Nra z .htaccess)rc r[ r\ rB rM )r r- rN rE rn r r r! r9 � s z.BaseApplicationsManager.get_htaccess_by_appdirc C sH | � |�}yt|||�}||d< W n tk r6 dS X | �||� dS )z� Retrieves application status :param user: User name :param app_dir: Application directory :param new_status: New appication status :type new_status: str :return: None r� N)r+ r rM r, )r r- rN � new_statusr� rn r r r! �set_app_status� s z&BaseApplicationsManager.set_app_statusc C s� | � |�}t�� dks"t�� dkr,t�d��t|�}ytj||ddd� W n: t t t fk r� } zt� |d� |���W dd}~X Y nX dS )z� Write data to user's config :param user: name of unix user :param config_data: data in json format :return: None r z$root should not write to user configi� �_tmp)�content� dest_path�perm�suffixz%Could not write json user config ({})N)r~ r[ r� �getegidr �SelectorExceptionr r� �write_file_via_tempfiler' r( r� �UnableToSaveDatar@ )r r- rO r� Z dumped_datar� r r r! r, � s z3BaseApplicationsManager.write_full_user_config_datac C s� | � � }x�t|�D ]�\}}yJ| �|j|j|j�}|dk rbx(|�� D ] }|�d| j �|krBdS qBW W q t k rx Y q t k r� } z| � |j |j � W dd}~X Y qX qW dS )z� Returns True if specified NodeJS version is in use by any app of any user. It will stop search on the first match. :param version: NodeJS interpreter major version :return True | False Nz %s_versionTF)r| r r� r� �pw_uid�pw_gid�valuesrj r� r r r* r� r� )r ry � user_inforC r� �user_apps_datarW r� r r r! �is_version_in_use� s (z)BaseApplicationsManager.is_version_in_usec C s� y| � |�}W n" tjjk r0 t�|��Y nX | �� }|d }x�t|�D ]�\}}y&| �|j |j |j �}| � |||� W qL t k r� Y qL tk r� } z<t�� dkr�| �|j|j � d| j| jf |d< n|�W dd}~X Y qLX qLW |S )a� Retrieves info about all installed NodeJS interpreters and user(s) applictions :param user: User name for read applictions. If None and current euid == 0 - all users will be processed. If current euid != 0 (called under some user), this argument will be ignored and only user with uid == euid will be processed :return: Dictionary with user(s) applications info :raises ClSelectExcept.NoSuchUser or BaseSelectorError �available_versionsr z`Some user's %s can't be read. Some user(s) data absent in output. Please see file %s for details�warningN)r| r� �clpwd�NoSuchUserExceptionr � NoSuchUser�_get_interpreter_infor r� r� r� r� r� r r r[ r� r* r� r� r� r& ) r r- r� �users_apps_infor� rC r� r� r� r r r! �get_applications_users_info s( z3BaseApplicationsManager.get_applications_users_infoTc C sf | j �|�j}tj�|| j|�}|r8tj�|�s8t|��tj�||�}|r^tj�|�s^t |��||fS )a Calculate, check exists and return application folders This method does not check that application exists in config. :raises: NoSuchUserException, MissingVirtualenvError, MissingAppRootError :return: tuple(app_root, app_venv) with absolute paths ) r r{ r� r[ r\ rB rw r] r r )r rU rm Zchk_envZ chk_app_rootr� �app_venvr r r! �get_app_folders3 s z'BaseApplicationsManager.get_app_foldersc C s | j �� }|S )z)Get initial information about interpreter)r � get_summary)r r� r r r! r� E s z-BaseApplicationsManager._get_interpreter_infoc C s | j j�|�S )zi Just a public proxy to internal method that blocks any actions with interpreter )r r� �acquire_interpreter_lock)r � it_versionr r r! r� J s z0BaseApplicationsManager.acquire_interpreter_lock)F)N)N)N)N)N)N)TT)-�__name__� __module__� __qualname__�__doc__r� r& r� rw rv r" � classmethodr* r0 rL rP rZ � staticmethodr` rq r rh ri r| r+ rc r� r� r� r� ru r~ r� r� r� r� r� r� r9 r� r, r� r� r� r� r� r r r r! r & sR " * r )*� __future__r r r r� r[ rR �abcr r r Zclselect.clpassengerre r� � future.utilsr � past.builtinsr �clcommonr �clcommon.utilsr r Zclselect.utilsr r r �clselectr �clselect.baseclselectr r r r �selector_managerr r r r r �objectr r r r r! �<module> s*