Edit File: cl_quota.cpython-37.pyc
B ��dV' � @ s d dl mZ d dl mZ d dl mZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl m Z m Z d dlmZmZmZmZmZmZmZ d dlmZmZmZmZmZm Z ddd �Z dd d �Z d d � Z dd� Z dd� Z!dd� Z"dd� Z#dd� Z$e%dk�re$ej&dd� � dS ) � )�print_function)�division)�absolute_importN)�get_boolean_param�CL_CONFIG_FILE)� QuotaWrapper�NoSuchPackageException�NoSuchUserException� IncorrectLimitFormatException� InsufficientPrivilegesException�GeneralException�QuotaDisabledException)� print_text� print_csv� print_json�print_text_error�print_csv_error�print_json_errorc C sF | � � dkrdS yt�| �j}t|�S tk r@ t| ��Y nX dS )zC Checks if username is valid one and returns UID as string �default�0N)�lower�pwd�getpwnam�pw_uid�str�KeyErrorr )�username�uid� r �A/opt/alt/python37/lib/python3.7/site-packages/clquota/cl_quota.py�!check_username_and_convert_to_uid' s r c C sj t �d�}|�| �}|s t| ��|�d�} | dkr6| S yt�t| �� W n tk rd t| ��Y nX | S )z< Checks if UID is valid one and retuns it as string z(\d+)� r ) �re�compile�searchr �groupr �getpwuid�intr )r �pZpmr r r � check_uid4 s r) c C sH t �d�} | �t j� t jjdd�}t jdd�}|�|� | �|� | S )z4 Sets syslog logger and returns its handler �clquotaz/dev/log)�addressz"%(name)s:%(levelname)s %(message)s)�fmt) �logging� getLogger�setLevel�INFO�handlersZ SysLogHandler� Formatter� setFormatter� addHandler)�loggerZ syslogger� formatterr r r �setup_system_logF s r7 c C s� t d� t dtjd d � t d� t d� t d� t d� t d� t d � t d � t d � t d � t d � t d� t d� t d� t d� t d� t d� d S )N� zUsage: r z [OPTIONS]z9If no options given prints quota statistics for all userszOptions:z/ -u | --user : specifies the userz2 -U | --user-id : specifies the user IDz� -S | --soft-limit : sets the soft limit for a user. Pass 0 or 'default' to set package default limit. Pass -1 or 'unlimited' to cancel limitz� -H | --hard-limit : sets the hard limit for a user. Pass 0 or 'default' to set package default limit. Pass -1 or 'unlimited' to cancel limitzC -V | --csv : returns data as comma separated valuesz1 -J | --json : returns data as jsonzE -p | --package : specifies a package to set or get limitsz2 -P | --package-limits : prints package limitszZ -a | --all-package-limits : prints all package limits (including packages without limits)zU -Y | --sync : synchronizes packages and users limits with the databasezD -C | --cache-content : cache quota data to a file the databasezR -F | --force : save user quotas even when they are equal to defaultsz� --check : Deprecated. Check if quotas is enabled/activated/suported; if disabled show diagnostic information; using with --user or --user-id options)�print�sys�argvr r r r �usageS s$ r< c C s t � } | �� | �� dS )z, Quota synchronization :return: N)r � synchronize�save_user_cache)�qr r r � quota_synck s r@ c C s( t tddd�s$tdt � t�d� dS )zc Check autosynchronization parameter in config file and exit if it`s disabled :return: Zcl_quota_limits_autosyncT)� default_valz:cl-quota limits autosynchronization is disabled in config r N)r r r9 r: �exitr r r r �check_autosync_disabledu s rC c C s� y2t � | ddddddddd d d d d ddg�\}}W n& t jk rX t� t�d� Y nX i }d |d< d |d< d |d< d |d< d|d< d|d < d|d< d|d< �x |D �]\}}|dkr�d |d< q�|dkr�d |d < q�|d kr�||d< q�|d kr�||d< q�|d k�r ||d< q�|d k�r ||d< q�|d!k�r4||d< q�|d"k�rHd#|d < q�|d$k�r\d#|d < q�|d%k�rpd#|d < q�|d&k�r�d|d< q�|d'k�r�d|d< q�|d(k�r�d#|d< q�|d)kr�d#|d< q�W |S )*NzPaVJYCp:u:U:S:H:Fzpackage-limitszall-package-limits�csv�jsonzpackage=zuser-id=zuser=z soft-limit=z hard-limit=�sync�cache�check� profilingZforcer! �soft�hard�userzuser-id�text�formatF� force_save)z-Pz--package-limits�package)z-az--all-package-limitsz all-package)z-pz --package)z-uz--user)z-Uz --user-id)z-Sz --soft-limit)z-Hz --hard-limit)z-Yz--syncT)z-Cz--cachez--check)z-Vz--csv)z-Jz--json)z --profiling)z-Fz--force)�getoptZ GetoptErrorr< r: rB )�argv_�opts�args�config�o�ar r r � parse_args sb rX c G s� t tttttd�}t| �}�y�d|kr�d|kr�t� tj � t j �sFd S |d r�d}t d| � dd l}|�d|� dd l}|�|�}t d � t � t d � |�d ��d � t d � t � t d � |�d ��d � nt� d S t � }d|k�rL|d �sL|d �sL|d d k�r*||d |�d �dd� n ||d |�|d �dd� d S d|k�rr||d |�� dd� d S d }|d �r�t|d �}n|d �r�t|d �}|d �r�|d k �r�t|�dk�r�|�|� |d �s�|d �r>d|k�r |d d k �r |j|d |d |d d� d S |j ||d |d |d d� d S d|k�rT|� � d S d|k�rj|� � d S |�r�|d �s�||d |� |�dtd� n||d |� � dtd� W n� t!k �r� } z|d �r�t t"| �� W d d } ~ X Y n� t#t$t%t&t'fk �r: } z$|d |d | � t(� �)| � W d d } ~ X Y nD t*k �r| } z$|d |d | � t(� �+| � W d d } ~ X Y nX d S ) N)rM rD rE Zerr_textZerr_csvZerr_jsonrF rG rI zquota-profiling.logz Profiling end; result saved to r z quota_sync()z&--------------------------------------zCumulative time:Zcalls� z Total time:rP rJ rK rN )�titlez all-packagerL zuser-idrH )rP rJ rK rO )r rJ rK rO �id)rZ �sort_keyZerr_),r r r r r r rX rC �os�path�existsr �DATAFILEr9 �profile�run�pstatsZStatsZ sort_statsZ print_statsr@ �get_package_limits�get_all_packages_limitsr r) r' �_check_if_quota_enabled�set_package_limit�set_user_limitr= r> �get_user_limits�get_all_users_limitsr r r r r r r r7 �info� Exception�error) rR Zoutput_dispatcherrU Z profiler_logra rc r( r? r �er r r �main� s� " ro �__main__r! )N)N)'� __future__r r r r] r: r" rQ r r- Zlogging.handlers� cldetectlibr r r* r r r r r r r Z clquota.utilsr r r r r r r r) r7 r< r@ rC rX ro �__name__r; r r r r �<module> s, $ 6k