관리-도구
편집 파일: ve_config_reader.cpython-311.opt-1.pyc
� �EBg� � �� � d dl mZ d dlmZmZmZmZmZ d dlm Z d dl m Z ddl m Z edg d�� � Z ed g d�� � Z ed d d d d d d d � � Zd Z G d � d � � Zd S )� )� namedtuple)�Tuple�Optional�Type�List�Iterator)�etree)�ClPwd� )�get_xml_config_etree�Limits��cpu�ncpu�io�vmem�pmem�nproc�ep�iops�InheritedLimitsN�rootc � � e Zd ZdZdd�Zd� Zd� Zeefd�� � Z e d� � � Z d� Z e fd �Zd � Zd � Zd � Zd � Zd� Zd� Zd� 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 � � t � � | _ |�t � � | _ n|| _ t | � � � � � | _ t | � � � � � | _ t | � � � � � | _ t | � � � � � | _ d S �N) r �clpwdr � _xml_config�dict� _users_limits� _users_map�_packages_limits� _packages_map�_resellers_limits�_resellers_limits_map�_resellers_defaults�_resellers_defaults_map)�self�_xml_config_etrees �!py/clveconfig/ve_config_reader.py�__init__zXmlConfigReader.__init__* s� � ��W�W�� � � $� 3� 5� 5�D� � � 0�D� � �t� 1� 1� 3� 3�4�4��� !�$�"7�"7�"9�"9� :� :���%)�$�*@�*@�*B�*B�%C�%C��"�'+�D�,D�,D�,F�,F�'G�'G��$�$�$� c �Z � t | j � d� � j � � S )N�version)�intr �find�text�r( s r* r. zXmlConfigReader.version8 s$ � ��4�#�(�(��3�3�8�9�9�9r, c �\ � | � | j � d� � � � S )N�defaults)�_parse_limits_sectionr r0 r2 s r* r4 zXmlConfigReader.defaults< s. � ��)�)� � � !� !�*� -� -�/� /� /r, c �x � | � |� � \ } ||� � S # t t f$ r Y d S w xY wr )�xpath�AttributeError� ValueError)�elementr7 �type_�values r* � _get_attribute_by_xpath_or_nonez/XmlConfigReader._get_attribute_by_xpath_or_noneA sO � � � �]�]�5�)�)�F�E��5��<�<� �� � �+� � � ��4�4� ���s �!$ �9�9c �| � t | � |dt � � | � |d� � | � |d� � | � |d� � | � |d� � | � |d� � | � |d� � | � |d� � � � � S ) Nz cpu/@limitz ncpu/@limitz io/@limitz mem/@limitz pmem/@limitz nproc/@limitzother/@maxentryprocsz iops/@limitr )r r= �str)�clsr: s r* r5 z%XmlConfigReader._parse_limits_sectionJ s� � � ��3�3�G�\�3�O�O��4�4�W�m�L�L��2�2�7�K�H�H��4�4�W�l�K�K��4�4�W�m�L�L��5�5�g�~�N�N��2�2�7�<R�S�S��4�4�W�m�L�L� � � � r, c �6 � | j � |� � S r )r! �get)r( �user_ids r* �get_user_limitsz XmlConfigReader.get_user_limitsX s � ���"�"�7�+�+�+r, c �: � | j � ||f� � S r )r# rB )r( �name�owners r* �get_package_limitsz"XmlConfigReader.get_package_limits\ s � ��!�%�%�t�U�m�4�4�4r, c �6 � | j � |� � S r )r' rB �r( rF s r* �get_reseller_default_limitsz+XmlConfigReader.get_reseller_default_limits` s � ��+�/�/��5�5�5r, c �6 � | j � |� � S r )r% rB rJ s r* �get_reseller_limitsz#XmlConfigReader.get_reseller_limitsd s � ��)�-�-�d�3�3�3r, c � � g }| j � d� � D ]�}|� d� � �W |� | j � |� d� � � � � � �Y# | j j $ r Y �jw xY w|� t |� d� � � � � � ��|S �N�lve�id�user)r �findallrB �appendr �get_uid�NoSuchUserExceptionr/ )r( �resultrP s r* � users_lve_idsz XmlConfigReader.users_lve_idsh s� � � ���#�+�+�E�2�2� 2� 2�C��w�w�t�}�}�$���M�M�$�*�"4�"4�S�W�W�V�_�_�"E�"E�F�F�F�F���z�5� � � � �D����� � � �c�#�'�'�$�-�-� 0� 0�1�1�1�1�� s �A A6�6B�Bc # � K � | j � d� � D ]p}|� d� � }|�-| j � |� d� � � � nt |� � }| � |� � }||fV � �qd S rO )r rS rB r rU r/ r5 )r( rP �uid�limitss r* r z XmlConfigReader._users_limitsy s� � � � ��#�+�+�E�2�2� � �C��'�'�$�-�-�C�9<��$�*�$�$�S�W�W�V�_�_�5�5�5�#�c�(�(�C��/�/��4�4�F��v�+� � � � � � r, c # �� K � | j � d� � }|D ]R}|� d� � }|� d� � pd }| � |� � }||pt f|fV � �Sd S )N�packagerQ �reseller)r rS rB r5 �DEFAULT_PROVIDER)r( � ve_packager] rF rG r[ s r* r" z XmlConfigReader._packages_limits� s� � � � ��%�-�-�i�8�8� �!� <� <�G��;�;�t�$�$�D��K�K� �+�+�3�t�E��/�/��8�8�F���2�"2�3�V�;� ;� ;� ;� ;� <� <r, c # � K � | j � d� � }|D ]2}|� d� � }| � |� � }||fV � �3d S )Nr^ rR )r rS rB r5 )r( � ve_resellerr^ rF r[ s r* r$ z!XmlConfigReader._resellers_limits� sl � � � ��&�.�.�z�:�:� �#� !� !�H��<�<��'�'�D��/�/��9�9�F���.� � � � � !� !r, c # �� K � | j � d� � }|D ]E}|� d� � � d� � }| � |� � }||fV � �Fd S )Nzreseller/defaultsz..rR )r rS r0 rB r5 )r( � ve_defaultsr4 rF s r* r& z#XmlConfigReader._resellers_defaults� s| � � � ��&�.�.�/B�C�C� �#� #� #�H��=�=��&�&�*�*�6�2�2�D��1�1�(�;�;�H���"� "� "� "� "� #� #r, r )�__name__� __module__� __qualname__�__doc__r+ r. r4 � staticmethodr/ r= � classmethodr5 rD r_ rH rK rM rX r r" r$ r& � r, r* r r s# � � � � � � � � H� H� H� H� :� :� :�/� /� /� �>A� � � � �\�� � � � �[� �,� ,� ,� .>� 5� 5� 5� 5�6� 6� 6�4� 4� 4�� � �" � � �<� <� <�!� !� !�#� #� #� #� #r, r )� collectionsr �typingr r r r r �lxmlr �clcommon.clpwdr � ve_configr r r � EMPTY_LIMITSr_ r rk r, r* �<module>rr s �� #� "� "� "� "� "� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� � � � � � � � � � � � � +� +� +� +� +� +� ��H� Z� Z� Z� [� [�� �*� .�0l�0l�0l�m�m���v��$��d�D�$��d�� � � � �{#� {#� {#� {#� {#� {#� {#� {#� {#� {#r,