| Current File : //proc/self/root/opt/alt/python37/lib/python3.7/site-packages/__pycache__/lveapi.cpython-37.pyc |
B
��oeN � @ sd d dl mZ d dl mZ d dlmZmZ d dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
d dlmZ d dlmZ d dlmZmZ d d lmZ y d dlZW n ek
r� dZY nX d d
lmZ e� Zd Z d Z d
Z dZ 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)dS ) � )�print_function)�absolute_import)�map�rangeN)�Optional)�uid_max)�ProcLve)�cpapi�ClPwd)� NotSupported)� ve_configi����reseller� � c @ s e Zd ZdS )� NameMapErrorN)�__name__�
__module__� __qualname__� r r � lveapi.pyr , s r c @ s e Zd ZdS )�NameMapConfigErrorN)r r r r r r r r 0 s r c @ s e Zd ZdS )�NameMapNotInitializedN)r r r r r r r r 4 s r c @ sX e Zd ZdZefdd�Zdd� Zdd� Zdd � Zdd d
�Z dd� Z
ddd�Z dd� Z d
S )�NameMapz�
Container for backend storing resellers_name<=>resellers_id map
As backend store use ve.cfg
Usage:
>>> name_map = NameMap()
>>> name_map.link_xml_node()
>>> name_map.id_list()
[1001]
c C s || _ d | _d | _d S )N)�
_xml_tag_name� _xml_node�_reseller_id_name_map)�selfZ xml_tag_namer r r �__init__D s zNameMap.__init__c C s& x | � � D ]\}}||kr
|S q
W d S )N)�load_from_node)r �name�name_�id_r r r �get_idJ s zNameMap.get_idc C s& x | � � D ]\}}||kr
|S q
W d S )N)r )r r! r Z_idr r r �get_nameO s zNameMap.get_namec C s dd� | � � D �S )Nc S s g | ] \}}|�qS r r )�.0�_r! r r r �
<listcomp>U s z#NameMap.id_list.<locals>.<listcomp>)r )r r r r �id_listT s zNameMap.id_listNTc C s* |dkrd| _ | �|� n || _ d| _dS )aC
Initialize NameMap. If xml_node is none,
config will be loaded automatically
:param use_cache: Bool whether bypass ve.cfg xml cache
:param xml_node: !! DEPRECATED PARAM !!
this param is left only for
compatibility with our old code
N)r � _load_resellers_map_from_ve_cfgr )r �xml_node� use_cacher r r �
link_xml_nodeW s
zNameMap.link_xml_nodec C sh t � | _| j|d�\}}xF|D ]>}|�d�}t|�d��}|r |r || jkr | j�||f� q W ~~dS )zT
Fills self._reseller_id_name_map from ve.cfg file
:return:
)r* �user�idN)�listr �_try_get_xml_node� getAttribute�int�append)r r* �ve_cfgr) �el_r r! r r r r( i s
z'NameMap._load_resellers_map_from_ve_cfgc C sJ yt j|d�\}}W n$ t jk
r8 d | _td��Y nX ||�| j�fS )N)r* z-Error happened while loading data from ve.cfg)r Zget_xml_configZBadVeConfigExceptionr r �getElementsByTagNamer )r r* r3 r) r r r r/ y s zNameMap._try_get_xml_nodec c s� | j dkr | jdkr td��| j rdx@| j �| j�D ].}|�d�}t|�d��}|r2|r2||fV q2W | jr�x | jD ]\}}||fV qrW dS )z?
Obtain data from xml node as (name, id_) list
NzLName map is not initialized. Use obj.link_xml_node() to get data from configr, r- )r r r r5 r r0 r1 )r r4 r r! r r r r � s
zNameMap.load_from_node)NT)T)
r r r �__doc__�LVP_XML_TAG_NAMEr r" r# r' r+ r( r/ r r r r r r 8 s
r c @ sj e Zd ZdZdd� Zdd� Zddd�Zd d
� Zd d � Zd
d� Z dd� Z
e dd� �Z e dd� �Z
dd� ZdS )�LvpMapzl
Container for storing information about lve:lvp mapping
In which reseller container stored lve
c C s* t � | _t� | _t� | _d | _t� | _d S )N)r �name_map�dict� _id_name_map� _name_id_map�_reseller_id_map_panelr
�_pwd)r r r r r � s
zLvpMap.__init__c C s || j |< || j|< d S )N)r; r< )r r r! r r r �_add_map� s
zLvpMap._add_mapNc C s* y| j �|�jS tjk
r$ |S X d S )N)r> Zget_pw_by_name�pw_uidr
ZNoSuchUserException)r r �defaultr r r r@ � s z
LvpMap.pw_uidc C s6 | � |�}|d k r|S | jd kr*t�� | _| j�|�S )N)r@ r= r Zget_reseller_id_pairs�get)r r
�uidr r r �_get_panel_reseller_id� s
z LvpMap._get_panel_reseller_idc C sd | j �|�p| j�|�}|dk r$|S y| �|�}W n tk
rJ d}Y nX |dk r`| �||� |S )z�
Convert reseller name to an LVE id.
It supports resellers without a system account (for Plesk compatibility).
N)r9 r" r< rB rD r r? )r r rC r r r �get_reseller_id� s
zLvpMap.get_reseller_idc C sn | j �|�p| j�|�}|dk r$|S y,t�|�j}t�|�rJ| � ||� nd}W n t
k
rh d}Y nX |S )z�
Convert reseller id to reseller name
It support resellers without system account (for Plesk compatibilyty)
N) r9 r# r; rB �pwd�getpwuidZpw_namer Z is_resellerr? �KeyError)r r! r r r r �get_reseller_name� s
zLvpMap.get_reseller_namec c s� t t�� �}i }xF|D ]>}y| �|�||< W q tk
rR t�tjd| � Y qX qW xTtjdd�D ]D\}}| �|�}|dkr�t�tjd| � qf|� |d�}||fV qfW dS )z�
This method loops over all user:reseller pairs in control panel
and returns appropriate lve_id:lvp_id pairs.
THIS METHOD WON'T CHECK IF 'RESELLER LIMITS' IS ENABLED IN ve.cfg
zIReseller %s still exists in control panel, but absent in /etc/passwd file)�cploginr
)ZkeylsNzEuser %s still exists in control panel, but absent in /etc/passwd filer )
�setr � resellersrE r �syslog� LOG_WARNINGZcpinfor@ rB )r rL �
reseller_uidsr
rJ �lve_id�lvp_idr r r �
lve_lvp_pairs� s$
zLvpMap.lve_lvp_pairsc c s xt �� D ]
} | V q
W d S )N)r rL )�
reseller_namer r r rL � s zLvpMap.resellersc C sd t � }t�| �}xN|D ]F}yt�|�j}|�|� W q tk
rZ t�tj d| � Y qX qW |S )z:
Obtain from control panel resellers uids
zEuser %s still exists in control panel, but absent in /etc/passwd file)
r. r �reseller_usersrF �getpwnamr@ r2 rH rM rN )r ZuidsrT r, r! r r r rO � s
zLvpMap.reseller_uidsc C s | � |�}| �|�S )N)rI rO )r rQ rS r r r �lvp_lve_id_list s
zLvpMap.lvp_lve_id_list)N)r r r r6 r r? r@ rD rE rI rR � staticmethodrL rO rV r r r r r8 � s
r8 c @ s e Zd ZdS )�
PyLveErrorN)r r r r r r r rX s rX c @ sz e Zd ZdZedd� �Zdd� Zdd� Zdd � Ze d
d d fd
d�Z
dd� Z dd� Z dd� Z
eefdd�Zejdd� �ZdS )�PyLvezA
Wrapper for generate traceback with pretty descriptions
c C s t | t�o | tj ko | dkS )Nr )�
isinstancer1 �errnoZENOSYS)�coder r r �_code_is_error s zPyLve._code_is_errorc sB t � | jj�r6d�� fdd�t� �D ��}d�|�}nt� �}|S )Nz, c s( g | ] }|� d �sd�|t� |���qS )r% z{}={})�
startswith�format�getattr)r$ r% )�arg_varr r r& ! s z%PyLve._arg_to_str.<locals>.<listcomp>z<liblve_settings object {}>)rZ �_pylve�liblve_settings�join�dirr_ �str)r ra Zliblve_settings_attrZ arg_var_strr )ra r � _arg_to_str s
zPyLve._arg_to_strc
s� |� d� j�}|� d� j�}|||�}� �|�}|rZ� jrZt�� j� |||�}� �|�}t||j |j
d� t t
� j|��� fdd�|�� D � �d�}� jdkr�t� jjf |�� � jdkrƈ j�� |s�|r�|jf |�} t| ��|S ) NZerr_msg� ignore_errorz, c s" g | ]\}}d � |� �|���qS )z{}={})r_ rg )r$ �k�v)r r r r& 4 s z&PyLve._wrapped_fun.<locals>.<listcomp>)r\ Zfun_name�moduleZargs_� r )�pop�default_msg_templaterh r] �_retry�timeZsleep� _retry_timer: r r rd r. r rg �items�debug�print�debug_msg_templater_ � traceback� print_stackrX )
r �call�args�kwargsZ msg_templaterh r\ Zis_errorZ format_args�msgr )r r � _wrapped_fun' s*
zPyLve._wrapped_func s � �fdd�}|S )Nc s �j � f| �|�S )N)r| )ry rz )rx r r r �fun? s z PyLve._wrap_code.<locals>.funr )r rx r} r )rx r r �
_wrap_code> s zPyLve._wrap_codeTg�������?r c C sN || _ | j dkrtd�| _d| _d| _d| _|| _|| _t� | _ || _
| j� � | _ | jj
| _
| �| jj�| _| jj| _| �| jj�| _| �| jj�| _| jj| _| �| jj�| _| �| jj�| _| �| jj�| _| �| jj�| _| �| jj�| _t|d��rJ| �| jj�| _| �| jj�| _| �| jj �| _ | �| jj �| _ | �| j �| _ d S )Nr rv z/Error code {code}; {module}.{fun_name}({args_})zBDEBUG [lvectl]: call {module}.{fun_name}({args_}) with code {code}F�lve_lvp_create) rs �
__import__rv rn ru rh rb ro r �_procrq Zlve_get_api_version� api_versionZ
initializer~ Z lve_startrc Z
lve_create� lve_destroy�lve_infoZlve_set_default� lve_setupZ
lve_enter_pidZlve_enter_pid_flagsZ
lve_leave_pid�hasattrr Zlve_lvp_destroy� lve_lvp_map� lve_lvp_move�
lve_lvp_setup)r �pylveZretryZ
retry_tymers r r r r C s8
zPyLve.__init__c C s t | jd�S )zB
Check in pylve binding reseller limits supported
r )r� rb )r r r r �resellers_supportedd s zPyLve.resellers_supportedc C s( y| � |� dS tk
r" dS X dS )zD
Check if lve exists in kernel
:rtype: bool
TFN)r� �OSError)r rP r r r �
lve_existsj s
zPyLve.lve_existsc C s� |dkr| j �||�S i }xr| j�|�D ]b}yH| �|�}|||< |j|jkrr| �|�}t|j|j�|_| �||� W q( tk
r� Y q(X q(W | � | j j�}|||dd�}x@| j�|�D ]0}||kr�| �||| � q�| �|| �|�� q�W |S dS )a�
Wrapper for lve_lvp_setup.
When reseller's limits changed,
we should iterate over his user's limits
and set them again;
This behaviour is needed cause kernel
does not update users limits after
changing reseller's one
Adjust parameters for top level container.
:param int lvp_id: top level container ID, 0 by default;
:param settings: liblve_settings instance.
:return: 0 or errno value
r T)rh N)
rb r� r� Z lve_id_listr� Zls_cpu�minr� r� r~ ) r rQ ZsettingsZreal_lve_settingsrP Z
real_settingsZ
temp_settingsZ_lve_lvp_setup�resultr r r r� w s(
zPyLve.lve_lvp_setupc C sP x:t |d |�D ](}y| �|� W q tk
r6 |S X qW td||f ��dS )z�
Iter over lves and find available one.
:param int start: value to start search from; UID_MAX by default
:param int stop: max value when we will stop search
:return int: lve_id
rl z)Unable to find free lve in range (%i, %i)N)r r� r� rX )r �start�stoprP r r r �get_available_lve_id� s
zPyLve.get_available_lve_idc c s& || j | _ }z
d V W d || _ X d S )N)rh )r rh Zsaved_ignore_errorr r r �context_ignore_error� s
zPyLve.context_ignore_errorN)r r r r6 rW r] rg r| r~ r� r r� r� r� �UID_MAX�
MAX_LVE_IDr� �
contextlib�contextmanagerr� r r r r rY s !
7rY c @ sV e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dS )�LveNc C s( |pt � | _|pt� | _|p t� | _d S )N)r �procrY �pyr8 r )r r� r� r r r r r � s z Lve.__init__c c sJ t | jj�� �}x4| j�� D ]&\}}||kr8||fV q |dfV q W dS )a5
Obtain {lve id}:{lvp id} pairs iterator based on ve.cfg config
(detect enabled resellers containers)
This method (unlike LvpMap.lve_lvp_pairs) will check
if reseller is enabled in ve.cfg and return lvp_id=0
for users of reseller with disabled reseller limits
r N)rK r r9 r' rR )r Zenabled_lvp_idrP rQ r r r �lve_id_lvp_id_pairs� s
zLve.lve_id_lvp_id_pairsc C s& x | � � D ]\}}||kr
|S q
W dS )z\
Obtain lvp id based on ve.cfg config (detect enabled resellers containers)
r )r� )r rP Zlve_id_Zlvp_id_r r r �lve2lvp� s z Lve.lve2lvpc O s| t j�| j�� �r&| j�� �|d�}nd}| jj|f|�|� |dkrxy t �
|� | j� ||� W n t k
rv Y nX dS )zH
safe destroy lve container with preserving lvp mapping
r N)
�os�path�existsr� Z proc_lve_mapr rB r� r� rF rG r� rH )r rP ry rz rQ r r r r� � s
zLve.lve_destroyc C sd | j �� }xT| �� D ]H\}}|�|d�|kr| j j|d�sF| j�|� | j�||� |||< qW dS )z4
Load lve_id:lvp_id map to kmod-lve
r )rQ N)r� r r� rB Z exist_lvpr� r r� )r Z
proc_map_dictrP rQ r r r � _sync_map� s
z
Lve._sync_mapc C s& y | � � W n tk
r Y nX dS )zZ
wrapped _sync_map function for prevent error if some cpapi not supported
N)r� r )r r r r �sync_map s z Lve.sync_mapc C s" yt �� S tk
r dS X dS )z^
Check if current panel supported for reseller's limits;
:rtype: bool
FN)r Z is_reseller_limits_supportedr )r r r r �is_panel_supported s zLve.is_panel_supportedc C s t | j�� | j�� | �� f�S )zo
Check present all needed (kmod-lve, liblve, /proc/lve, panel) for manipulate resellers limits
)�allr� r� r� r� )r r r r �reseller_limit_supported s
z Lve.reseller_limit_supportedc C s t | j�� | j�� f�S )zh
Check present all needed (kmod-lve, liblve, /proc/lve) for manipulate resellers limits
)r� r� r� r� )r r r r �is_lve10! s z Lve.is_lve10)NNN) r r r r r� r� r� r� r� r� r� r� r r r r r� � s
r� )*Z
__future__r r �builtinsr r r� r� r[ rM rp rF �typingr Zclcommon.clfuncr Zclcommon.clprocr Zclcommonr r
Z clcommon.cpapi.cpapiexceptionsr r� � ImportErrorZ
clveconfigr r� r� r7 Z
LVE_NO_UBCZLVE_NO_MAXENTER� Exceptionr r r �objectr r8 rX rY r� r r r r �<module> s@
^} /