Edit File: ve_config_reader.cpython-37.opt-1.pyc
B 5/9ea � @ s� d dl mZ d dlmZmZmZmZmZ d dlm Z d dl m Z ddl m Z d dlmZ e dd d d d d dddg�Ze dd d d d d dddg�Zedddddddd�ZdZG dd� de�ZdS )� )�absolute_import)�Tuple�Optional�Type�List�Iterator)�etree)� namedtuple� )�get_xml_config_etree)�ClPwd�Limits�cpu�ncpu�io�vmem�pmem�nproc�ep�iops�InheritedLimitsN�rootc @ s� e Zd ZdZd dd�Zdd� Zdd� Zeefd d ��Z e d d � �Z d d� Z e fdd�Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � ZdS ) �XmlConfigReadera5 This class parses our xml into user-friendly primitive structures (dicts, tuples, namedtuples) This class may do: - data conversion (cpu & ncpu in config -> speed) - validation This class should NOT: - take care of user/package/reseller existence - write anything to xml Nc C st t � | _|d krt� | _n|| _dd� | �� D �| _dd� | �� D �| _dd� | �� D �| _ dd� | � � D �| _ d S )Nc S s i | ]\}}||�qS � r )�.0Zid_�limitsr r �!py/clveconfig/ve_config_reader.py� <dictcomp>1 s z,XmlConfigReader.__init__.<locals>.<dictcomp>c S s i | ]\}}||�qS r r )r �keyr r r r r 2 s c S s i | ]\}}||�qS r r )r �namer r r r r 3 s c S s i | ]\}}||�qS r r )r r �defaultsr r r r 4 s ) r �clpwdr � _xml_config� _users_limits� _users_map�_packages_limits� _packages_map�_resellers_limits�_resellers_limits_map�_resellers_defaults�_resellers_defaults_map)�selfZ_xml_config_etreer r r �__init__( s zXmlConfigReader.__init__c C s t | j�d�j�S )N�version)�intr" �find�text)r+ r r r r- 6 s zXmlConfigReader.versionc C s | � | j�d��S )Nr )�_parse_limits_sectionr" r/ )r+ r r r r : s zXmlConfigReader.defaultsc C s2 y| � |�\}||�S ttfk r, d S X d S )N)�xpath�AttributeError� ValueError)�elementr2 Ztype_�valuer r r � _get_attribute_by_xpath_or_none? s z/XmlConfigReader._get_attribute_by_xpath_or_nonec C sZ t | �|dt�| �|d�| �|d�| �|d�| �|d�| �|d�| �|d�| �|d�d �S ) Nz cpu/@limitz ncpu/@limitz io/@limitz mem/@limitz pmem/@limitz nproc/@limitzother/@maxentryprocsz iops/@limit)r r r r r r r r )r r7 �str)�clsr5 r r r r1 H s z%XmlConfigReader._parse_limits_sectionc C s | j �|�S )N)r$ �get)r+ Zuser_idr r r �get_user_limitsV s z XmlConfigReader.get_user_limitsc C s | j �||f�S )N)r& r: )r+ r �ownerr r r �get_package_limitsZ s z"XmlConfigReader.get_package_limitsc C s | j �|�S )N)r* r: )r+ r r r r �get_reseller_default_limits^ s z+XmlConfigReader.get_reseller_default_limitsc C s | j �|�S )N)r( r: )r+ r r r r �get_reseller_limitsb s z#XmlConfigReader.get_reseller_limitsc C sx g }xn| j �d�D ]^}|�d�d kr\y |�| j�|�d��� W qp | jjk rX Y qpX q|�t|�d��� qW |S )N�lve�id�user)r" �findallr: �appendr! �get_uidZNoSuchUserExceptionr. )r+ �resultr@ r r r � users_lve_idsf s z XmlConfigReader.users_lve_idsc c sZ xT| j �d�D ]D}|�d�}|d kr6| j�|�d��nt|�}| �|�}||fV qW d S )Nr@ rA rB )r" rC r: r! rE r. r1 )r+ r@ Zuidr r r r r# w s " z XmlConfigReader._users_limitsc c sR | j �d�}x@|D ]8}|�d�}|�d�p,d }| �|�}||p@tf|fV qW d S )N�packagerA �reseller)r" rC r: r1 �DEFAULT_PROVIDER)r+ Z ve_packagerH r r< r r r r r% s z XmlConfigReader._packages_limitsc c s< | j �d�}x*|D ]"}|�d�}| �|�}||fV qW d S )NrI rB )r" rC r: r1 )r+ Z ve_resellerrI r r r r r r' � s z!XmlConfigReader._resellers_limitsc c sB | j �d�}x0|D ](}|�d��d�}| �|�}||fV qW d S )Nzreseller/defaultsz..rB )r" rC r/ r: r1 )r+ Z ve_defaultsr r r r r r) � s z#XmlConfigReader._resellers_defaults)N)�__name__� __module__� __qualname__�__doc__r, r- r � staticmethodr. r7 � classmethodr1 r; rJ r= r> r? rG r# r% r' r) r r r r r s r )Z __future__r �typingr r r r r Zlxmlr � collectionsr Z ve_configr Zclcommon.clpwdr r r Z EMPTY_LIMITSrJ �objectr r r r r �<module> s