Current File : //opt/alt/python37/lib/python3.7/site-packages/__pycache__/clcontrollib.cpython-37.pyc
B
L#Lb�� � @ sL d dl mZ d dl mZ d dlmZ d dlZd dlZd dlZd dlZd dl Z d dl
m Z d dl m
Z
d dlmZ d dlmZ d d lmZmZmZmZ d dlZd
ZG d d � d e�Zd
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)�ClPwd)�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 �=/opt/alt/python37/lib/python3.7/site-packages/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�printr4 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)r? �
writelines)rA rC Z open_methodrB 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)r1 )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)r1 )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)r1 )r' rI �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)r1 )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)r1 )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)r1 )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)r1 )r' rK 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)r1 )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)r1 )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) rR z{},{})r" r8 rS r% )r' r) rK r; rI �_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)r1 )r' r; r) r r r � get_reseller) s zGeneralPanel.get_reseller)F)NNFT)rE )F)F)F)F)F)F)F)F)FN)FN)F)r r r r( r* r+ r1 � staticmethodr4 r= rD rG rH rJ rL rM rN rO rP rQ rS rU rV 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/r> NzUnable to read file 'z
', error is: z-. It looks like you use not an UTF-8 charset.z
Please read the instruction: )�file) r? r@ �UnicodeDecodeError�
startswith�USERDIRr8 r- r. r4 �exit�BAD_CODING_ERROR_CODE)r' rA Z
ENCODING_LINKrB rC r< �messager r r rD 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_USERSrD �strip�match�group)r' �result�pattern� file_content�line�strippedrk 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 )rj �split)�.0�lr r r �
<listcomp>~ s z)Cpanel._get_resellers.<locals>.<listcomp>)rf rg rh �RESELLERS_PATHrD )r' � resellersro 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 �=rt )rf rg rh rD �listr rj ru )r' rg rI ro rm 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� rt ) r= rf rg �joinr^ rh rD r� r rj ru r{ )r' r; r) rK r, rg ro rm 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)r8 r� )r' r; r) r r r rV � 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)rf rg �isdir�
PACKAGEDIR�listdirrh 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 )NrT rt r zWarning: %s is not a file.)r{ r� ru r8 )r' r) rz r� �is_filer r r rN � 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| )rr rf rg r� r^ r� r r6 rY r r7 r8 r� r� r+ r� ) r' r) rR Zusers_resellers_dictr� r; r< rg rK r r r rS � s(
zCpanel._get_user_packagesc C sN xH| j ||d�D ]6\}}}|dkr|d kr6t||� q||krt|� qW d S )N)rR r| )rS r8 )r' rI r) rK r; � self_packageZ
self_resellerr r r rH � 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 r9 r: r r7 r8 r5 r^ rf rg rh r� )r' r; r) �users�userrg rI r r r rM 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 )NrT rt r z%s %szWarning: %s is not a file.)r{ r� ru r8 )r' r) rK rz r� r� Zres_namer r r rO s z Cpanel.list_resellers_packagesc C s6 | � � }||kr$|r td| � d S | j||d� d S )NzError: %s is not reseller)r) rK )r{ r8 rO )r' rK r) rz r r r rP 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 )NrT c 3 s$ | ] }d � � d|� ��kV qdS )rT N)r� )rv �i)�partsrz r r � <genexpr>+ s z/Cpanel.list_users_in_package.<locals>.<genexpr>rt z/getcontrolpaneluserspackages: Undefined package)r{ ru �anyr �lenrH r8 )r' rI r) r )r� rz r rJ ' s
" z Cpanel.list_users_in_packagec C s2 | � � }||kr"| j|||d� n |r.td� d S )N)r) rK z8getcontrolpaneluserspackages: Undefined reseller package)r{ rH r8 )r' rI rK r) rz r r r rL 2 s
z%Cpanel.list_users_in_reseller_packagec C s0 x*| � |�D ] \}}}td|||f � q W d S )Nz%s,%s,%s)rS r8 )r' r) r; r� rK r r r rQ 9 s zCpanel.list_users)F)F)F)FN)NFN)F)FN)F)F)F)F)r r r r^ r� ry ri r r( rD rr r{ r� r� rV r� rN rS rH rM rO rP rJ rL rQ �
__classcell__r r )r! r r 4 s,
$
r c s e Zd ZdZdZdZdZdZdZdZ dZ
e � Z � fdd �Z
d
d � Zd d
� Zdd� Zdd� Zd9dd�Zd:dd�Zd;dd�Zdd� Zdd � Zd d � Zd<d d �Zd=d!d"�Zd#d$� Zd>d%d&�Zd?d'd(�Z d@d)d*�Z dAd+d,�Z dBd-d.�Z dCd/d0�Z d1d2� Z!d3d4� Z"dDd5d6�Z#dEd7d8�Z$� Z%S )Fr 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) rX r r( r Zdetect_DA_admin�DA_ADMINr
rZ 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 )Nr> )�encodingzUnable to read file 'z '. It looks like you use not an z charset.z Error is: �.)r? r� r@ r\ r4 � Exception)r' rA rB rC r< ra r r r rD 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� r4 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��}ttdddd|d��}ttd ddd|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 y|d
�� � d �d }
W n t
k
�r@ td��Y nX |
dk�r�|�r�y|d
�� � d �d }
W n t
k
�r� td��Y nX |dk�r�d} | j � |�} | �
| ��r�d}
|} | 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\=.*$z package\=.*$zoriginal_package\=.*$r r� rt z#No "usertype" record in user configz"No "creator" record in user configz"No "package" record in user configZcustomz+No "original_package" record in user configrK r| ZDEFAULT)rf rg r� � DA_USERS_DIRrD r5 r� r rj ru �
IndexErrorr r6 �_is_reseller_limits_enabled)
r' r, Z userprofileZuserprofile_content�result_usertypeZresult_creatorZresult_packageZresult_original_package�usertyperK rI �lvp_idr; r r r r� k sp
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_enabledFc 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* rf r� r� r= r� r )r' r) rR r* r, r; rI rK r r r rS � 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{},{}) rf rg r� r� �existsrD r� �rstripr r8 r% r+ ) r' r) rK rg Zreseller_usersr, r; rI rT r r r rU � 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 r8 )r' r; r) r, rT rK r r r rV � 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� rt �adminT)
rf rg r� r� rh rD r� r rj ru )r' r� rm rg ro 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
rt z%s
N) Zclsudor� r� r� r� � ImportErrorr8 r- r_ 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_adminsC 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� )rv rp r r r rx S s z+DirectAdmin.list_admins.<locals>.<listcomp>)rf rg rh �
DA_ADMIN_LISTrD )r' r) r� ro r r r r* I s
zDirectAdmin.list_adminsc c s* | � | j�}x|D ]}|�d�V qW d S )Nr� )rD �DA_ADMINS_PACKAGESr� )r' ro rp r r r r� V s
zDirectAdmin._get_packagesc C s x| � � D ] }t|� q
W d S )N)r� r8 )r' r) rp r r r rN [ s zDirectAdmin.list_packagesc C s& x | � � D ]\}}}t||� q
W d S )N)rS r8 )r' r) r; rI rT r r r rH _ s zDirectAdmin.list_allc C s2 x,| � � D ] \}}}tdt|�||f � q
W d S )Nz%s,%s,%s)rS r8 r� )r' r) r; rI rK r r r rQ c 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 r8 )r' r; r) r, rT rI r r r rM g s z DirectAdmin.list_user_packagec C s0 x*| j dd�D ]\}}}||krt|� qW d S )Nr| )rR )rS r8 )r' rI r) r; �package_rK r r r rJ q s z!DirectAdmin.list_users_in_packagec C s0 x*| j |d�D ]\}}}||krt|� qW d S )N)rR )rS r8 )r' rI rK r) r; r� r r r rL x 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� )rv rw r r r rx � s z.DirectAdmin._get_resellers.<locals>.<listcomp>) rf rg rh �DA_RESELLERS_LISTrD �extendr* r� r� )r' Zresellers_listro 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� )rv rw r r r rx � s z6DirectAdmin._get_reseller_packages.<locals>.<listcomp>)rf rg r� r� rh rD )r' rK Z
packages_listZ package_pathro 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% )rm ZpkgZresr r r �<lambda>� � z5DirectAdmin.list_resellers_packages.<locals>.<lambda>r| )r{ r� r� � functools�reducerj r8 )r' r) Zfirst_exceptionrK �packagesr< Z
out_stringr r r rO � s
z#DirectAdmin.list_resellers_packagesc C s t d�| �|��� d S )Nr� )r8 r� r� )r' rK r) r r r rP � s z"DirectAdmin.list_reseller_packages)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( rD r� r� r� rS rU rV r� r� r� r� r* r� rN rH rQ rM rJ rL r{ r� rO rP r� r r )r! r r > sB R
r c @ s e Zd Zdd� ZdS )� ISPManagerGetPackagesExceptionc C s t �| |� d S )N)r� r( )r' ra 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 )rX r r( r ZgetCPZ
CP_VERSIONru r: �
isp_major_verZispmanager5_is_masterZisp5_is_masterr rZ 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 %sr> ZAccountr �"r| r� ZPreset� rt NzERROR: Can't read %s - %s�default)r� rf rg rh � ISP4_CONFr� r? �readru �popr: r r6 �replacerj r] r� r� �OSError�IOError�grpZgetgrnam�ISP5_GRPZgr_memr r7 �KeyError) r' �dict_uid_package�linesrp 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)r8 )�clsr; r) rK r r r rV � 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 )Nrt )r� r� �valuesr8 r� r- r_ )r' r) r� rI r< r r r rN 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� rt )r� r8 r� r� r- r_ )r' r) r� r; r< r r r rH 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 )Nrt zIncorrect user id)r: r� r8 r� r- r_ r5 )r' Zs_uidr) r; r� r< r r r rM 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 )Nrt )r� r8 r� r� r- r_ )r' rI r) r� r; r< r r r rJ 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,adminrt )r� r8 r� r� r- r_ )r' r) r� r; r< r r r rQ ( s
(zISPManager.list_users)F)F)F)F)F)F)r r r r� r r� r( r� � classmethodrV rN rH rM rJ rQ 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�Zd'dd�Zd(dd�Zd)dd�Zd*dd�Zd+d d �Zd,d d �Zd-d d!�Z� ZS ).r 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)rX r r( r rZ r& )r' )r! r r r( < s zPlesk.__init__c
C s� y dd l }W n& tk
r2 td� t�d� Y nX tj�| j�sFdS t | j��
� � � }yB|j ddt
|�ddd d
�}|�� }|�||� |�� }|�� |S |jk
r� } z*td |jd |jd f � t�d� W d d }~X Y nX d S ) Nr zVError: MySQL-client libraries is not installed. Please install MySQL-client libraries.rt FZ localhostr� ZpsaT�utf8)Z use_unicode�charsetz Error %d: %s)ZMySQLdbr� r8 r- r_ rf rg rh �
PSA_SHADOWr? r� rj Zconnectr� ZcursorZexecuteZfetchall�close�Error�args) r' Zsql�dataZmdbZpasswdZconZcur�resultsr< r r r �fetch_data_from_db@ s$ zPlesk.fetch_data_from_dbFc c s� d}|dk r&d� |�}| �||g�}n,|dk rHd� |�}| �||g�}n
| �|�}xJ|D ]B\}}}|pj| j}|dkrxdn|}| j||d�}|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) rR r, �queryrm rI rK r; r r r rS U 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� r8 )r' r) r� rp r r r rN } 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 r9 r: r� r� r� �tupler8 )r' r; r) rK Z usernamesZformat_stringsr� rm r r r rV � 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)rS r8 r% r� )r' r) r� r; rI rT r r r rH � 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)rS r8 r% r� )r' r) r� r; rI rK r r r rQ � 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 )rt r )rv rp r r r rx � s z+Plesk.list_user_package.<locals>.<listcomp>) r r9 r: r r7 r8 r5 rS r� ) r' r; r) r� �namesr< rc rm rI r r r rM � 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';
)rI r� r N)r� r� r� r8 r r6 )r' rI r) r� Zloginr r r rJ � 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'rt 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';
)rI � reseller_id)
r� r� r: r� r r6 r r7 r8 r� ) r' rI rK r) Zreseller_queryr� r� rm rp Zluidr< r r r rL � 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� rt r r� �rootz{} {}N)r� r� r� r8 r% )r' r) r� rp rI rK r r r rO s
z Plesk.list_resellers_packagesc 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;)rK r� r )rd rk r� r� r� r8 )r' rK r) r� rp r r r rP ' s z Plesk.list_reseller_packagesc C s dgS )Nr� r )r' r) r r r r* ; 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� rS rN r� �boolrV rH rQ rM rJ rL rO rP r* r� r r )r! r r 2 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 )r- r. r/ r_ )r' r) r0 r r r r1 @ s zInterWorx._warningN)r r r r1 r r r r r ? 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) r0 r r r r1 G s zUnknown._warningN)r r r r1 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 HostingNG panel)r� )r' r) r0 r r r r1 L s zHostingNG._warningN)r r r r1 r r r r r K s r )'Z
__future__r r �builtinsr rf rd r- r2 r� �typingr Zclcommonr Zclcommon.clprocr Zclcommon.utilsr Zclcommon.cpapi.pluginsr r
r r Z cldetectlibr r` r� r r �objectr r r r� r r r r r r r r r �<module> s@
| u|
Mini Shell