Edit File: arg_parsers.cpython-37.opt-1.pyc
B 5/9e�: � $ @ s� d dl mZ d dlmZmZ d dlmZmZ d dlmZm Z m Z m Z m Z d dl mZmZ ddlmZ dZd d d d d ddhZd d d d ddhZdhZddhZeeB eB eB ZedhB Zeej�s�dhZnedddhB ZedddhB Zddddd d d gZ d d � Z d!d"� Z ee e e e e e d#d$� d%d&�e e e d'e e d(d$� �e �d)d&�e e d'e e e �d*d$� �e e d+d$� �d,d$� �d-d&�e d'e e e �d.d$� �d/d&�e d'e e!d0d$� �d1d&�e e d'e e!d2d$� �d1d&�e d'e!�e d'e!�e d'e d)d&�e d'd3d$� d4�"ed5�d&�e d'd6d$� d4�"ed7�d&�e d'd8d$� d4�"ed9�d&�e e d'e!�e d'e!�e d'e!�e d'e!�e d'e!�e d'e!�e d'e!�e d'e e e �d:d$� �d;d&�e d'e e e �d<d$� �d=d&�e d'e!�e d>� �Z#G d?d@� d@�Z$G dAdB� dBe$�Z%d'S )C� )�absolute_import)�Tuple�Union)� DocoptExit�docopt)�Schema�And�Use�Or� SchemaError)�Feature�is_panel_feature_supported� )� VALUES_STRzcloudlinux-limitsZspeedZnprocZpmemZvmemZ maxEntryProcs�ioZiopsZinodesz mysql-cpuzmysql-io�allZcagefsz mysql-govzmysql-restrictzmysql-unrestrict-allz--mysql-restrictz--mysql-unrestrict-allz --mysql-govz --mysql-ioz --mysql-cpuz--cagefsz--inodesc C s t t| �t �dkS )z, Validate limits keys for --default r )�len�set�AVAILABLE_DEFAULTS)�keys� r � py/cllimits/lib/arg_parsers.py�_default_keys_validate# s r c C s t t| �d��tt� �dkS )z Validate limits keys �,r )r r �split�AVAILABLE_LIMITS)r r r r �_limits_keys_validate* s r c C s | S )Nr )�xr r r �<lambda>9 � r z4use --json option, other modes currently unsupported)�errorNc C s | � d�S )Nr )r )r r r r r < r z Invalid keysc C s t | �dkS )Nr )�int)r r r r r > r c C s t | ��� S )N)�str�lower)r r r r r ? r c C s | dkS )N�defaultr )r r r r r @ r z8--lve-id must be non-negative integer value or 'default'c C s t | �dkS )Nr )r! )r r r r r B r z0--reseller-id must be non-negative integer valuec C s | � � S )N)�isdigit)r r r r r D r zInvalid reseller namec C s | � � S )N)r% )r r r r r F r c C s | dkS )N)ZignoredZwatchedr )r r r r r J r z{}{}z: 'ignored', 'watched'c C s | dkS )N)ZenabledZdisabledr )r r r r r L r z: 'enabled', 'disabled'c C s | dkS )N)Z restrictedZ unrestrictedr )r r r r r N r z : 'restricted', 'unrestricted'c C s | dkS )Nr r )r r r r r X r z.--mysql-cpu must be non-negative integer valuec C s | dkS )Nr r )r r r r r \ r z---mysql-io must be non-negative integer value) r �getzdisable-reseller-limitszenable-reseller-limitsz--jsonz--human-readable-numbersz --unlimitedz --defaultz--save-all-parametersz--lve-idz --reseller-idz--reseller-namez--allz--for-resellerz --usernamez--domainz--limitsz --mysql-govz--cagefsz--mysql-restrictz--mysql-unrestrict-allz--speedz--pmemz--vmemz--nprocz--ioz--iopsz--maxEntryProcsz --mysql-cpuz --mysql-ioz--inodesz--helpc @ s� e Zd ZdZd�e�ZdZee e e e e f d�dd��Z e e e e f d�dd �Zee e e f d �d d �Ze e e ee f d �dd�Zeed�dd��Zeeee ee f d�dd��Zde e e e ee ef f d �dd�ZdS )�CloudlinuxLimitsOptsParserz� Parse arguments for cloudlinux-limits command :param argv: sys.argv :param _is_json_need: sys.argv contains --json key :return cortege: (error_flag, s_message) a� Utility to get/set any Cloudlinux limits Usage: {0} set [--json] (--lve-id <int> | --username <str>) (--unlimited) [--for-reseller <str>] {0} set [--json] (--lve-id <int> | --username <str> | --reseller-name <str>) (--default <str>) {0} set [--json] (--lve-id <int> | --username <str>) (--mysql-gov <ignored,watched> | --cagefs <enabled,disabled> | --mysql-restrict <restricted,unrestricted>) {0} set [--json] (--mysql-unrestrict-all) {0} set [--json] (--lve-id <int> | --username <str>) [--speed <str> --io <str> --nproc <str> --pmem <str> --vmem <str> --iops <str> --inodes <N,M> --maxEntryProcs <str> --mysql-cpu <int> --mysql-io <int> --save-all-parameters] {0} set [--json] (--lve-id <int> | --username <str> | --reseller-name <str>) [--speed <str> --io <str> --nproc <str> --pmem <str> --vmem <str> --iops <str> --maxEntryProcs <str>] [--default <str>] [--for-reseller <str>] {0} set [--json] (--reseller-name <str>) (--unlimited) {0} [get] [--json] [--lve-id <int> | --username <str> | --reseller-name <str> | --domain <str>] [--limits=<keys>] [--human-readable-numbers] [--for-reseller <str>] {0} disable-reseller-limits (--reseller-id <int> | --reseller-name <str>) [--json] {0} enable-reseller-limits ((--reseller-id <int> | --reseller-name <str>) | --all) [--json] {0} (-h | --help) Options: --json Return data in JSON format. --lve-id <int> LVE id. will display record only for that LVE id. --username <str> Execute command only for specific user. --reseller-name <str> Execute command only for specific reseller. --reseller-id <int> Execute command only for specific reseller. --all Execute command for all resellers --for-reseller <str> Use supplied reseller for get/set data. --domain <str> Show data only for specific domain --limits <keys> Available keys: speed,nproc,pmem,vmem,maxEntryProcs,io, iops,mysql-gov,mysql-cpu,mysql-io,cagefs,inodes --human-readable-numbers Return PMEM and VMEM limits in KBytes, MBytes or GBytes --unlimited Set all limits to unlimited. --default [limits] Reset limits to the Package defaults. List of comma-separated limits to reset them to default or "all" --mysql-gov <ignored|watched> Monitor or ignore by MySQL governor. --cagefs <enabled|disabled> Enable or disable CageFS for a user. --mysql-restrict <[un]restricted> Set user restrict status with dbctl (restricted or unrestricted). --mysql-unrestrict-all Unrestrict all restricted users with dbctl. --speed <str> Limit CPU usage for LVE | LVP. --pmem <str> Limit physical memory usage for applications inside LVE | LVP. --vmem <str> Limit virtual memory for applications inside LVE. --nproc <str> Limit number of processes for LVE | LVP. --io <str> Define io limits for LVE | LVP (KB/s). --iops <str> Limit io per second for LVE | LVP. --maxEntryProcs <str> Limit number of entry processes for LVE | LVP. --mysql-cpu <int> Set MySQL governor CPU limit (pct). --mysql-io <int> Set MySQL governor IO limit (read + write MB/s) --inodes <N,M> Set inode limits. N - soft, M - hard. --save-all-parameters Save all parameters even if they match with defaults settings. -h, --help Show this help message and exit )r r& zdisable-reseller-limitszenable-reseller-limits)�option�argv�returnc C sb d}d}x<|D ]4}|� | �r"d}q|sxtD ]}|� |�r,|}q,W qW |r^|r^dd|� d�fS dS )z� Checks options compatibility with admin options :return: Cortege (flag, message) flag: True/False - compatible/not compatible message - comment if flag is False F� TzERROR: option 'z)' is not compatible with '--for-reseller')Tr+ )� startswith�ADMIN_ONLY_OPTIONS)r( r) Z option_foundZadmin_option_name�argZ admin_optionr r r �!_is_option_compatible_with_admins� s z<CloudlinuxLimitsOptsParser._is_option_compatible_with_admins)r) r* c C s | � d|�\}}||fS )z: Performs checks before parsing arguments z--for-reseller)r/ )�selfr) Z is_compatible�messager r r � _perform_checks_before_docopt� s z8CloudlinuxLimitsOptsParser._perform_checks_before_docopt)�argsr* c C s |d r|d rdS dS )z9 Performs checks after parsing arguments z--reseller-namez--for-reseller)FzEERROR: '--reseller-name' and '--for-reseller' cannot be used together)Tr+ r )r0 r3 r r r � _perform_checks_after_docopt� s z7CloudlinuxLimitsOptsParser._perform_checks_after_docopt)r) �is_jsonr* c C s| | � |�\}}|si |fS yt| j|�}W n. tk rX d}|sP|d| j 7 }i |fS X | �|�\}}|sti |fS |dfS )zm Applies docopt to parse utility arguments. Performs checks before and after parsing z@ERROR: Invalid parameter passed or required parameter is missingz r+ )r2 r � docstringr r4 )r0 r) r5 �okr1 r3 Zs_error_stringr r r � _apply_docopt� s z(CloudlinuxLimitsOptsParser._apply_docopt)r* c C s t S )zQ Returns schema that should be used to validate parsed arguments )� global_schemar r r r � _get_schema� s z&CloudlinuxLimitsOptsParser._get_schema)�schemar3 r* c C sB y| � |�}W n* tk r8 } z i t|�fS d}~X Y nX |dfS )z@ Validate parsed arguments with provided schema Nr+ )Zvalidater r" )r; r3 �er r r �_validate_schema� s z+CloudlinuxLimitsOptsParser._validate_schemaFc sh | � ||�\� }� s d|fS t� fdd�| jD ��s<d� d< | �� }| �|� �\� }� s`d|fS d� fS )zF Parses and validates arguments passed to the utility Fc 3 s | ]}� � |�V qd S )N)r& )�.0Zcommand)r3 r r � <genexpr>� s z8CloudlinuxLimitsOptsParser.parse_args.<locals>.<genexpr>Tr& )r8 �any�commandsr: r= )r0 r) r5 r1 r; r )r3 r � parse_args� s z%CloudlinuxLimitsOptsParser.parse_argsN)F)�__name__� __module__� __qualname__�__doc__�format� PROG_NAMEr6 rA � staticmethodr" �listr �boolr/ r2 �dictr4 r8 r r: r= r rB r r r r r' h s . r' c s� e Zd Zd�e�ZdZeee e f d�dd�Z e ee e f d�dd�Z eed �d d ��Zdee ee ee e f f d �� fdd� Z� ZS )� CloudlinuxLimitsNoLveOptsParseraf Utility to get/set any Cloudlinux limits Usage: {0} set [--json] (--lve-id <int> | --username <str>) (--cagefs <enabled,disabled>) {0} [get] [--json] [--lve-id <int> | --username <str>] {0} (-h | --help) Options: --json Return data in JSON format. --lve-id <int> LVE id. will display record only for that LVE id. --username <str> Execute command only for specific user. --cagefs <enabled|disabled> Enable or disable CageFS for a user. -h, --help Show this help message and exit )r r& )r) r* c C s dS )N)Tr+ r )r0 r) r r r r2 s z=CloudlinuxLimitsNoLveOptsParser._perform_checks_before_docopt)r3 r* c C s dS )N)Tr+ r )r0 r3 r r r r4 s z<CloudlinuxLimitsNoLveOptsParser._perform_checks_after_docopt)r* c s2 dddddddh� � fdd �t j�� D �} t| �S ) Nr r& z--jsonz--lve-idz --usernamez--cagefsz--helpc s i | ]\}}|� kr||�qS r r )r> r( �value)�schema_optionsr r � <dictcomp>& s z?CloudlinuxLimitsNoLveOptsParser._get_schema.<locals>.<dictcomp>)r9 r; �itemsr )Zoptionsr )rO r r: " s z+CloudlinuxLimitsNoLveOptsParser._get_schemaF)r) r5 r* c sV t � �||�\}}|s d|fS x.tj�� D ] \}}|�||tkrDdnd � q*W d|fS )NFT)�superrB r9 r; rQ � setdefaultrK )r0 r) r5 r7 r3 �keyrN )� __class__r r rB , s z*CloudlinuxLimitsNoLveOptsParser.parse_args)F)rC rD rE rG rH r6 rA rJ r rK r" r2 rL r4 rI r r: r rB � __classcell__r r )rU r rM s rM )&Z __future__r �typingr r r r r; r r r r r Zclcommon.cpapir r Zutilsr rH ZAVAILABLE_LVE_KEYSZAVAILABLE_LVP_KEYSZAVAILABLE_QUOTA_KEYSZAVAILABLE_MYSQL_KEYSZAVAILABLE_KEYSr ZLVEr ZAVAILABLE_MYSQL_KEYS_ALLr- r r rK r! r" rG r9 r' rM r r r r �<module> s� !