HEX

File: //opt/alt/python37/lib/python3.7/site-packages/__pycache__/clcontrollib.cpython-37.opt-1.pyc
B

CMn����@sXddlmZddlmZddlmZddlZddlZddlZddlZddl	Z	ddl
mZmZm
Z
ddlmZmZddlmZddlmZdd	lmZmZmZmZddlZd
ZGdd�de�Zd
d�ZGdd�de �Z!Gdd�de!�Z"Gdd�de!�Z#Gdd�de�Z$Gdd�de!�Z%Gdd�de!�Z&Gdd�de!�Z'Gdd�de!�Z(Gdd �d e!�Z)dS)!�)�print_function)�absolute_import)�rangeN)�Tuple�Optional�List)�ClPwd�	mysql_lib)�ProcLve)�grep)�cpanel�directadmin�
ispmanager�plesk�0c@seZdZdZdS)�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__�rr�clcontrollib.pyrsrcCsrt��tjdkrt�Stjdkr(t�Stjdkr8t�StjdkrHt�StjdkrXt�Stjdkrht�St	�SdS)N�PleskZcPanel�	InterWorx�
ISPManager�DirectAdmin�	HostingNG)
�detect�	getCPNameZCP_NAMEr�Cpanelrrrr�Unknownrrrr�detect_panelclass's





r!c@s�eZdZdd�Zd+dd�Zdd�Zdd	�Zed
d��Zd,dd�Z	edd��Z
ed-dd��Zd.dd�Zd/dd�Z
d0dd�Zd1dd�Zd2dd�Zd3dd �Zd4d!d"�Zd5d#d$�Zd6d%d&�Zd7d'd(�Zd8d)d*�ZdS)9�GeneralPanelcCsNt�|_|jj|_t��|_|j��|j��krDt	d�
|j|j���d|_dS)Nz+Class "{}" not should be used on panel "{}")r�
users_data�	__class__r�
class_namerrZcurrent_panel_name�lower�UserWarning�format�cpapi_plugin)�selfrrr�__init__;s


zGeneralPanel.__init__FcCs
|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*�debugrrr�list_adminsGszGeneralPanel.list_adminscCs|j�|�S)z|
        Return True if username is in admin names
        :param str username: user to check
        :return: bool
        )r)�is_admin)r*�usernamerrrr.OszGeneralPanel.is_admincCs |rtj�d||jjf�dS)Nz%s is not implemented for %s
)�sys�stderr�writer$r)r*r,�method_namerrr�_warningWszGeneralPanel._warningcCst�tjd�|��dS)NzCL_CONTROLLIB: {})�syslogZLOG_WARNINGr()�msgrrr�
_slog_warning\szGeneralPanel._slog_warningNTc
Cs|dk	r|dk	rtd��|dkr0|dkr0td��|dk	r�y|j�|�Stjk
r�}z$|rdt|�|rx|�d�|��dSd}~XYnXy|j�t	|��dStjk
r�}z$|r�t|�|r�|�d�|��dSd}~XYntk
�r�td�dSXdS)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 systemrz/User with uid="{}" is not present in the systemzIncorrect user id)
�
ValueErrorr#�get_uidr�NoSuchUserException�printr7r(�	get_names�int)r*�uidr/r,Zsyslog_�errr�_check_and_get_user`s8z GeneralPanel._check_and_get_userc	Cs"t|d��}|��}WdQRX|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�contentrrr�get_file_lines�s
zGeneralPanel.get_file_lines�wc	Cs$t||��}|�|�WdQRXdS)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)rDrFZopen_methodrErrr�write_file_lines�szGeneralPanel.write_file_linescCs|�|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,rrr�list_all�s
zGeneralPanel.list_allcCs|�|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,rrr�list_users_in_package�s
z"GeneralPanel.list_users_in_packagecCs|�|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,rrr�list_users_in_reseller_package�sz+GeneralPanel.list_users_in_reseller_packagecCs|�|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,rrr�list_user_package�s
zGeneralPanel.list_user_packagecCs|�|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,rrr�
list_packages�s	zGeneralPanel.list_packagescCs|�|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,rrr�list_resellers_packages�s	z$GeneralPanel.list_resellers_packagescCs|�|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*rNr,rrr�list_reseller_packages�sz#GeneralPanel.list_reseller_packagescCs|�|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,rrr�
list_userss	zGeneralPanel.list_userscCs|�|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_namerrr�_get_user_packagess	zGeneralPanel._get_user_packagescCsT|jdkrP|dkr"|rtd�dSx,|j||d�D]\}}}td�||��q2WdS)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,rUz{},{})r%r;rVr()r*r,rNr>rL�_rrr�list_reseller_userss
z GeneralPanel.list_reseller_userscCs|�|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,rrr�get_reseller)szGeneralPanel.get_reseller)F)NNFT)rH)F)F)F)F)F)F)F)F)FN)FN)F)rrrr+r-r.r4�staticmethodr7r@rGrJrKrMrOrPrQrRrSrTrVrXrYrrrrr"9s*

