Edit File: lveapi.cpython-37.pyc
B 5/9eN � @ 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@ ^} /