JezK
Edit File: lvectl_packages_lib.cpython-311.pyc
� �l�M܅�� �x � d dl Z d dlmZ d dlmZ d dlmZmZ d dlm Z d dl m Z m Z d dl mZ G d� d � � ZdS ) � N)�NameMap)�reseller_lvp_list)� exec_utility�isolated_child_env)�AVAILABLE_LVE_KEYS)�(_convert_memory_value_to_adaptive_format�"check_pmem_lower_bound_from_string)�LvectlExceptionc �0 � e Zd Zd� Zdd�Zdd�Zd� Zd� ZdS ) � LveCtlPackagec � � d | _ d| _ d | _ t � � | _ | j � � � d S )Nz/usr/sbin/lvectl)� _lve_version� _UTILITY_PATH�_packages_data_listr � _name_map� link_xml_node)�selfs �$py/clpackages/lvectl_packages_lib.py�__init__zLveCtlPackage.__init__ sA � � ��� /��� $(�� � ����� ��$�$�&�&�&�&�&� Fc �h � � � j �dS ddg}|s|� d� � t � j |t � � �� � \ }}t j |� � }|dk rt |d � � �|d }g � _ t � fd �t � � D � � � � }|D ]�}|d } | d v rd } |d } i } t |d � � | d< t |d |� � | d< d|v rt |d |� � | d< t |d � � | d< d|v rt |d � � | d< dt |d � � i| d< d|v rt |d � � | d < d d v rd } n| |v } | | | | d �} � j � | � � ��dS )!z� Loads all package info from lvectl :param human_readable_numbers: PMEM and VMEM limits in KBytes, MBytes or GBytes :return: None Nzall-package-listz--jsonz--bytes��envr �msg�datac 3 �L �K � | ] }�j � |� � V � � d S )N)r �get_name)�.0�uidr s �r � <genexpr>z+LveCtlPackage._load_info.<locals>.<genexpr>= s3 �� � � � [� [��t�~� 6� 6�s� ;� ;� [� [� [� [� [� [r �RESELLER)zN/A�root�ID�SPEED�speed�VMEM�vmem�PMEM�pmem�EP�ep�NPROC�nproc�all�IO�io�IOPS�iops�provider)r" � F)�packager3 �limits� enabledLimits) r �appendr r r �json�loadsr �setr �intr )r �human_readable_numbers� lvectl_cmd�ret_code�lvectl_data_str� loaded_json� json_data�active_resellers� package_data� reseller_name� package_name�package_limitsr7 �package_result_dicts ` r � _load_infozLveCtlPackage._load_info% s: �� � � #� /� �F�(�(�3� �%� )� � � �i� (� (� (�$0��1C�Z�Ug�Ui�Ui�$j�$j�$j�!��/��j� �1�1� � �q�=�=�!�+�e�"4�5�5� 5� ��'� �#%�� � � [� [� [� [�GX�GZ�GZ� [� [� [�[�[�� %� " A� " A�L� (�� 4�M� ��'�'� &� �'��-�L� �N�&)�,�w�*?�&@�&@�N�7� #�%M� �V�$�&�&� &�N�6� "� � �%�%�)Q� ��(�*�*� *��v�&� $'�|�D�'9�#:�#:�N�4� ��,�&�&�*-�l�7�.C�*D�*D��w�'�$)�3�|�D�/A�+B�+B�#C�N�4� �� �%�%�),�\�&�-A�)B�)B��v�&��\�)�)� %� � � -�1A� A� �'� )�(�!.� #� #� � � $� +� +�,?� @� @� @� @�E" A� " Ar c �� � | � |�� � |s| j S g }| j D ]G}|d |k r9|r"|d |k r|� |� � �2|� |� � �H|S )a� Get limits for supplied reseller and package :param reseller_name_arg: Reseller name. If None - all resellers :param package_name_arg: Package name. Only if reseller name is provided. If None - all packages :param human_readable_numbers: PMEM and VMEM limits in KBytes, MBytes or GBytes :return: Data list List item format: { "package": "Basic_16", "provider": "Reseller_Peter", "limits": { "vmem": 768, "pmem": 768, "ep": 25, "nproc": 50, "iops": 100, "speed": { "all": 50 }, "io": { "all": 5 }, "inodes": { "soft": 100000, "hard": 200000 } } )r= r3 r5 )rI r r8 )r �reseller_name_arg�package_name_argr= � out_data_listrD s r �get_reseller_package_limitsz)LveCtlPackage.get_reseller_package_limitsb s� � �$ ���/E��F�F�F� � ,��+� +�� � � 4� 7� 7�L��J�'�+<�<�<�#� 7�#�I�.�2B�B�B�%�,�,�\�:�:�:�� "�(�(� �6�6�6�� � r c �$ � d|||rd|� d�ndd�d�S )Nz3%(what)s set error for package=%(package)s%(error)sz [�]r4 )�whatr5 �error)�message�context� )r �paramrF �errs r � __set_errorzLveCtlPackage.__set_error� s< � �P�$)�l�47�%?�[�#�[�[�[�[�R� A� A�B� B� Br c �P � |sdS d|v r&t |d � � }|rt |� � �d|g}|dk r|� d|� �� � |� � � D ]l\ }}|t vr$t | � ||d� � � � �|dk r|� � � r|� d�}|� d |� d |� �� � �mt | j |d t � � � � � \ }} } |d k s| r$t | � d|| � � � � �dS )a� Sets limits for supplied package :param reseller_name: Reseller name :param package_name: Package name :param limits_to_set: new LVE limits. Available keys: speed, vmem, pmem, maxEntryProcs, io, nproc, iops All other keys are ignoring. If some parameter absent on current LVE version (for example pmem on LVE4), it will be ignored too. :return: None Nr) zpackage-set-extr" z --reseller=z unknown limitr% �%z--�=Tr r �Limits) r r r8 �itemsr �_LveCtlPackage__set_error�isdigitr r r ) r rE rF � limits_to_setrR r> � limit_name� limit_valuer? � lvectl_stdout� lvectl_stderrs r �set_reseller_package_limitsz)LveCtlPackage.set_reseller_package_limits� s} � � � � �F� �]� "� "�6�}�V�7L�M�M�E�� -�%�e�,�,�,� (� �6� � �F� "� "� � � � ;�M� ;� ;� <� <� <� (5�':�':�'<�'<� ?� ?� #�J� ��!3�3�3�%�d�&6�&6�z�<�Q`�&a�&a�b�b�b��W�$�$��)<�)<�)>�)>�$�!,� /� /� /� � � � � =�:� =� =� � =� =� >� >� >� >�1=� � � �D�6H�6J�6J�2 � 2 � 2 �.��-� � �q�=�=�M�=�!�$�"2�"2�8�\�=�"Y�"Y�Z�Z� Z� �=r N)F)�__name__� __module__� __qualname__r rI rN r^ re rU r r r r sw � � � � � � '� '� '� ;A� ;A� ;A� ;A�z" � " � " � " �HB� B� B� &[� &[� &[� &[� &[r r )r9 �lveapir � lvectllibr � cllimits.libr r �cllimits.lib.arg_parsersr �cllimits.lib.utilsr r �cllimitsr r rU r r �<module>ro s� �� � � � � � � � � � � '� '� '� '� '� '� 9� 9� 9� 9� 9� 9� 9� 9� 7� 7� 7� 7� 7� 7�D� D� D� D� D� D� D� D� $� $� $� $� $� $�\[� \[� \[� \[� \[� \[� \[� \[� \[� \[r