2









r"cs�eZdZdZdZdZdZdZ�fdd�Zdd	�Z	d
d�Z
dd
�Zdd�Zd)dd�Z
d*dd�Zdd�Zd+dd�Zd,dd�Zd-dd�Zd.dd�Zd/dd �Zd0d!d"�Zd1d#d$�Zd2d%d&�Zd3d'd(�Z�ZS)4rz/var/cpanel/usersz/var/cpanel/packagesz/var/cpanel/resellersz/etc/trueuserownersNcs(tt|���|j��|_t��|_dS)N)	�superrr+r#Zget_sys_min_uid�min_uidr�PanelPluginr))r*)r$rrr+<szCpanel.__init__c
Cs�d}y"t|d��}|��}WdQRXWnptk
r�}zRd|�d|�d�}|�|j�rd|d|��7}t|tjd�|�|�t�	t
�Wdd}~XYnX|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/rANzUnable to read file 'z
', error is: z-. It looks like you use not an UTF-8 charset.z
Please read the instruction: )�file)rBrC�UnicodeDecodeError�
startswith�USERDIRr;r0r1r7�exit�BAD_CODING_ERROR_CODE)r*rDZ
ENCODING_LINKrErFr?�messagerrrrGAs	
zCpanel.get_file_linescCsli}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�strippedrnrrr�_get_users_resellers_dictYs

z Cpanel._get_users_resellers_dictcCs0g}tj�|j�r,|�|j�}dd�|D�}|S)zw
        Get resellers list from RESELLERS_PATH

        :rtype: list
        :return: list of resellers names
        cSs g|]}|���dd�d�qS)�:�r)rm�split)�.0�lrrr�
<listcomp>~sz)Cpanel._get_resellers.<locals>.<listcomp>)rirjrk�RESELLERS_PATHrG)r*�	resellersrrrrr�_get_resellersts
zCpanel._get_resellerscCsNd}tj�|�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)rirjrkrG�listrrmrx)r*rjrLrrrprrr�get_package�s

zCpanel.get_packageFcCs�d}|j||d�}|rltj�|j|�}tj�|�rl|�|�}ttdddd|d��}|rl|d�	��
d�d}|r�||��kr�|SdS)	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�rr�rw)r@rirj�joinrarkrGr�rrmrxr~)r*r>r,rNr/rjrrrprrr�
_get_reseller�s	

zCpanel._get_resellercCst|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,rrrrY�s
zCpanel.get_resellerccsbtj�|j�r^xNt�|j�D]>}tj�tj�|j|��rD|dfVqtj�|j|�dfVqWdS)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)rirj�isdir�
PACKAGEDIR�listdirrkr�)r*�filenamerrr�
_get_packages�s
zCpanel._get_packagescCs^|��}xP|��D]D\}}|rFd|kr<|�dd�d|kr<qt|�q|rtd|�qWdS)NrWrwrzWarning: %s is not a file.)r~r�rxr;)r*r,r}r��is_filerrrrQ�s
zCpanel.list_packagesc	cs�|��}|sdStj�|j�r�x�t�|j�D]�}y|j�|�}||jkrJw,Wn0t	j
k
r|}z|rlt|�Wdd}~XYq,Xtj�|j|�}||kr�||}n|�
||�}|�|�r�dn|}|dk	r�||kr�q,||�|�|fVq,WdS)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)rurirjr�rar�r#r9r\rr:r;r�r�r.r�)	r*r,rUZusers_resellers_dictr�r>r?rjrNrrrrV�s(


