Edit File: clcontrollib.cpython-37.pyc
B 5/9e�� � @ sX d dl mZ d dl mZ d dlmZ d dlZd dlZd dlZd dlZd dl Z d dl m Z m Z m Z d dlmZmZ d dlmZ d dlmZ d d lmZmZmZmZ d dlZd Z G d d � d e �Z d d� Z G dd� de �Z!G dd� de!�Z"G dd� de!�Z#G dd� de �Z$G dd� de!�Z%G dd� de!�Z&G dd � d e!�Z'G d d � d e!�Z(G d d � d e!�Z)dS )!� )�print_function)�absolute_import)�rangeN)�Tuple�Optional�List)�ClPwd� mysql_lib)�ProcLve)�grep)�cpanel� directadmin� ispmanager�plesk�0 c @ s e Zd ZdZdS )�ConfigDataErrorz� Should be raised when something went wrong during user's config data loading/parsing. Supposed to be used when we want to skip some user with broken configuration files N)�__name__� __module__� __qualname__�__doc__� r r �clcontrollib.pyr s r c C sr t �� t jdkrt� S t jdkr(t� S t jdkr8t� S t jdkrHt� S t jdkrXt� S t jdkrht� S t � S d S )N�PleskZcPanel� InterWorx� ISPManager� DirectAdmin� HostingNG) �detect� getCPNameZCP_NAMEr �Cpanelr r r r �Unknownr r r r �detect_panelclass' s r! c @ s� e Zd Zdd� Zd+dd�Zdd� Zdd � Zed d � �Zd,dd�Z edd� �Z ed-dd��Z d.dd�Z d/dd�Z d0dd�Zd1dd �Zd2d d �Zd3d d �Zd4d!d"�Zd5d#d$�Zd6d%d&�Zd7d'd(�Zd8d)d*�Zd S )9� GeneralPanelc C sN t � | _| jj| _t�� | _| j�� | j�� krDt d� | j| j���d | _ d S )Nz+Class "{}" not should be used on panel "{}") r � users_data� __class__r � class_namer r Zcurrent_panel_name�lower� UserWarning�format� cpapi_plugin)�selfr r r �__init__; s zGeneralPanel.__init__Fc C s | j �� S )z� List all admins names in given control panel :param bool debug: Do produce debug output or don't :return: list of strings )r) Zadmins)r* �debugr r r � list_adminsG s zGeneralPanel.list_adminsc C s | j �|�S )z| Return True if username is in admin names :param str username: user to check :return: bool )r) �is_admin)r* �usernamer r r r. O s zGeneralPanel.is_adminc C s |r t j�d|| jjf � d S )Nz %s is not implemented for %s )�sys�stderr�writer$ r )r* r, � method_namer r r �_warningW s zGeneralPanel._warningc C s t � t jd�| �� d S )NzCL_CONTROLLIB: {})�syslogZ LOG_WARNINGr( )�msgr r r � _slog_warning\ s zGeneralPanel._slog_warningNTc C s |dk r|dk rt d��|dkr0|dkr0t d��|dk r�y | j�|�S tjk r� } z$|rdt|� |rx| �d�|�� dS d}~X Y nX y| j�t |��d S tjk r� } z$|r�t|� |r�| �d�|�� dS d}~X Y n t k � r� td� dS X dS )aH Check that provided user is really exists in system and return it as: - uid if username was provided - username if uid was provided - None if user doesn't exists Produce syslog/debug output depending on flags if user isn't exists. This helper function is mostly needed because it's possible that user is already deleted in system but his entity is still present in control panel files (like DA_USERS_DIR) so we should do additional checks to skip such users. :return: (uid | username) or None NzIThis function accepts either uid or username, but not both simultaneouslyz0Incorrect call: uid or username is not specifiedz4User with username="{}" is not present in the systemr z/User with uid="{}" is not present in the systemzIncorrect user id) � ValueErrorr# �get_uidr �NoSuchUserException�printr7 r( � get_names�int)r* �uidr/ r, Zsyslog_�er r r �_check_and_get_user` s8 z GeneralPanel._check_and_get_userc C s" t | d��}|�� }W dQ R X |S )a Safely read file content and close it before return the content string. No exceptions will have caught! :param string file_name: Absolute or relative path to file :rtype: string :return: Content of given file path �rN)�open� readlines)� file_name�f�contentr r r �get_file_lines� s zGeneralPanel.get_file_lines�wc C s$ t | |��}|�|� W dQ R X dS )a= Safely write to file and close it before the function end. No exceptions will have caught! :param string file_name: Absolute or relative path to file :param string content: What to write there :param string open_method: Either "w" or "a" or "b" :return: None N)rB � writelines)rD rF Z open_methodrE r r r �write_file_lines� s z GeneralPanel.write_file_linesc C s | � |d� dS )a Implements /usr/bin/getcontrolpaneluserspackages and /usr/bin/getcontrolpaneluserspackages --list-all commands Result is stdout output: uid package name :param bool debug: Do produce debug output or don't :return: None zList allN)r4 )r* r, r r r �list_all� s zGeneralPanel.list_allc C s | � |d� dS )a Implements /usr/bin/getcontrolpaneluserspackages --package=package_name Result is stdout output: uid :param string package: Package name :param bool debug: Do produce debug output or don't :return: None zList users in given packageN)r4 )r* �packager, r r r �list_users_in_package� s z"GeneralPanel.list_users_in_packagec C s | � |d� dS )aX Implements /usr/bin/getcontrolpaneluserspackages --package=package_name --reseller=reseller_name Result is stdout output: uid :param string package: Package name :param string reseller: Reseller name :param bool debug: Do produce debug output or don't :return: None z&List users in given reseller's packageN)r4 )r* rL �resellerr, r r r � list_users_in_reseller_package� s z+GeneralPanel.list_users_in_reseller_packagec C s | � |d� dS )z� Implements /usr/bin/getcontrolpaneluserspackages --userid=id Result is stdout output: uid package name :param int uid: User ID :param bool debug: Do produce debug output or don't :return: None zList given user's packageN)r4 )r* r> r, r r r �list_user_package� s z GeneralPanel.list_user_packagec C s | � |d� dS )z� Implements /usr/bin/getcontrolpaneluserspackages --list-packages Result is stdout output: package name :param bool debug: Do produce debug output or don't :return: None zList all packagesN)r4 )r* r, r r r � list_packages� s zGeneralPanel.list_packagesc C s | � |d� dS )z� Implements /usr/bin/getcontrolpaneluserspackages --list-resellers-packages Result is stdout output: reseller_name package name :param bool debug: Do produce debug output or don't :return: None z!List packages belong to resellersN)r4 )r* r, r r r �list_resellers_packages� s z$GeneralPanel.list_resellers_packagesc C s | � |d� dS )a' Implements /usr/bin/getcontrolpaneluserspackages --list-reseller-packages=reseller Result is stdout output: package name :param string reseller: Reseller name :param bool debug: Do produce debug output or don't :return: None z&List packages belong to given resellerN)r4 )r* rN r, r r r �list_reseller_packages� s z#GeneralPanel.list_reseller_packagesc C s | � |d� dS )z� Implements /usr/bin/getcontrolpaneluserspackages --list-users Result is stdout output: uid,package name,reseller_name :param bool debug: Do produce debug output or don't :return: None z List usersN)r4 )r* r, r r r � list_users s zGeneralPanel.list_usersc C s | � |d� dS )a( Generate triple with uid, it's package, it's reseller :param bool debug: Do produce debug output or don't :param string reseller_name: filter by this reseller :rtype: generator of (int, string, string) :return: (uid, package name, reseller name) zList reseler's usersN)r4 )r* r, � reseller_namer r r �_get_user_packages s z GeneralPanel._get_user_packagesc C sT | j dkrP|dkr"|r td� dS x,| j||d�D ]\}}}td�||�� q2W dS )z� Print uid and package for defined reseller :param bool debug: Do produce debug output or don't :param reseller: username of reseller :return: None r" NzReseller is not defined)r, rU z{},{})r% r; rV r( )r* r, rN r> rL �_r r r �list_reseller_users s z GeneralPanel.list_reseller_usersc C s | � |d� dS )z� Determine user's reseller :param string uid: uid of user :param bool debug: Do produce debug output or don't :rtype: string :return: reseller name zGet user's resellerN)r4 )r* r> r, r r r � get_reseller) s zGeneralPanel.get_reseller)F)NNFT)rH )F)F)F)F)F)F)F)F)FN)FN)F)r r r r+ r- r. r4 � staticmethodr7 r@ rG rJ rK rM rO rP rQ rR rS rT rV rX rY r r r r r" 9 s* 2 r" c s� e Zd ZdZdZdZdZdZ� fdd�Zdd � Z d d � Z d d � Z dd� Z d)dd�Z d*dd�Zdd� Zd+dd�Zd,dd�Zd-dd �Zd.d d �Zd/d d �Zd0d!d"�Zd1d#d$�Zd2d%d&�Zd3d'd(�Z� ZS )4r z/var/cpanel/usersz/var/cpanel/packagesz/var/cpanel/resellersz/etc/trueuserownersNc s( t t| ��� | j�� | _t�� | _d S )N) �superr r+ r# Zget_sys_min_uid�min_uidr � PanelPluginr) )r* )r$ r r r+ < s zCpanel.__init__c C s� d}y"t |d��}|�� }W dQ R X W np tk r� } zRd|� d|� d�}|�| j�rd|d|� �7 }t|tjd� | �|� t� t � W dd}~X Y nX |S ) a Safely read file content and close it before return the content string. No exceptions will have caught! :param string file_name: Absolute or relative path to file :rtype: string :return: Content of given file path z]http://kb.cloudlinux.com/2017/08/how-to-fix-the-issue-with-non-utf-symbols-in-packages-names/rA NzUnable to read file 'z ', error is: z-. It looks like you use not an UTF-8 charset.z Please read the instruction: )�file) rB rC �UnicodeDecodeError� startswith�USERDIRr; r0 r1 r7 �exit�BAD_CODING_ERROR_CODE)r* rD Z ENCODING_LINKrE rF r? �messager r r rG A s zCpanel.get_file_linesc C sl i }t �d�}tj�| j�rh| �| j�}x>|D ]6}|�� }t �||�}|dkrPq.|� d�||� d�< q.W |S )z~ Get dict, where keys - user's names, values - reseller's names :return: dict {user1: reseller1, ...} z!^(?P<name>\w+):\s*(?P<owner>\w+)$N�owner�name) �re�compile�os�path�isfile�RESELLERS_AND_USERSrG �strip�match�group)r* �result�pattern� file_content�line�strippedrn r r r �_get_users_resellers_dictY s z Cpanel._get_users_resellers_dictc C s0 g }t j�| j�r,| �| j�}dd� |D �}|S )zw Get resellers list from RESELLERS_PATH :rtype: list :return: list of resellers names c S s g | ]}|� � �d d�d �qS )�:� r )rm �split)�.0�lr r r � <listcomp>~ s z)Cpanel._get_resellers.<locals>.<listcomp>)ri rj rk �RESELLERS_PATHrG )r* � resellersrr r r r �_get_resellerst s zCpanel._get_resellersc C sN d}t j�|�rJ| �|�}ttdddd|d��}|rJ|d �� �d�d }|S )z� Get package name from user's config file :param string path: Path to file contains info about cPanel user :rtype: string :return: Package name � z PLAN\=.*$F)� fixed_string�match_any_position�multiple_search�data_from_filer �=rw )ri rj rk rG �listr rm rx )r* rj rL rr rp r r r � get_package� s zCpanel.get_packageFc C s� d}| j ||d�}|rltj�| j|�}tj�|�rl| �|�}ttdddd|d��}|rl|d � � � d�d }|r�|| � � kr�|S dS ) z� Determine user's reseller based on `owner` field from user's config :param string uid: uid of user :param bool debug: Do produce debug output or don't :rtype: string :return: reseller name r )r> r, z OWNER\=.*$F)r� r� r� r� r r� rw ) r@ ri rj �joinra rk rG r� r rm rx r~ )r* r> r, rN r/ rj rr rp r r r � _get_reseller� s zCpanel._get_resellerc C s t | j||d�� dS )z� Determine user's reseller based on `owner` field from user's config :param string uid: uid of user :param bool debug: Do produce debug output or don't :rtype: string :return: reseller name )r, N)r; r� )r* r> r, r r r rY � s zCpanel.get_resellerc c sb t j�| j�r^xNt �| j�D ]>}t j�t j�| j|��rD|dfV q t j�| j|�dfV q W dS )z� Generate list of packages in current panel :rtype: generator of (string, bool) :return: (name of package, True) for normal files and (/path/to/wrang/file, False) otherwise TFN)ri rj �isdir� PACKAGEDIR�listdirrk r� )r* �filenamer r r � _get_packages� s zCpanel._get_packagesc C s^ | � � }xP| �� D ]D\}}|rFd|kr<|�dd�d |kr<qt|� q|rtd| � qW d S )NrW rw r zWarning: %s is not a file.)r~ r� rx r; )r* r, r} r� �is_filer r r rQ � s zCpanel.list_packagesc c s� | � � }|sdS tj�| j�r�x�t�| j�D ]�}y | j�|�}|| jk rJw,W n0 t j k r| } z|rlt |� W dd}~X Y q,X tj� | j|�}||kr�|| }n | � ||�}| �|�r�dn|}|dk r�||kr�q,|| �|�|fV q,W dS )ao Generate triple with uid, it's package and it's reseller for every user in USERDIR which uid is greater than MIN_UID :param bool debug: Do produce debug output or don't :param string reseller_name: filter by this reseller :rtype: generator of (int, string, string) :return: (uid, package name, reseller name) Nr )ru ri rj r� ra r� r# r9 r\ r r: r; r� r� r. r� ) r* r, rU Zusers_resellers_dictr� r> r? rj rN r r r rV � s( zCpanel._get_user_packagesc C sN xH| j ||d�D ]6\}}}|dkr|d kr6t||� q||krt|� qW d S )N)rU r )rV r; )r* rL r, rN r> � self_packageZ self_resellerr r r rK � s zCpanel.list_allc C s� y| j �t|��}W n< tjk r6 |r2td� Y n^ tk rP td� Y nDX x@|D ]8}| jd | }tj � |�rX| � |�}|dkrXt|� qXW d S )Nz,getcontrolpaneluserspackages: User not foundzIncorrect user id�/r ) r# r< r= r r: r; r8 ra ri rj rk r� )r* r> r, �users�userrj rL r r r rP s zCpanel.list_user_packagec C s� | � � }xv| �� D ]j\}}|rhd|krh|�dd�d }||krBqq||d k rV||krVqq|td||f � q|s|rtd| � qW d S )NrW rw r z%s %szWarning: %s is not a file.)r~ r� rx r; )r* r, rN r} r� r� Zres_namer r r rR s z Cpanel.list_resellers_packagesc C s6 | � � }||kr$|r td| � d S | j||d� d S )NzError: %s is not reseller)r, rN )r~ r; rR )r* rN r, r} r r r rS s z Cpanel.list_reseller_packagesc sd | � � �d|kr@|�d�� t� �fdd�tdt� ��D ��r@d S |d k rT| �|� n |r`td� d S )NrW c 3 s$ | ] }d � � d|� ��kV qdS )rW N)r� )ry �i)�partsr} r r � <genexpr>+ s z/Cpanel.list_users_in_package.<locals>.<genexpr>rw z/getcontrolpaneluserspackages: Undefined package)r~ rx �anyr �lenrK r; )r* rL r, r )r� r} r rM ' s " z Cpanel.list_users_in_packagec C s2 | � � }||kr"| j|||d� n |r.td� d S )N)r, rN z8getcontrolpaneluserspackages: Undefined reseller package)r~ rK r; )r* rL rN r, r} r r r rO 2 s z%Cpanel.list_users_in_reseller_packagec C s0 x*| � |�D ] \}}}td|||f � q W d S )Nz%s,%s,%s)rV r; )r* r, r> r� rN r r r rT 9 s zCpanel.list_users)F)F)F)FN)NFN)F)FN)F)F)F)F)r r r ra r� r| rl r# r+ rG ru r~ r� r� rY r� rQ rV rK rP rR rS rM rO rT � __classcell__r r )r$ r r 4 s, $ r c s0 e Zd ZdZdZdZdZdZdZdZ dZ e � Z � fdd �Z d d � Zd d � Zdd� Zdd� Zd<eeee ed�dd�Zd=dd�Zd>dd�Zd?dd�Zd d � Zd d � Zd d!� Zd@d"d#�Z dAd$d%�Z d&d'� Z dBd(d)�Z dCd*d+�Z dDd,d-�Z!dEd.d/�Z"dFd0d1�Z#dGd2d3�Z$d4d5� Z%d6d7� Z&dHd8d9�Z'dId:d;�Z(� Z)S )Jr z"/usr/local/directadmin/data/users/z,/usr/local/directadmin/conf/directadmin.confz,/usr/local/directadmin/data/admin/admin.listz//usr/local/directadmin/data/admin/reseller.listz//usr/local/directadmin/data/admin/packages.listr Nc s2 t t| ��� t�� | _t�� | _| j� � | _ d S )N) r[ r r+ r Zdetect_DA_admin�DA_ADMINr r] r) Zget_encoding_name�ENCODING)r* )r$ r r r+ I s zDirectAdmin.__init__c C s� y(t |d| jd��}|�� }W d Q R X |S tk r~ } z8d|� d| j� d�}| �|d|� d� � t|�|�W d d }~X Y nX d S )NrA )�encodingzUnable to read file 'z '. It looks like you use not an z charset.z Error is: �.)rB r� rC r_ r7 � Exception)r* rD rE rF r? rd r r r rG P s zDirectAdmin.get_file_linesc C sN y | � |�S tk rH } z d}| �|�||�� t� �W dd}~X Y nX dS )zU Just a wrapper around _get_user_info_inner to send errors to syslog z;Skipping user '{}' with bad configuration files due to '{}'N)�_get_user_info_innerr� r7 r( r )r* r/ r? Ztplr r r �_get_user_info[ s zDirectAdmin._get_user_infoc C s* t j�| j|d�}y| �|�}W n d}Y nX |s>td��ttdddd|d��}ttdddd|d��}y|d �� � d �d }W n t k r� td ��Y nX y|d �� � d �d }W n t k r� td ��Y nX | � ||�}|d k�rd}| j � |�} | �| ��rd}|}| j � |�} | ||fS )zG Return uid, package name and owner of user (reseller) z user.confNz&Unable to read any data from user.confz usertype\=.*$F)r� r� r� r� z creator\=.*$r r� rw z#No "usertype" record in user configz"No "creator" record in user configrN r ZDEFAULT)ri rj r� � DA_USERS_DIRrG r8 r� r rm rx � IndexError�_get_user_packager# r9 �_is_reseller_limits_enabled) r* r/ � userprofile�userprofile_content�result_usertypeZresult_creator�usertyperN rL �lvp_idr> r r r r� k sH z DirectAdmin._get_user_info_innerc C s | j �|�S )N)�proc_lveZ exist_lvp)r* r� r r r r� � s z'DirectAdmin._is_reseller_limits_enabled)r/ r� �returnc C s� |sBt j�| j|d�}y| �|�}W n d}Y nX |sBtd��ttdddd|d��}ttdddd|d��}y|d �� � d �d }W n t k r� td ��Y nX |d kr�|r�y|d �� � d �d }W n t k r� td ��Y nX |S )z� Reads user's package just as in the `_get_user_info_inner` function, but without masking to DEFAULT package Be aware that it may return different package than the `_get_user_info_inner` function z user.confNz&Unable to read any data from user.confz package\=.*$F)r� r� r� r� zoriginal_package\=.*$r r� rw z"No "package" record in user configZcustomz+No "original_package" record in user config) ri rj r� r� rG r8 r� r rm rx r� )r* r/ r� r� Zresult_packageZresult_original_packagerL r r r r� � s< z DirectAdmin._get_user_packageFc c s� t | �� �}x|t�| j�D ]l}||kr(q| j||d�dkr<qy| �|�\}}}W n tk rf wY nX |dk rz||krzq|||fV qW dS )a' Generate triple with uid, it's package, it's reseller :param bool debug: Do produce debug output or don't :param string reseller_name: filter by this reseller :rtype: generator of (int, string, string) :return: (uid, package name, reseller name) )r/ r, N)�setr- ri r� r� r@ r� r )r* r, rU r- r/ r> rL rN r r r rV � s z DirectAdmin._get_user_packagesc C s� t j�| j|d�}t j�|�s"dS | �|�}xN|D ]F}y| �|�d��\}}}W n tk rf w2Y nX t d� ||�� q2W | � |�s�y| �|�\}}}W n tk r� dS X t d� ||�� dS )z� Print uid and package for users of a specified reseller :param debug: whether to produce debug output :param reseller: username of a reseller :return: None z users.listN� z{},{}) ri rj r� r� �existsrG r� �rstripr r; r( r. ) r* r, rN rj Zreseller_usersr/ r> rL rW r r r rX � s z DirectAdmin.list_reseller_usersc C sH | j ||d�}|rDy| �|�\}}}W n tk r: dS X t|� dS )z� Determine user's reseller based on package from user's config :param string uid: uid of user :param bool debug: Do produce debug output or don't :rtype: string :return: reseller name )r> r, N)r@ r� r r; )r* r> r, r/ rW rN r r r rY s zDirectAdmin.get_resellerc C s | � |�S )z� Check login_name is da admin. Backward compatibility wrapper :param login_name: Admin name :return: )r. )r* � login_namer r r � is_da_admin s zDirectAdmin.is_da_adminc C sl d}t j�| j|d�}t j�|�rh| �|�}ttdddd|d��}|rh|d �� � d�d }|dkrhd }|S ) z� Check login_name is da admin for using in hook return True - if login_name is admin False - otherwise Fz user.confz usertype\=.*$)r� r� r� r� r r� rw �adminT) ri rj r� r� rk rG r� r rm rx )r* r� rp rj rr r� r� r r r �is_da_admin_for_hook" s z DirectAdmin.is_da_admin_for_hookc C s� y ddl m}m}m}m} W n& tk rB td� t�d� Y nX | � � }xR|D ]J}y t |� W qR |||fk r� } ztdt |� � W dd}~X Y qRX qRW dS )z� Add all DA admins to sudoers. Now calls only from CageFS plugin installer :param s_cmd: String such as "Clsudo.add_cagefs_user(user)" :return: r )�Clsudo� NoSuchUser�UnableToReadFile�UnableToWriteFilezDCannot import Clsudo. Check if alt-python27-cllib package installed rw z%s N) Zclsudor� r� r� r� � ImportErrorr; r0 rb r- �eval�str) r* Zs_cmdr� r� r� r� � da_adminsr� r? r r r � add_all_da_admins_to_sudoers@ s z(DirectAdmin.add_all_da_admins_to_sudoersc C s | � |�S )z< Wrapper for compatibility with old code )r- )r* r, r r r �get_all_da_adminsT s z DirectAdmin.get_all_da_adminsc C s0 g }t j�| j�r,| �| j�}dd� |D �}|S )zr Get all DA admins list :rtype: list of strings :return: List of DA admins names c S s g | ]}|� d ��qS )r� )r� )ry rs r r r r{ d s z+DirectAdmin.list_admins.<locals>.<listcomp>)ri rj rk � DA_ADMIN_LISTrG )r* r, r� rr r r r r- Z s zDirectAdmin.list_adminsc c s* | � | j�}x|D ]}|�d�V qW d S )Nr� )rG �DA_ADMINS_PACKAGESr� )r* rr rs r r r r� g s zDirectAdmin._get_packagesc C s x| � � D ] }t|� q W d S )N)r� r; )r* r, rs r r r rQ l s zDirectAdmin.list_packagesc C s& x | � � D ]\}}}t||� q W d S )N)rV r; )r* r, r> rL rW r r r rK p s zDirectAdmin.list_allc C s2 x,| � � D ] \}}}tdt|�||f � q W d S )Nz%s,%s,%s)rV r; r� )r* r, r> rL rN r r r rT t s zDirectAdmin.list_usersc C sH | j ||d�}|rDy| �|�\}}}W n tk r: d S X t|� d S )N)r> r, )r@ r� r r; )r* r> r, r/ rW rL r r r rP x s z DirectAdmin.list_user_packagec C s0 x*| j dd�D ]\}}}||krt|� qW d S )Nr )rU )rV r; )r* rL r, r> �package_rN r r r rM � s z!DirectAdmin.list_users_in_packagec C s0 x*| j |d�D ]\}}}||krt|� qW d S )N)rU )rV r; )r* rL rN r, r> r� r r r rO � s z*DirectAdmin.list_users_in_reseller_packagec C sF g }t j�| j�r,| �| j�}dd� |D �}|�| �� � tt|��S )z0 Return list of resellers names c S s g | ]}|� � r|� � �qS r )r� )ry rz r r r r{ � s z.DirectAdmin._get_resellers.<locals>.<listcomp>) ri rj rk �DA_RESELLERS_LISTrG �extendr- r� r� )r* Zresellers_listrr r r r r~ � s zDirectAdmin._get_resellersc C s> g }t j�| j|d�}t j�|�r:| �|�}dd� |D �}|S )z� Get all packages for giver reseller :param string reseller: Reseller name :rtype: list of strings :return: List of packages names that belongs to given reseller z packages.listc S s g | ]}|� � r|� � �qS r )r� )ry rz r r r r{ � s z6DirectAdmin._get_reseller_packages.<locals>.<listcomp>)ri rj r� r� rk rG )r* rN Z packages_listZ package_pathrr r r r �_get_reseller_packages� s z"DirectAdmin._get_reseller_packagesc C s� d }xr| � � D ]f}y| �|�}W n0 tk rP } z|d kr>|}wW d d }~X Y nX t�|fdd�|d��� }t|� qW |d k r�|�d S )Nc S s d� | ||�S )Nz{}{} {} )r( )rp Zpkg�resr r r �<lambda>� � z5DirectAdmin.list_resellers_packages.<locals>.<lambda>r )r~ r� r� � functools�reducerm r; )r* r, Zfirst_exceptionrN �packagesr? Z out_stringr r r rR � s z#DirectAdmin.list_resellers_packagesc C s t d�| �|��� d S )Nr� )r; r� r� )r* rN r, r r r rS � s z"DirectAdmin.list_reseller_packages)N)FN)FN)F)F)F)F)F)F)F)F)F)F)F)*r r r r� ZDA_CONFr� r� r� r� r� r# r r� r+ rG r� r� r� r� r r r� rV rX rY r� r� r� r� r- r� rQ rK rT rP rM rO r~ r� rR rS r� r r )r$ r r > sD : ) r c @ s e Zd Zdd� ZdS )� ISPManagerGetPackagesExceptionc C s t �| |� d S )N)r� r+ )r* rd r r r r+ � s z'ISPManagerGetPackagesException.__init__N)r r r r+ r r r r r� � s r� c sp e Zd ZdZdZdZ� fdd�Zdd� Zedd d ��Z dd d �Z dd d�Z ddd�Z ddd�Z ddd�Z� ZS )r z!/usr/local/ispmgr/etc/ispmgr.confNZ mgrsecurec sH t t| ��� t�� tj�d�}t|d �| _t� � | _ t � � | _ d S )Nr� r )r[ r r+ r ZgetCPZ CP_VERSIONrx r= � isp_major_verZispmanager5_is_masterZisp5_is_masterr r] r) )r* Z ver_parts)r$ r r r+ � s zISPManager.__init__c C s� i }| j dk�r6tj�| j�s,td| j ��y�t| jd��� }|�d�}|� d� x�|D ]�}|�� d }t | j � |� dd���}|�d�}xb|D ]Z}|�� �d �r�d �|�� �� d d � �}||kr�|| d t|� ||< q�t|�||< q�W qXW W n> ttfk �r2 } ztd | jt|�f ��W d d }~X Y nX ntdd l}yT|�| j�j} x@| D ]8} y| j � | �}d||< W n tjk �r� Y nX �qTW W n tk �r� Y nX |S )z� Retrives list of pairs "uid - package" :return: Dictionary: uid -> package_name Example: {512: 'custom', 513: 'test1'} Case 2: (Error): (1, "error message") � zERROR: Can't read %srA ZAccountr �"r r� ZPreset� rw NzERROR: Can't read %s - %s�default)r� ri rj rk � ISP4_CONFr� rB �readrx �popr= r# r9 �replacerm r` r� r� �OSError�IOError�grpZgetgrnam�ISP5_GRPZgr_memr r: �KeyError) r* �dict_uid_package�linesrs r> Zline1Zpkg_namer? r� Z panel_usersr� r r r �_get_user_packages_dict� s@ * z"ISPManager._get_user_packages_dictFc C s d}t |� dS )z� Get reseller name from user's config file :param string uid: uid of user :param bool debug: Do produce debug output or don't :rtype: string :return: reseller name r� N)r; )�clsr> r, rN r r r rY s zISPManager.get_resellerc C sd y*| � � }x t|�� �D ] }t|� qW W n4 tk r^ } zt|� t�d� W d d }~X Y nX d S )Nrw )r� r� �valuesr; r� r0 rb )r* r, r� rL r? r r r rQ s zISPManager.list_packagesc C sl y2| � � }x$|D ] }tt|�d || � qW W n4 tk rf } zt|� t�d� W d d }~X Y nX d S )Nr� rw )r� r; r� r� r0 rb )r* r, r� r> r? r r r rK s zISPManager.list_allc C s| y(t |�}| �� }||kr&t|| � W nN tk r\ } zt|� t�d� W d d }~X Y n tk rv td� Y nX d S )Nrw zIncorrect user id)r= r� r; r� r0 rb r8 )r* Zs_uidr, r> r� r? r r r rP # s z ISPManager.list_user_packagec C sl y2| � � }x$|D ] }||| krtt|�� qW W n4 tk rf } zt|� t�d� W d d }~X Y nX d S )Nrw )r� r; r� r� r0 rb )r* rL r, r� r> r? r r r rM / s z ISPManager.list_users_in_packagec C st y:| � � }x,|D ]$}tt|�d t|| � d � qW W n4 tk rn } zt|� t�d� W d d }~X Y nX d S )N�,z,adminrw )r� r; r� r� r0 rb )r* r, r� r> r? r r r rT 9 s (zISPManager.list_users)F)F)F)F)F)F)r r r r� r# r� r+ r� � classmethodrY rQ rK rP rM rT r� r r )r$ r r � s / r c s� e Zd ZdZdZdZdZdZ� fdd�Zd+dd �Z d,d d �Z d-d d�Z d.e e d�dd�Zd/dd�Zd0dd�Zd1dd�Zd2dd�Zd3dd�Zd4d d �Zeeee e ef d �d d �Zee d!�d"d#�Ze ed$�d%d&�Zd5d'd(�Z d6d)d*�Z � Z S )7r zz See following link for information about database: https://github.com/plesk/db-schemas/blob/master/psadb.xml z/etc/psa/.psa.shadow�1�NoneNc s t t| ��� t�� | _d S )N)r[ r r+ r r] r) )r* )r$ r r r+ M s zPlesk.__init__c C s` t j�| j�sdS t| j��� �� }tjddt |�dddd�}|� � �}|j ||d�S Q R X d S ) NFZ localhostr� ZpsaT�utf8)Zhostr� �passwd�dbZ use_unicode�charset)�args) ri rj rk � PSA_SHADOWrB r� rm r ZMySQLConnectorr� ZconnectZ execute_query)r* Zsql�datar� Z connectorr� r r r �fetch_data_from_dbQ s zPlesk.fetch_data_from_dbFc c s� d}|dk r&d� |�}| �||g�}n,|dk rHd� |�}| �||g�}n | �|�}xR|D ]J\}}}|pj| j}|dkrxdn|}| j||d�}|rX|dk rX|||fV qXW dS )a Generate triple with uid, it's package, it's reseller :param bool debug: Do produce debug output or don't :param string reseller_name: filter by this reseller :param string username: filter by this username (ignored if reseller_name set) :rtype: generator of (int, string, string) :return: (uid, package name, reseller name) a� SELECT sys_user.login, template.name, reseller.login FROM sys_users AS sys_user JOIN hosting AS hosting ON hosting.sys_user_id = sys_user.id JOIN domains AS domain ON hosting.dom_id = domain.id AND domain.webspace_id=0 JOIN clients AS reseller ON domain.vendor_id = reseller.id LEFT JOIN Subscriptions AS subscription ON subscription.object_id = domain.id AND subscription.object_type='domain' LEFT JOIN PlansSubscriptions AS plan ON plan.subscription_id = subscription.id LEFT JOIN Templates AS template ON plan.plan_id = template.id WHERE sys_user.mapped_to IS NULL AND (template.type = 'domain' OR template.type IS NULL) Nz{} AND reseller.login = %sz{} AND sys_user.login = %sr� r )r/ r, )r( r� � NO_PACKAGEr@ ) r* r, rU r/ �queryrp rL rN r> r r r rV [ s zPlesk._get_user_packagesc C s4 dt | jd� }x | �|�D ]}t|d � q W dS )z� Print packages (exclude reseller's) :param bool debug: Do produce debug output or don't :return: packages names zs SELECT name FROM psa.Templates WHERE owner_id = %(admin_id)s AND type = 'domain'; )�admin_idr N)�dict�ADMIN_IDr� r; )r* r, r� rs r r r rQ � s zPlesk.list_packages)r> r, c C sl d}| j �t|��}|rTd�dgt|� �}d| }| �|t|��}|rT|d d }t|dkrbdn|� dS )z� Get reseller name from DB :param string uid: uid of user :param bool debug: Do produce debug output or don't :rtype: string :return: reseller name r r� z%saJ SELECT reseller.login FROM sys_users AS sys_user JOIN hosting AS hosting ON hosting.sys_user_id = sys_user.id JOIN domains AS domain ON hosting.dom_id = domain.id JOIN clients AS reseller ON domain.vendor_id = reseller.id WHERE sys_user.login IN (%s)r r� N)r# r< r= r� r� r� �tupler; )r* r> r, rN Z usernamesZformat_stringsr� rp r r r rY � s zPlesk.get_resellerc C s8 | j |d�}x&|D ] \}}}td�t|�|�� qW dS )z� Print info about user's packages (include resellers) :param bool debug: Do produce debug output or don't :return: pairs "uid package", where uid - unique id of user package - name of user's package )r, z{} {}N)rV r; r( r� )r* r, r� r> rL rW r r r rK � s zPlesk.list_allc C s: | j |d�}x(|D ] \}}}td�t|�||�� qW dS )aF Print info about user's packages (include resellers) and thouse resellers :param bool debug: Do produce debug output or don't :return: triples "uid,package,reseller", where uid - unique id of user package - name of user's package reseller - name of package's reseller )r, z{},{},{}N)rV r; r( r� )r* r, r� r> rL rN r r r rT � s zPlesk.list_usersc C s� g }y| j �t|��}W nJ tjk rH } z|r8t|� W dd}~X Y nL tk rb td� Y n2X x.|D ]&}| j||d�}|�dd� |D �� qjW x|D ] }t|� q�W dS )z� Print all packages for user's uid :param uid: user's unique id :param bool debug: Do produce debug output or don't :return: package name for user's uid NzIncorrect user id)r, r/ c S s g | ] }|d �qS )rw r )ry rs r r r r{ � s z+Plesk.list_user_package.<locals>.<listcomp>) r# r< r= r r: r; r8 rV r� ) r* r> r, r� �namesr? rf rp rL r r r rP � s zPlesk.list_user_packagec C s> dt || jd� }x&| �|�D ]}t| j�|d �� q W dS )z� Print all users in package :param package: name of package :param bool debug: Do produce debug output or don't :return: user's uid a� SELECT t5.login FROM psa.Templates AS t1 JOIN psa.PlansSubscriptions AS t2 ON t2.plan_id = t1.id JOIN psa.Subscriptions AS t3 ON t3.id = t2.subscription_id JOIN psa.domains AS t4 ON t4.id = t3.object_id JOIN psa.hosting AS t6 ON t6.dom_id = t4.id RIGHT JOIN psa.sys_users AS t5 ON t6.sys_user_id = t5.id WHERE t1.name = '%(package)s' AND t1.owner_id = %(admin_id)s AND t1.type = 'domain'; )rL r� r N)r� r� r� r; r# r9 )r* rL r, r� Zloginr r r rM � s zPlesk.list_users_in_packagec C s� d| }| � |�}t|�dkr"dS t|d d �}dt||d� }| � |�}x^|D ]V}y| j�|d �} W n0 tjk r� } z|r�t| � W dd} ~ X Y qRX tt | �� qRW dS )z� Print info about users in resellers package :param package: package name :param reseller: name of package's reseller :param bool debug: Do produce debug output or don't :return: users uids zCSELECT id FROM psa.clients WHERE type = 'reseller' AND login = '%s'rw Nr a� SELECT t5.login FROM psa.Templates AS t1 JOIN psa.PlansSubscriptions AS t2 ON t2.plan_id = t1.id JOIN psa.Subscriptions AS t3 ON t3.id = t2.subscription_id JOIN psa.domains AS t4 ON t4.id = t3.object_id JOIN psa.hosting AS t6 ON t6.dom_id = t4.id RIGHT JOIN psa.sys_users AS t5 ON t6.sys_user_id = t5.id WHERE t1.type = 'domain' AND t1.name = '%(package)s' AND t1.owner_id = '%(reseller_id)s'; )rL � reseller_id) r� r� r= r� r# r9 r r: r; r� ) r* rL rN r, Zreseller_queryr� r� rp rs Zluidr? r r r rO � s z$Plesk.list_users_in_reseller_packagec C sX dt | jd� }xB| �|�D ]4}|d }|d dkr8dn|d }td�||�� q W dS ) z� Print list of resellers packages :param bool debug: Do produce debug output or don't :return: pairs "reseller package", where package - package name reseller - package's reseller z�SELECT t2.login, t1.name FROM (SELECT name, owner_id, type FROM psa.Templates) AS t1 JOIN psa.clients AS t2 ON t1.owner_id = t2.id WHERE t1.owner_id != %(admin_id)s AND t1.type = 'domain';)r� rw r r� �rootz{} {}N)r� r� r� r; r( )r* r, r� rs rL rN r r r rR s z Plesk.list_resellers_packages)r� c C s\ g }d}xN| � |�D ]@}|d }|d dkr0dn|d }|d }|�||t|�f� qW |S )z� Return list of non-reseller packages with the plesk DB id [(reseller, package, plesk_id), (None, package, plesk_id)] z�SELECT t2.login, t1.name, t1.id FROM (SELECT id, name, owner_id, type FROM psa.Templates) AS t1 JOIN psa.clients AS t2 ON t1.owner_id = t2.id WHERE t1.type = 'domain';rw r r� r� � )r� �appendr= )r* r� r� rs rL rN Z_idr r r � list_domain_packages_with_id- s z"Plesk.list_domain_packages_with_id)� package_idr� c C s d}| � ||f�}|d d S )z8 Return package name by the plesk DB id z7SELECT id, name, type FROM psa.Templates WHERE id = %s;r rw )r� )r* r� r� r� r r r �$get_package_name_by_id_from_plesk_db> s z*Plesk.get_package_name_by_id_from_plesk_db)� package_namer� c C s d}| � ||f�}|d d S )z8 Return package id by the plesk DB name z9SELECT id, name, type FROM psa.Templates WHERE name = %s;r )r� )r* r� r� r� r r r �$get_package_id_by_name_from_plesk_dbF s z*Plesk.get_package_id_by_name_from_plesk_dbc C sR |dkr d}t �d|�s dS dt|| jd� }x | �|�D ]}t|d � q:W dS )z� Print list reseller's packages :param reseller: name of reseller :param bool debug: Do produce debug output or don't :return: packages names r� r� z^[\w_]*$Nz�SELECT t1.name FROM (SELECT name, owner_id, type FROM psa.Templates) as t1 JOIN psa.clients as t2 ON t1.owner_id = t2.id AND t2.login = '%(reseller)s' WHERE t1.type = 'domain' AND t1.owner_id != %(admin_id)s;)rN r� r )rg rn r� r� r� r; )r* rN r, r� rs r r r rS N s z Plesk.list_reseller_packagesc C s dgS )Nr� r )r* r, r r r r- b s zPlesk.list_admins)N)FNN)F)F)F)F)F)F)F)F)F)F) r r r r r� r� r� r# r+ r� rV rQ r� �boolrY rK rT rP rM rO rR r r r r= r� r� r� rS r- r� r r )r$ r r C s* ( # r c @ s e Zd Zdd� ZdS )r c C s |rt j�d� t �d� d S )Nz"Doesn't support InterWorx anymore r )r0 r1 r2 rb )r* r, r3 r r r r4 g s zInterWorx._warningN)r r r r4 r r r r r f s r c @ s e Zd Zdd� ZdS )r c C s t d| ��d S )Nz'%s wasn't implemented for Unknown panel)�NotImplementedError)r* r, r3 r r r r4 n s zUnknown._warningN)r r r r4 r r r r r m s r c @ s e Zd Zdd� ZdS )r c C s t d| ��d S )Nz)%s wasn't implemented for HostingNG panel)r� )r* r, r3 r r r r4 s s zHostingNG._warningN)r r r r4 r r r r r r s r )*Z __future__r r �builtinsr ri rg r0 r5 r� �typingr r r Zclcommonr r Zclcommon.clprocr Zclcommon.utilsr Zclcommon.cpapi.pluginsr r r r Z cldetectlibr rc r� r r! �objectr" r r r� r r r r r r r r r �<module> sB | | %