Edit File: lve_storage.cpython-37.pyc
B 5/9eIB � @ s d dl mZ d dlmZ d dlmZmZ d dlmZ d dl m Z m Z m Z m Z mZ d dlmZ d dlmZmZmZmZmZ d dlmZmZmZmZm Z m Z ed d d d g�Z ed d dd g�Z edddd ddg�Z edddg�Z!dZ"dZ#dZ$dZ%dZ&dZ'G dd � d e(�Z)d S ) � )�absolute_import)�ClPwd)� resellers� NotSupported)� iteritems)�XmlConfigReader�DEFAULT_PROVIDER� EMPTY_LIMITS�Limits�InheritedLimits)� namedtuple)�paneluserslist�panelpackagesdict� LVP_DEFAULT� LVE_DEFAULT�DEFAULT_PACKAGE)�Dict�Union�Tuple�List�Optional�Iterator�Package�name�provider�limits�Reseller�defaults�User�id�package�reseller� is_panel_userZ IneritedLimit�value�source�DEFAULTSZown�globalZbuiltinc @ s� e Zd ZdZeeddd�Zeeedd�Z d)dd�Z e dd� �Z d d � Z e d d � �Zd d� Zdd� Zdd� Zdd� Zedd� �Zedd� �Zedd� �Zedd � �Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Z d'd(� Z dS )*�LimitsDataStoragea� This class is designed to merge data from xml config & control panel and provide user-friendy interface to work with lve limits (packages, resellers, defaults) What this class can do: - process information from xml config and control panel - re-pack structures for easy data access What this class should not do: - set limits in kernel - know anything about kernel limits N)r r r )r r r c C s6 t � | _|pt� | _d | _d | _d | _d | _d | _d S )N) r Z_pwdr � _xml_source�_panel_resellers_list� _packages�_users� _resellers� _defaults)�selfZ_xml_config_reader� r/ � py/cllimitslib_v2/lve_storage.py�__init__A s zLimitsDataStorage.__init__c C s | dkrd} n | dkrd} | S )z2 Convert mem to vmem and reversed �vmem�memr/ )Z limit_namer/ r/ r0 �_limit_name_convertJ s z%LimitsDataStorage._limit_name_convertc C s8 | j dkr2y t� | _ W n tk r0 g | _ Y nX | j S )z� Get list of names of panel resellers. If cpapi panel plugin don't support that method, we return empty list N)r) r r )r. r/ r/ r0 �_get_panel_resellersX s z&LimitsDataStorage._get_panel_resellersc C s | j dk S )z@ Not activated reseller doesn't have any limits N)r )r! r/ r/ r0 �_is_active_resellerg s z%LimitsDataStorage._is_active_resellerc c s x| j �� D ] }|V q W d S )N)r( Z users_lve_ids)r. �uidr/ r/ r0 �_list_uids_from_ve_cfgo s z(LimitsDataStorage._list_uids_from_ve_cfgc C s6 || j kr| j | S t|| j| j�|�| jdd�S dS )ax This method will return User by uid. If we will pass uid of nonexistent user, method return User without any limits, with with DEFAULT_RESELLER_OBJ and with DEFAULT_PKG_OBJ. This trick is needed for validating of limits for nonexistent user in case of migrating limits and entities from another server. F)r r r r! r" N)�usersr �DEFAULT_PKG_OBJr( �get_user_limits�DEFAULT_RESELLER_OBJ)r. r7 r/ r/ r0 �get_user_by_uidt s z!LimitsDataStorage.get_user_by_uidc C sV ||f| j kr | j ||f S t|f| j kr8| j t|f S t|| j| j�|t�d�S dS )a' This method is needed because of lvectl bug "by design" our api designed only for situations when user's owner is the owner of the package that contains that user but sometimes user can be in admin's package. We will return entity Package without any limits and with DEFAULT_RESELLER_OBJ if we request name of nonexistent package. This trick is needed for validating of limits for nonexistent package in case of migrating limits and entities from another server )r r r N)�packagesr r r< r( �get_package_limits)r. r r! r/ r/ r0 � get_package_by_name_and_reseller� s z2LimitsDataStorage.get_package_by_name_and_resellerc C s || j kr| j | S | jS d S )N)r r< )r. r r/ r/ r0 �get_reseller_by_name� s z&LimitsDataStorage.get_reseller_by_namec C sJ | j d krDi | _ x"| �� D ]}| j�|�| j |< qW | j�� | j t< | j S )N)r- r5 r( �get_reseller_default_limitsr r% )r. r r/ r/ r0 r � s zLimitsDataStorage.defaultsc C sd | j d kr^i | _ xLtt� �D ]>\}}x4|D ],}t|| �|�| j�||�d�| j ||f< q*W q W | j S )N)r r r )r* r r r rA r( r? )r. r Zprovider_packagesZ package_namer/ r/ r0 r> � s zLimitsDataStorage.packagesc C s� | j d kr�i | _ xBt� D ]8\}}}t|| �|�| �||�| j�|�dd�| j |< qW x>| �� D ]2}|| j kr^t|| j| j�|�| j dd�| j |< q^W | j S )NT)r r! r r r" F)r r r r! r" ) r+ r r rA r@ r( r; r8 r: r< )r. r7 r! r r/ r/ r0 r9 � s$ zLimitsDataStorage.usersc C sL | j d krFi | _ x4| �� D ](}t|| j�|�| j�|�d�| j |< qW | j S )N)r r r )r, r5 r r( rB �get_reseller_limits)r. Z reseller_namer/ r/ r0 r � s zLimitsDataStorage.resellersc C s dd� | j �� D �S )Nc S s* i | ]"\}}|j d k r|j tkr||�qS )N)r r )�.0�keyr r/ r/ r0 � <dictcomp>� s zELimitsDataStorage.get_packages_with_custom_limits.<locals>.<dictcomp>)r> �items)r. r/ r/ r0 � get_packages_with_custom_limits� s z1LimitsDataStorage.get_packages_with_custom_limitsc C s dd� | j �� D �S )Nc S s0 i | ](\}}|j d k r|j tkr|jr||�qS )N)r r r" )rD r7 �userr/ r/ r0 rF � s zBLimitsDataStorage.get_users_with_custom_limits.<locals>.<dictcomp>)r9 rG )r. r/ r/ r0 � get_users_with_custom_limits� s z.LimitsDataStorage.get_users_with_custom_limitsc C s� |j j}| jt }|jpt�� }x�t|�D ]�\}}|dkr�t||d�}t||d�}t| � |� } |dk rzt |t d�||< q�|dk r�| � |j �s�t |t d�||< q�t | td�||< q*t |td�||< q*W tf |�S )zA Method will return inherited limits for package N)r# r$ )r r r% r r �_asdictr �getattrr r4 �InheritedLimit�RESELLER_SOURCEr6 � GLOBAL_SOURCE�BUILTIN_SOURCE� OWN_SOURCEr ) r. r �reseller_defaults�global_defaultsZpackage_own_limitsrE r# �reseller_default_value�global_default_value�builtin_default_valuer/ r/ r0 r? � s. z$LimitsDataStorage.get_package_limitsc C s� | j t }|jj }|jj}|jp"t�� }x�t|�D ]�\}}|dkr�t||d�}t||d�} t||d�} t | � |� } |dk r�t |t d�||< q�| dk r�t | t d�||< q�| dk r�| �|j�s�t | td�||< q�t | td�||< q2t |td�||< q2W tf |�S )z> Method will return inherited limits for user N)r# r$ )r r% r! r r r rK r rL r r4 rM �PACKAGE_SOURCErN r6 rO rP rQ r ) r. rI rS rR Zpackage_limitsZ user_limitsrE r# Z package_valuerT rU rV r/ r/ r0 r; ' s: z!LimitsDataStorage.get_user_limitsc C s� | � |�s�tttd td�ttd td�ttd td�ttd td�ttd td�ttd td�ttd td�ttd td�d �S |j�� }xLt|�D ]@\}}|d kr�t| �|� }t|td�||< q�t|t d�||< q�W tf |�S ) z� Method will return limits equal builtin for non-activated reseller and inherited limits for activated reseller �cpu)r# r$ �ncpu�ior3 �pmem�nproc�ep�iops)rX rY rZ r2 r[ r\ r] r^ N) r6 r rM r rP r rK r r4 rQ )r. r! Zreseller_limitsrE r# rV r/ r/ r0 rC V sJ z%LimitsDataStorage.get_reseller_limitsc C s� | j t }|tkr |�� }n"| �|�s4tf t�� �S |j �� }xLt|�D ]@\}}|dkrxt| �|� }t |t d�||< qHt |t d�||< qHW tf |�S )z� Method will return empty inherited defaults limits for non-activated reseller or inherited defaults limits for activated reseller. For global defaults method will return inherited limits N)r# r$ ) r r% rK r6 r r r r r4 rM rP rQ )r. r! rS rR rE r# rV r/ r/ r0 �get_defaults_limits� s z%LimitsDataStorage.get_defaults_limits)N) �__name__� __module__� __qualname__�__doc__r r r< r r r: r1 � staticmethodr4 r5 r6 r8 r= r@ rA �propertyr r> r9 r rH rJ r? r; rC r_ r/ r/ r/ r0 r' ! s6 (/=r' N)*Z __future__r Zclcommonr Zclcommon.cpapir r Z future.utilsr Z clveconfigr r r r r � collectionsr Z lvectllibr r r r r �typingr r r r r r r r r rM r% rQ rW rN rO rP �objectr' r/ r/ r/ r0 �<module> s$