zCpanel._get_user_packagescCsNxH|j||d�D]6\}}}|dkr|dkr6t||�q||krt|�qWdS)N)rUr)rVr;)r*rLr,rNr>�self_packageZ
self_resellerrrrrK�szCpanel.list_allcCs�y|j�t|��}Wn<tjk
r6|r2td�Yn^tk
rPtd�YnDXx@|D]8}|jd|}tj	�
|�rX|�|�}|dkrXt|�qXWdS)Nz,getcontrolpaneluserspackages: User not foundzIncorrect user id�/r)r#r<r=rr:r;r8rarirjrkr�)r*r>r,�users�userrjrLrrrrPs

zCpanel.list_user_packagecCs�|��}xv|��D]j\}}|rhd|krh|�dd�d}||krBqq||dk	rV||krVqq|td||f�q|s|rtd|�qWdS)NrWrwrz%s %szWarning: %s is not a file.)r~r�rxr;)r*r,rNr}r�r�Zres_namerrrrRszCpanel.list_resellers_packagescCs6|��}||kr$|r td|�dS|j||d�dS)NzError: %s is not reseller)r,rN)r~r;rR)r*rNr,r}rrrrSszCpanel.list_reseller_packagescsd|���d|kr@|�d��t��fdd�tdt���D��r@dS|dk	rT|�|�n|r`td�dS)NrWc3s$|]}d��d|���kVqdS)rWN)r�)ry�i)�partsr}rr�	<genexpr>+sz/Cpanel.list_users_in_package.<locals>.<genexpr>rwz/getcontrolpaneluserspackages: Undefined package)r~rx�anyr�lenrKr;)r*rLr,r)r�r}rrM's
"zCpanel.list_users_in_packagecCs2|��}||kr"|j|||d�n|r.td�dS)N)r,rNz8getcontrolpaneluserspackages: Undefined reseller package)r~rKr;)r*rLrNr,r}rrrrO2s
z%Cpanel.list_users_in_reseller_packagecCs0x*|�|�D]\}}}td|||f�qWdS)Nz%s,%s,%s)rVr;)r*r,r>r�rNrrrrT9szCpanel.list_users)F)F)F)FN)NFN)F)FN)F)F)F)F)rrrrar�r|rlr#r+rGrur~r�r�rYr�rQrVrKrPrRrSrMrOrT�
__classcell__rr)r$rr4s,





$





rcs0eZdZdZdZdZdZdZdZdZ	dZ
e�Z�fdd	�Z
d
d�Zdd
�Zdd�Zdd�Zd<eeeeed�dd�Zd=dd�Zd>dd�Zd?dd�Zdd�Zdd�Zd d!�Zd@d"d#�ZdAd$d%�Zd&d'�ZdBd(d)�ZdCd*d+�Z dDd,d-�Z!dEd.d/�Z"dFd0d1�Z#dGd2d3�Z$d4d5�Z%d6d7�Z&dHd8d9�Z'dId:d;�Z(�Z)S)Jrz"/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.listrNcs2tt|���t��|_t��|_|j�	�|_
dS)N)r[rr+rZdetect_DA_admin�DA_ADMINr
r]r)Zget_encoding_name�ENCODING)r*)r$rrr+Is

zDirectAdmin.__init__c
Cs�y(t|d|jd��}|��}WdQRX|Stk
r~}z8d|�d|j�d�}|�|d|�d��t|�|�Wdd}~XYnXdS)NrA)�encodingzUnable to read file 'z '. It looks like you use not an z	 charset.z Error is: �.)rBr�rCr_r7�	Exception)r*rDrErFr?rdrrrrGPszDirectAdmin.get_file_linesc
CsNy
|�|�Stk
rH}z d}|�|�||��t��Wdd}~XYnXdS)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�r7r(r)r*r/r?Ztplrrr�_get_user_info[s
zDirectAdmin._get_user_infocCs*tj�|j|d�}y|�|�}Wnd}YnX|s>td��ttdddd|d��}ttdddd|d��}y|d���	d	�d
}Wnt
k
r�td��YnXy|d���	d	�d
}Wnt
k
r�td��YnX|�||�}|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�zcreator\=.*$rr�rwz#No "usertype" record in user configz"No "creator" record in user configrNrZDEFAULT)rirjr��DA_USERS_DIRrGr8r�rrmrx�
IndexError�_get_user_packager#r9�_is_reseller_limits_enabled)r*r/�userprofile�userprofile_content�result_usertypeZresult_creator�usertyperNrL�lvp_idr>rrrr�ksH



z DirectAdmin._get_user_info_innercCs|j�|�S)N)�proc_lveZ	exist_lvp)r*r�rrrr��sz'DirectAdmin._is_reseller_limits_enabled)r/r��returncCs�|sBtj�|j|d�}y|�|�}Wnd}YnX|sBtd��ttdddd|d��}ttdddd|d��}y|d���	d	�d
}Wnt
k
r�td��YnX|dkr�|r�y|d���	d	�d
}Wnt
k
r�td
��YnX|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.confzpackage\=.*$F)r�r�r�r�zoriginal_package\=.*$rr�rwz"No "package" record in user configZcustomz+No "original_package" record in user config)rirjr�r�rGr8r�rrmrxr�)r*r/r�r�Zresult_packageZresult_original_packagerLrrrr��s<


zDirectAdmin._get_user_packageFc	cs�t|���}x|t�|j�D]l}||kr(q|j||d�dkr<qy|�|�\}}}Wntk
rfwYnX|dk	rz||krzq|||fVqWdS)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-rir�r�r@r�r)r*r,rUr-r/r>rLrNrrrrV�szDirectAdmin._get_user_packagesc		Cs�tj�|j|d�}tj�|�s"dS|�|�}xN|D]F}y|�|�d��\}}}Wntk
rfw2YnXt	d�
||��q2W|�|�s�y|�|�\}}}Wntk
r�dSXt	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{},{})rirjr�r��existsrGr��rstriprr;r(r.)	r*r,rNrjZreseller_usersr/r>rLrWrrrrX�s 


zDirectAdmin.list_reseller_userscCsH|j||d�}|rDy|�|�\}}}Wntk
r:dSXt|�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�rr;)r*r>r,r/rWrNrrrrY	szDirectAdmin.get_resellercCs
|�|�S)z�
        Check login_name is da admin. Backward compatibility wrapper
        :param login_name: Admin name
        :return:
        )r.)r*�
login_namerrr�is_da_adminszDirectAdmin.is_da_admincCsld}tj�|j|d�}tj�|�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�rr�rw�adminT)
rirjr�r�rkrGr�rrmrx)r*r�rprjrrr�r�rrr�is_da_admin_for_hook"s

z DirectAdmin.is_da_admin_for_hookc	Cs�yddlm}m}m}m}Wn&tk
rBtd�t�d�YnX|�	�}xR|D]J}yt
|�WqR|||fk
r�}ztdt|��Wdd}~XYqRXqRWdS)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
rwz%s
N)Zclsudor�r�r�r��ImportErrorr;r0rbr-�eval�str)	r*Zs_cmdr�r�r�r��	da_adminsr�r?rrr�add_all_da_admins_to_sudoers@s
z(DirectAdmin.add_all_da_admins_to_sudoerscCs
|�|�S)z<
           Wrapper for compatibility with old code
        )r-)r*r,rrr�get_all_da_adminsTszDirectAdmin.get_all_da_adminscCs0g}tj�|j�r,|�|j�}dd�|D�}|S)zr
        Get all DA admins list

        :rtype: list of strings
        :return: List of DA admins names
        cSsg|]}|�d��qS)r�)r�)ryrsrrrr{dsz+DirectAdmin.list_admins.<locals>.<listcomp>)rirjrk�
DA_ADMIN_LISTrG)r*r,r�rrrrrr-Zs
zDirectAdmin.list_adminsccs*|�|j�}x|D]}|�d�VqWdS)Nr�)rG�DA_ADMINS_PACKAGESr�)r*rrrsrrrr�gs
zDirectAdmin._get_packagescCsx|��D]}t|�q
WdS)N)r�r;)r*r,rsrrrrQlszDirectAdmin.list_packagescCs&x |��D]\}}}t||�q
WdS)N)rVr;)r*r,r>rLrWrrrrKpszDirectAdmin.list_allcCs2x,|��D] \}}}tdt|�||f�q
WdS)Nz%s,%s,%s)rVr;r�)r*r,r>rLrNrrrrTtszDirectAdmin.list_userscCsH|j||d�}|rDy|�|�\}}}Wntk
r:dSXt|�dS)N)r>r,)r@r�rr;)r*r>r,r/rWrLrrrrPxszDirectAdmin.list_user_packagecCs0x*|jdd�D]\}}}||krt|�qWdS)Nr)rU)rVr;)r*rLr,r>�package_rNrrrrM�sz!DirectAdmin.list_users_in_packagecCs0x*|j|d�D]\}}}||krt|�qWdS)N)rU)rVr;)r*rLrNr,r>r�rrrrO�sz*DirectAdmin.list_users_in_reseller_packagecCsFg}tj�|j�r,|�|j�}dd�|D�}|�|���tt|��S)z0
        Return list of resellers names
        cSsg|]}|��r|���qSr)r�)ryrzrrrr{�sz.DirectAdmin._get_resellers.<locals>.<listcomp>)	rirjrk�DA_RESELLERS_LISTrG�extendr-r�r�)r*Zresellers_listrrrrrr~�szDirectAdmin._get_resellerscCs>g}tj�|j|d�}tj�|�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.listcSsg|]}|��r|���qSr)r�)ryrzrrrr{�sz6DirectAdmin._get_reseller_packages.<locals>.<listcomp>)rirjr�r�rkrG)r*rNZ
packages_listZpackage_pathrrrrr�_get_reseller_packages�s
z"DirectAdmin._get_reseller_packagescCs�d}xr|��D]f}y|�|�}Wn0tk
rP}z|dkr>|}wWdd}~XYnXt�|fdd�|d���}t|�qW|dk	r�|�dS)NcSsd�|||�S)Nz{}{} {}
)r()rpZpkg�resrrr�<lambda>��z5DirectAdmin.list_resellers_packages.<locals>.<lambda>r)r~r�r��	functools�reducermr;)r*r,Zfirst_exceptionrN�packagesr?Z
out_stringrrrrR�s

z#DirectAdmin.list_resellers_packagescCstd�|�|���dS)Nr�)r;r�r�)r*rNr,rrrrS�sz"DirectAdmin.list_reseller_packages)N)FN)FN)F)F)F)F)F)F)F)F)F)F)F)*rrrr�ZDA_CONFr�r�r�r�r�r#r
r�r+rGr�r�r�r�rrr�rVrXrYr�r�r�r�r-r�rQrKrTrPrMrOr~r�rRrSr�rr)r$rr>sD:)













rc@seZdZdd�ZdS)�ISPManagerGetPackagesExceptioncCst�||�dS)N)r�r+)r*rdrrrr+�sz'ISPManagerGetPackagesException.__init__N)rrrr+rrrrr��sr�cspeZdZdZdZdZ�fdd�Zdd�Zedd	d
��Z	ddd�Z
dd
d�Zddd�Zddd�Z
ddd�Z�ZS)rz!/usr/local/ispmgr/etc/ispmgr.confNZ	mgrsecurecsHtt|���t��tj�d�}t|d�|_t�	�|_
t��|_
dS)Nr�r)r[rr+rZgetCPZ
CP_VERSIONrxr=�
isp_major_verZispmanager5_is_masterZisp5_is_masterrr]r))r*Z	ver_parts)r$rrr+�s
zISPManager.__init__c
Cs�i}|jdk�r6tj�|j�s,td|j��y�t|jd���}|�d�}|�	d�x�|D]�}|��d}t
|j�|�
dd���}|�d�}xb|D]Z}|���d	�r�d
�|����dd��}||kr�||d
t|�||<q�t|�||<q�WqXWWn>ttfk
�r2}ztd
|jt|�f��Wdd}~XYnXntddl}yT|�|j�j}	x@|	D]8}
y|j�|
�}d||<Wntjk
�r�YnX�qTWWntk
�r�YnX|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 %srAZAccountr�"rr�ZPreset� rwNzERROR: Can't read %s - %s�default)r�rirjrk�	ISP4_CONFr�rB�readrx�popr=r#r9�replacermr`r�r��OSError�IOError�grpZgetgrnam�ISP5_GRPZgr_memrr:�KeyError)r*�dict_uid_package�linesrsr>Zline1Zpkg_namer?r�Zpanel_usersr�rrr�_get_user_packages_dict�s@




*
z"ISPManager._get_user_packages_dictFcCsd}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,rNrrrrYs
zISPManager.get_resellerc
Csdy*|��}xt|���D]}t|�qWWn4tk
r^}zt|�t�d�Wdd}~XYnXdS)Nrw)r�r��valuesr;r�r0rb)r*r,r�rLr?rrrrQszISPManager.list_packagesc
Csly2|��}x$|D]}tt|�d||�qWWn4tk
rf}zt|�t�d�Wdd}~XYnXdS)Nr�rw)r�r;r�r�r0rb)r*r,r�r>r?rrrrKs
 zISPManager.list_allc
Cs|y(t|�}|��}||kr&t||�WnNtk
r\}zt|�t�d�Wdd}~XYntk
rvtd�YnXdS)NrwzIncorrect user id)r=r�r;r�r0rbr8)r*Zs_uidr,r>r�r?rrrrP#szISPManager.list_user_packagec
Csly2|��}x$|D]}|||krtt|��qWWn4tk
rf}zt|�t�d�Wdd}~XYnXdS)Nrw)r�r;r�r�r0rb)r*rLr,r�r>r?rrrrM/s
z ISPManager.list_users_in_packagec
Csty:|��}x,|D]$}tt|�dt||�d�qWWn4tk
rn}zt|�t�d�Wdd}~XYnXdS)N�,z,adminrw)r�r;r�r�r0rb)r*r,r�r>r?rrrrT9s
(zISPManager.list_users)F)F)F)F)F)F)rrrr�r#r�r+r��classmethodrYrQrKrPrMrTr�rr)r$rr�s	/
	
	


rcs�eZdZdZdZdZdZdZ�fdd�Zd+dd	�Z	d,dd�Z
d-d
d�Zd.ee
d�dd�Zd/dd�Zd0dd�Zd1dd�Zd2dd�Zd3dd�Zd4dd�Zeeeeeefd�dd �Zeed!�d"d#�Zeed$�d%d&�Zd5d'd(�Zd6d)d*�Z�ZS)7rzz
    See following link for information about database:
    https://github.com/plesk/db-schemas/blob/master/psadb.xml
    z/etc/psa/.psa.shadow�1�NoneNcstt|���t��|_dS)N)r[rr+rr]r))r*)r$rrr+MszPlesk.__init__c	Cs`tj�|j�sdSt|j�����}tjddt	|�dddd�}|�
��}|j||d�SQRXdS)	NFZ	localhostr�ZpsaT�utf8)Zhostr��passwd�dbZuse_unicode�charset)�args)rirjrk�
PSA_SHADOWrBr�rmr	ZMySQLConnectorr�ZconnectZ
execute_query)r*Zsql�datar�Z	connectorr�rrr�fetch_data_from_dbQs

zPlesk.fetch_data_from_dbFc	cs�d}|dk	r&d�|�}|�||g�}n,|dk	rHd�|�}|�||g�}n
|�|�}xR|D]J\}}}|pj|j}|dkrxdn|}|j||d�}|rX|dk	rX|||fVqXWdS)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,rUr/�queryrprLrNr>rrrrV[s



zPlesk._get_user_packagescCs4dt|jd�}x|�|�D]}t|d�qWdS)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_idrN)�dict�ADMIN_IDr�r;)r*r,r�rsrrrrQ�szPlesk.list_packages)r>r,cCsld}|j�t|��}|rTd�dgt|��}d|}|�|t|��}|rT|dd}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
        rr�z%saJSELECT 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)rr�N)r#r<r=r�r�r��tupler;)r*r>r,rNZ	usernamesZformat_stringsr�rprrrrY�szPlesk.get_resellercCs8|j|d�}x&|D]\}}}td�t|�|��qWdS)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)rVr;r(r�)r*r,r�r>rLrWrrrrK�s
zPlesk.list_allcCs:|j|d�}x(|D] \}}}td�t|�||��qWdS)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)rVr;r(r�)r*r,r�r>rLrNrrrrT�s
zPlesk.list_usersc	
Cs�g}y|j�t|��}WnJtjk
rH}z|r8t|�Wdd}~XYnLtk
rbtd�Yn2Xx.|D]&}|j||d�}|�dd�|D��qjWx|D]}t|�q�WdS)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/cSsg|]}|d�qS)rwr)ryrsrrrr{�sz+Plesk.list_user_package.<locals>.<listcomp>)	r#r<r=rr:r;r8rVr�)	r*r>r,r��namesr?rfrprLrrrrP�s


zPlesk.list_user_packagecCs>dt||jd�}x&|�|�D]}t|j�|d��qWdS)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';
        )rLr�rN)r�r�r�r;r#r9)r*rLr,r�ZloginrrrrM�szPlesk.list_users_in_packagecCs�d|}|�|�}t|�dkr"dSt|dd�}dt||d�}|�|�}x^|D]V}y|j�|d�}	Wn0tjk
r�}
z|r�t|
�Wdd}
~
XYqRXtt	|	��qRWdS)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'rwNra�
            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#r9rr:r;r�)r*rLrNr,Zreseller_queryr�r�rprsZluidr?rrrrO�s




z$Plesk.list_users_in_reseller_packagecCsXdt|jd�}xB|�|�D]4}|d}|ddkr8dn|d}td�||��qWdS)	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�rwrr��rootz{} {}N)r�r�r�r;r()r*r,r�rsrLrNrrrrRs

zPlesk.list_resellers_packages)r�cCs\g}d}xN|�|�D]@}|d}|ddkr0dn|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';rwrr�r��)r��appendr=)r*r�r�rsrLrNZ_idrrr�list_domain_packages_with_id-sz"Plesk.list_domain_packages_with_id)�
package_idr�cCsd}|�||f�}|ddS)z8
        Return package name by the plesk DB id
        z7SELECT id, name, type FROM psa.Templates WHERE id = %s;rrw)r�)r*r�r�r�rrr�$get_package_name_by_id_from_plesk_db>sz*Plesk.get_package_name_by_id_from_plesk_db)�package_namer�cCsd}|�||f�}|ddS)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�rrr�$get_package_id_by_name_from_plesk_dbFsz*Plesk.get_package_id_by_name_from_plesk_dbcCsR|dkrd}t�d|�sdSdt||jd�}x|�|�D]}t|d�q:WdS)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;)rNr�r)rgrnr�r�r�r;)r*rNr,r�rsrrrrSNs	zPlesk.list_reseller_packagescCsdgS)Nr�r)r*r,rrrr-bszPlesk.list_admins)N)FNN)F)F)F)F)F)F)F)F)F)F)rrrrr�r�r�r#r+r�rVrQr��boolrYrKrTrPrMrOrRrrrr=r�r�r�rSr-r�rr)r$rrCs*


(





#
 
rc@seZdZdd�ZdS)rcCs|rtj�d�t�d�dS)Nz"Doesn't support InterWorx anymore
r)r0r1r2rb)r*r,r3rrrr4gszInterWorx._warningN)rrrr4rrrrrfsrc@seZdZdd�ZdS)r cCstd|��dS)Nz'%s wasn't implemented for Unknown panel)�NotImplementedError)r*r,r3rrrr4nszUnknown._warningN)rrrr4rrrrr msr c@seZdZdd�ZdS)rcCstd|��dS)Nz)%s wasn't implemented for HostingNG panel)r�)r*r,r3rrrr4sszHostingNG._warningN)rrrr4rrrrrrsr)*Z
__future__rr�builtinsrrirgr0r5r��typingrrrZclcommonrr	Zclcommon.clprocr
Zclcommon.utilsrZclcommon.cpapi.pluginsrr
rrZcldetectlibrrcr�rr!�objectr"rrr�rrrr rrrrr�<module>sB
||%