관리-도구
편집 파일: __init__.cpython-37.pyc
B �v6d� � @ s� d dl mZ d dl mZ d dl mZ d dlZd dlZd dlZd dlZd dlm Z d dl Z d dl Z d dl Z d dlmZmZmZmZmZ d dlmZ d dlmZ d d lmZmZ d d lm Z m Z m Z m Z d d l m!Z! d d l"m#Z# d d l$m%Z%m&Z&m'Z' d dl(m)Z) d dl*Z*d dl+Z+e+j,�-� Z.dZ/G dd� de0�Z1G dd� de0�Z2G dd� de0�Z3G dd� de0�Z4G dd� de �Z5G dd� de0�Z6G d d � d e0�Z7G d d � d e7�Z8d d!� Z9d"d#� Z:e;d$�d%d&�Z<G d'd(� d(e=�Z>dS ))� )�print_function)�division)�absolute_importN)�FormattedException)�S_IRUSR�S_IWUSR�S_IRGRP�S_IROTH�ST_DEV)�map)� iteritems)�CPAPIExternalProgramFailed� EncodingError)� run_command�ExternalProgramFailed�write_file_lines�get_file_lines)�ClPwd)� defaultdict)� list_users�admin_packages�resellers_packages)�check_quota_enabled�defaultc @ s e Zd Zdd� ZdS )�NoSuchPackageExceptionc C s t �| d|f � d S )NzNo such package (%s))� Exception�__init__)�self�package� r �A/opt/alt/python37/lib/python3.7/site-packages/clquota/__init__.pyr ) s z NoSuchPackageException.__init__N)�__name__� __module__� __qualname__r r r r r r ( s r c @ s e Zd Zdd� ZdS )�NoSuchUserExceptionc C s t �| d|f � d S )NzNo such user (%s))r r )r �userr r r r . s z NoSuchUserException.__init__N)r! r" r# r r r r r r$ - s r$ c @ s e Zd Zdd� ZdS )� InsufficientPrivilegesExceptionc C s t �| d� d S )NzInsufficient privileges)r r )r r r r r 3 s z(InsufficientPrivilegesException.__init__N)r! r" r# r r r r r r& 2 s r& c @ s e Zd Zdd� ZdS )� IncorrectLimitFormatExceptionc C s t �| d|f � d S )NzIncorrect limit format (%s))r r )r �limitr r r r 8 s z&IncorrectLimitFormatException.__init__N)r! r" r# r r r r r r' 7 s r' c s( e Zd ZdZejd�� fdd� Z� ZS )�MalformedConfigExceptionz\ Raised when config files is malformed and cl-quota is not able to work with it )�errorc s$ t t| ��dtt|�d�d�� d S )Nz�cl-quota can't work because for malformed config. Please, contact CloudLinux support if you need help with resolving this issue. Details: %(error_message)s)� error_message)�message�context)�superr) r �dict�str)r r* )� __class__r r r A s z!MalformedConfigException.__init__)r! r" r# �__doc__� ConfigParser� ParsingErrorr � __classcell__r r )r1 r r) < s r) c @ s e Zd Zdd� ZdS )�GeneralExceptionc C s t �| |� d S )N)r r )r r, r r r r O s zGeneralException.__init__N)r! r" r# r r r r r r6 N s r6 c s e Zd Z� fdd�Z� ZS )�QuotaDisabledExceptionc s t t| ��d� d S )Nz&Quota disabled for all users on server)r. r7 r )r )r1 r r r T s z QuotaDisabledException.__init__)r! r" r# r r5 r r )r1 r r7 S s r7 c @ s e Zd ZdZddd�ZdS )�UserQuotaDisabledExceptionz? Raised when quota is disabled for one particular user Nc C sD d}|r|d| 7 }|r$|d| 7 }|r4|d| 7 }t �| |� d S )NzQuota disabledz for user id %sz (home directory %s)z; %s)r r )r �uid�homedirr, Zall_msgr r r r \ s z#UserQuotaDisabledException.__init__)NNN)r! r" r# r2 r r r r r r8 X s r8 c C sD | d dkr| d7 } d}| dkr$dS x|D ]}| � |�r*dS q*W dS )zd >>> _is_sys_path('/home/username') False >>> _is_sys_path('/var/davecot') True ����/) z/root/z/usr/z/var/z/sbin/z/dev/z/bin/z/srv/z/sys/z /etc/ntp/TN)� startswith)�pathZ sys_path_Zpath_r r r � _is_sys_pathg s r? c s$ t � } | �� � � fdd�� D �}|S )z) Return no system users uid list c s$ g | ] }t � | j�s� | j�qS r )r? �pw_dir�pw_uid)�.0Zusr)�pw_dictr r � <listcomp> s z#_get_users_list.<locals>.<listcomp>)r � get_user_dict)Zcl_pwdZ users_uidr )rC r �_get_users_listx s rF )�returnc C s t jt jddd�} | S )zJ Check `cl_quota_inodes_inheritance` parameter in the config file Zcl_quota_inodes_inheritanceF)Z default_val)� cldetectlibZget_boolean_paramZCL_CONFIG_FILE)�resr r r � is_quota_inheritance_enabled� s rJ c @ sF e Zd ZdZdZdZdZdZdZdZ dZ ed Z d Z d d � Z ed d� �Zdd� Zdd� Zdd� Zdd� Zdd� Zd�dd�Zdd � Zd�d d �Zd�d d!�Zd�d"d#�Zd$d%� Zd&d'� Zd(d)� Z d*d+� Z d,d-� Z d.d/� Z d0d1� Z ed2d3� �Z!d�d4d5�Z"d�d6d7�Z#d�d8d9�Z$d�d:d;�Z%d�d<d=�Z&d�d>d?�Z'd�d@dA�Z(d�dBdC�Z)d�dDdE�Z*d�dFdG�Z+dHdI� Z,dJdK� Z-dLdM� Z.d�dNdO�Z/dPdQ� Z0d�dRdS�Z1dTdU� Z2dVdW� Z3dXdY� Z4dZd[� Z5d\d]� Z6d^d_� Z7d`da� Z8dbdc� Z9ddde� Z:dfdg� Z;dhdi� Z<djdk� Z=dldm� Z>dndo� Z?dpdq� Z@drds� ZAdtdu� ZBedvdw� �ZCdxdy� ZDdzd{� ZEd|d}� ZFd~d� ZGd�d�d��ZHd�d�� ZIdS )�� QuotaWrapperz4 Base quota class for inode quotas handling z /proc/mountsz/usr/sbin/setquotaz/usr/sbin/repquotaz%/usr/bin/getcontrolpaneluserspackagesz /etc/container/cl-quotas.datz /etc/container/cl-quotas.cacheNz.lockFc C s� | � tj� | � tj� | � tj� t� | _d | _i | _i | _ i | _ i | _ i | _ i | _ | �� | _ddddddg| _t�� | _| �� | _| �| j�| _d | _d | _d S )N� bytes_used� bytes_soft� bytes_hardZ inodes_used� inodes_soft� inodes_hard)�_assert_file_existsrK � PROC_MOUNTS�REPQUOTA�SETQUOTA�list�_quota_enabled_list�_panel_present�_grace�_quota� _device_quota�_package_to_uids_map�_uid_to_packages_map�_uid_to_homedir_map�_get_saved_data_handler�_dh�_fields�os�geteuid�_euid�_load_quota_devices�_devices�_get_mountpoint_device_map�_mountpoint_device_mapped�_device_user_map�_all_package_list)r r r r r � s$ zQuotaWrapper.__init__c C s t j�| �std| f ��dS )z> Checks if command is present and exits if no zNo such command (%s)N)ra r> �exists� RuntimeError)r> r r r rQ � s z QuotaWrapper._assert_file_existsc C s | S )Nr )r r r r � __enter__� s zQuotaWrapper.__enter__c C s | j �� d S )N)�LOCK_FD�close)r �type�value� tracebackr r r �__exit__� s zQuotaWrapper.__exit__c C s | � | �|��S )z9 Returns user limits converted to tuples )�_convert_data_to_tuples�_get_current_quotas)r r9 r r r �get_user_limits� s z QuotaWrapper.get_user_limitsc C s | � | �� �S )z= Returns all user limits converted to tuples )rs rt )r r r r �get_all_users_limits� s z!QuotaWrapper.get_all_users_limitsc C s | � | j|d��S )z� :param packname: Package name for get limits. If None, returns all packages, else - only supplied package Returns package limits converted to tuples (called only from main) )�packname)rs �_get_package_quotas)r r r r r �get_package_limits� s z QuotaWrapper.get_package_limitsc C s | � | j|dd��S )zY Returns all packages limits converted to tuples (called only from main) T)rw � all_packages)rs rx )r r r r r �get_all_packages_limits� s z$QuotaWrapper.get_all_packages_limitsc C s" |dkr dS |dkrdS | � |�S )z� Preprocessed passed limit: 'default' --> '0', 'unlimited' --> -1, else calls _check_limit :param limit: :return: r �0)� unlimitedz-1z-1)� _check_limit)r r( r r r �_preprocess_limit� s z QuotaWrapper._preprocess_limitc C s� i }i }| j �d�r�t| j �d��dkr�| �� }xf| j �d�D ]V}|r^||kr^| j �d|� q>| j �d|��d�}t|�dkr�q>|d |d f||< q>W |r�| � � | � � x0| j � � D ]"}||kr�|| ||< q�d||< q�W |S )z� Retrive all available packages with their limits :param clean_dead_packages: if True - remove all nonexistent packages from cl-quotas.dat :return: Dictionary: { 'package_name': (soft_limit, hard_limit) } �packagesr �:� � )r| r| ) r_ � has_section�len�items�_get_all_package_list�options� remove_option�get�split� _write_data�_get_package_to_users_mapr[ �keys)r Zclean_dead_packages�package_limits_dictZ db_packages�list_of_packagesr �package_limitsr r r � _get_all_packages_with_limits� s( z*QuotaWrapper._get_all_packages_with_limitsTc C s0 |s|dks|dkrdS | � � |}|} | �|�}| �| �}|dk�r`|dkrRd}|dkr^d}| j�d�r�| j�dd�r�| j�dd��d�} |p�| d } |p�| d } n|p�d} |p�d} | j| | d� |�r|d kr�d}n |dkr�| }| d kr�d}n |dkr�| }| jd||d � | � � } xD| D ]<}| | \}}|dk�s@|dk�r | j ||||||d � �q W dS |�r|| j |||d �\}}nd \}}|| }}|dk�s�|dk�r�d}|dk�s�|dk�r�d}| � |�}|| }|d |k�s�|d |k�s�|�r�| � |�d }|tk�rl| j�d��rN| j�d|��rN| j�d|��d�} | d } | d } nd} d} |�p^| }|�ph| }n|}|}tjd||d |d ||| �| �|��g}t|� |�r�|dk�r�d}n|dk�r�|}|dk�r�d}n|dk�r�|}| �|||� |dk�r |dk�s |dk�r,|dk�r,| �|||� dS )z= Sets limits for users :return: None Nr| z-1�usersr� r r� )�soft�hard)z-1r} )r9 r� r� )� only_store)r| r| rP rO z-urM rN )r| z-1)� _check_adminr r_ r� � has_optionr� r� �_apply_to_all_if_not_set�_save_user_limitsr� �set_package_limit�_combine_user_limitsrt �_get_uid_to_packages_map�DEFAULT_PACKAGErK rT �_get_home_device�_fetch_homedirr )r r9 r� r� �save�syncZ force_saver� Zsoft_argZhard_arg�limitsZ cache_softZ cache_hardr� r �p_soft�p_hardZuser_combine_softZuser_combine_hard� user_soft� user_hard� saved_quotas�cachedZ user_package� soft_limit� hard_limit�cmdr r r �set_user_limit s� " (zQuotaWrapper.set_user_limitc C s� t � }|tkrdS |s*|dks&|dkr*dS | �� | �|�}| �|�}g }|r`| �|||�\}}|r�|| jdd�kr�|| �� kr�| �|||� dS y| �|� W n tk r� |r�dS Y nX | � � } | � � } �x�| � � D �]�} �xB| �|�D �]2} | | | k�rq�|� t | ��d } t�rdt� �rdt�� }|�| �}||k�rd| �|d�}|| d }|| d }| j| ||d�}|�s|q�|dk�r�d }n|d �p�d }|dk�r�d }n|d �p�d }yR| �| | d | | d f||f��s|�d | | | d | | d ||f � W q� tk �r Y q�X q�W t|�dk�r6q�d�|�d }|�rv| | jk�r`d| j| < | j| | | j| < ntjd| g}t ||d� g }q�W |�r�| �|||� dS )z? Sets limits for package :rtype : None NT)rz r rO rP )r9 r� r� z-1r| r� z%s %s %s %s %srM rN � � z-bu)�std_in) r r� r� r �!_get_saved_package_limits_if_nonerx r� �_save_package_limitsr �_get_device_user_maprt r� � get_names�int�IS_DArJ � clcontrollib� DirectAdminZ_get_user_package�_combine_package_limits�limits_are_equal�append�KeyErrorr� �joinrZ rK rT r )r r r� r� r� r� r� �clpwdr� �device_user_mapr� �devicer9 �_userZpanelZ _real_packageZ _real_quotas�datar� r� r� r r r r� { s� z QuotaWrapper.set_package_limitc C s | � � | �d�}x0|�� D ]$}|| \}}| j|||ddd� q W i | _| j�d�r�x�| j�d�D ]t}yL| �|� | j� d|�� d�}t |�dk r�wd|\}}| j |||ddd� W qd t k r� | j�d|� Y qdX qdW | �� x0| j�� D ]"}tjd|g}t|| j| d� q�W d S ) zN Read limits from file and applies them to packages and users TF)r� r� r� r� z-bu)r� N)r� r� r� r� rY r_ r� r� r� r� r� r� r� r$ r� r� rZ rK rT r ) r r� r r� r� r9 r� r� r� r r r � synchronize� s* zQuotaWrapper.synchronizec s� | � � g }| �� � xBtt� �� �td�D ]*�|��gtt� �fdd�| j�� � q*W | � d� | � t j �}t j|t jd�}|�|� | �t j � | �� dS )zF Caches the limits to non-privileged user to see them )�keyc s � � | S )Nr )�x)�current_quotas�kr r �<lambda>� � z.QuotaWrapper.save_user_cache.<locals>.<lambda>T)�quotingN)r� rt �sortedrU r� r� r� r r` �_get_global_lock�_prepare_writerrK � CACHEFILE�csv�writer� QUOTE_MINIMAL� writerows� _end_writer� _release_lock)r � cache_content� file_handler�csv_outr )r� r� r �save_user_cache� s * z QuotaWrapper.save_user_cachec C s& | j dkr dtddg��� k| _ | j S )z6 Return True if control panel present N�Unknownz/usr/bin/cldetectz--detect-cp-nameonly)rW r �rstrip)r r r r �_check_present_panel� s z!QuotaWrapper._check_present_panelc C s | j dkrt� �dS )z2 Raise exception if no admin user r N)rc r& )r r r r r� s zQuotaWrapper._check_adminc C sp | � d� tjddd�}t|_zBy|�tj� W n, tjk r\ } z t|��W dd}~X Y nX W d| � � X |S )z: Gets ConfigParser handler for future use TNF)� interpolation�strict) r� r3 r0 � optionxform�readrK �DATAFILEr4 r) r� )r �dh�er r r r^ s z$QuotaWrapper._get_saved_data_handlerc C s� | j dk r| j S i }g }xF| �� D ]:}y |�| �| �|��|f� W q" tk rZ w"Y q"X q"W x:|D ]2}|d |kr�g ||d < ||d �|d � qfW || _ | j S )zF Returns dictionary mapping devices to lists of users Nr r� )rh �_get_list_of_uidsr� r� r� r� )r Z devices_mapZdevice_user_pairsr9 �pairr r r r� s z!QuotaWrapper._get_device_user_mapc C s> |d ks|dkr|S t �d�}|�|�}|s4t|��|�d�S )Nz-1z(\d+)r� )�re�compile�searchr' �group)r r( Z limit_patternZ pattern_matchr r r r~ 0 s zQuotaWrapper._check_limitc C s | |kr dS dS )z� Compare tuples :param limits1: tuple :param limits2: tuple :return: True of tuple 1 equal to tuple 2 TFr )Zlimits1Zlimits2r r r r� 9 s z QuotaWrapper.limits_are_equalc C s� g }| � � }| �� }x�|�� D ]�}tjd|g}x�| �� D ]�}||| ks8t| �|�krXq8| j|||d�} | snq8| � || d || d f| �s8|� d||| d || d | d p�d | d p�d f � q8W t |�dkr�q d � |�d }t ||d � g }q W d S )zr Applies limits to all users if no other (user or package) ones has not been set for them z-bu)r9 r� r� rO rP z%s %s %s %s %srM rN r r| r� r� )r� N)r� rt r� rK rT r� r� r� �_combine_default_limitsr� r� r� r� r ) r r� r� r� r� r� r� r� r9 r� r r r r� E s. z%QuotaWrapper._apply_to_all_if_not_setc C sx | j |||d�\}}d\}}x&| �|�D ]}| j|||d�\}}q(W | jd||d�\} } | dkrdd} | dkrpd} | | fS )zc Determines user limits taking into account saved package and default ones )r9 r� r� )NN)r r� r� r| N)� _get_saved_user_limits_if_noner� �*_get_saved_package_limits_if_none_or_unlim�'_get_saved_user_limits_if_none_or_unlim) r r9 r� r� Zu_softZu_hardr� r� r Zd_softZd_hardr r r r� c s z!QuotaWrapper._combine_user_limitsc C sD | j |d�\}}|r |r dS |dkr(|}|dkr4|}| jd||d�S )zc Determines package limits taking into account saved user and default ones )r9 r Nr| )r9 r� r� )r� r� )r r9 r� r� �t_soft�t_hardr r r r� u s z$QuotaWrapper._combine_package_limitsc C sH | j |d�\}}|dk r |}|dk r(|}|dkr4d}|dkr@d}||fS )z� Determines default limits taking into account saved user and package ones :param soft: soft limit from uid=0 :param hard: hard limit from uid=0 )r9 Nz-1r| )r� )r r9 r� r� r� r� r r r r� � s z$QuotaWrapper._combine_default_limitsc C s~ yD| j �d|��d�\}}|dkr.|dkr.|}|dkrB|dkrB|}W n tjtjfk r` Y nX | �|�}| �|�}||fS )z? Retrives saved user limits if none has passed r� r� Nr| )r_ r� r� r3 �NoSectionError� NoOptionErrorr~ )r r9 r� r� r� r� r r r r� � s z+QuotaWrapper._get_saved_user_limits_if_nonec C s� yl| j �d|��d�\}}|dkr&d}|dkr2d}|dksB|dkrN|dkrN|}|dks^|dkrj|dkrj|}W n tjtjfk r� Y nX | �|�}| �|�}||fS )zN Applies saved user limits if none or unlimit has been passed r� r� z-1r| N)r_ r� r� r3 r� r� r~ )r r9 r� r� r� r� r r r r� � s z4QuotaWrapper._get_saved_user_limits_if_none_or_unlimc C s� |t krjyD| j�d|��d�\}}|dkr6|dkr6|}|dkrJ|dkrJ|}W n tjtjfk rh Y nX | �|�}| �|�}||fS )zA Applies saved package limits if none has passed r� r� Nr| )r� r_ r� r� r3 r� r� r~ )r r r� r� � pack_soft� pack_hardr r r r� � s z.QuotaWrapper._get_saved_package_limits_if_nonec C s� |t krzyT| j�d|��d�\}}|dks2|dkr>|dkr>|}|dksN|dkrZ|dkrZ|}W n tjtjfk rx Y nX | �|�}| �|�}||fS )zQ Applies saved package limits if none or unlimit has been passed r� r� Nr| )r� r_ r� r� r3 r� r� r~ )r r r� r� r� r� r r r r� � s z7QuotaWrapper._get_saved_package_limits_if_none_or_unlimc C sb | j dkr| �� S | js"| �� | _|r\y|| j| iS tk rZ | �|� t|��Y nX | jS )z� Retrieves current quotas. If euid == 0, use data from repquota utility, else from /etc/container/cl-quotas.cache file r )rc �_load_user_cacherY �_load_current_quotasr� �_check_if_quota_enabledr$ )r r9 r r r rt � s z QuotaWrapper._get_current_quotasc C s i }|r| � � }n| �� }x�|D ]�}dg}yx|dkrN| j�dd��d�\}}n| j�d|��d�\}}| �|�}| �|�}|dkr�d}|dkr�d}|�||g� W n* tjtj fk r� |�ddg� Y nX |� | � ||�� q W |�ry ||| iS t k �r t |��Y nX |S )aa Prepares package limits data for outputting (call only from get_package_limits/get_all_packages_limits - main) :param packname: Package name for get limits. If present, function returns limits only for this package, else - all packages :param all_packages: If False reads only used and admin's packages, True - all packages (including reseller packages without users) :return Dictionary of package limits: {package_name: {'inodes_used': 'xxx', 'inodes_soft': 'yyy', 'inodes_hard': 'zzz'} �-r r� r| r� r� z-1)r� �_get_list_of_packagesr_ r� r� r~ �extendr3 r� r� �update� _populater� r ) r rw rz �qr� r �valuesr� r� r r r rx � s4 z QuotaWrapper._get_package_quotasc s, |t tt� fdd�t| jdd � ����iS )Nc s | d � | d fS )Nr� r r )r� )r� r r r� r� z(QuotaWrapper._populate.<locals>.<lambda>� )r/ rU r � enumerater` )r �itemr� r )r� r r� s zQuotaWrapper._populatec C s t | �� �� �S )N)rU r� r� )r r r r r� " s z"QuotaWrapper._get_list_of_packagesc C s t | �� �� �S )N)rU r� r� )r r r r r� % s z QuotaWrapper._get_list_of_uidsc C sB | j s| �� | _ |r<y | j | S tk r: t|��Y nX | j S )N)r[ �_load_package_uids_datar� r )r r r r r r� ( s z&QuotaWrapper._get_package_to_users_mapc C sF || j krd S | �|�}t|d�}|r6t|||d��n | j �|� d S )N)r> )r9 r: r, )rV r� r r8 r� )r r9 Zhome_dirZquota_disabled_messager r r r� 2 s z$QuotaWrapper._check_if_quota_enabledc C sB | j s| �� | _|r<y | j | S tk r: t|��Y nX | j S )N)r\ r r[ r� r$ )r r9 r r r r� < s z%QuotaWrapper._get_uid_to_packages_mapc C s y t � }W n8 tttfk rB } ztdt|� ��W dd}~X Y nX tt�}tt�| _xBt |�D ]6\}}t|�}|d }|| � |� | j| � |� q`W yt dd�}W n6 ttfk r� } ztdt|� ��W dd}~X Y nX x |D ]}||kr�q�g ||< q�W t |k�rg |t < |S )z� Retrieve package-uids map from cpapi. Only for custom panels. See LU-610 for details :return: Dictionary with data. Example: {'default': ['1038', '1043', '1046'], 'res1_pack1': ['1044'], 'pack1': ['1042']} z%s. Can not get usersNr T)� raise_excz %s. Can not get admin packages) r �OSErrorr r r r0 r rU r\ r r� r r� ) r Zusers_packagesr� Zpackages_usersr9 Zuid_dataZs_uidr Z admin_pkgsr r r � _get_packages_uids_from_cpapiF s, " " z*QuotaWrapper._get_packages_uids_from_cpapic C sR i }| j dkr|S | �� sJtttt� ��|t< tdd� |t D ��| _|S | � � S )z� Gets map of packages and users :rtype dict :return Dictionary with data. Example: {'default': ['1038', '1043', '1046'], 'res1_pack1': ['1044'], 'pack1': ['1042']} r c s s | ]}|t fV qd S )N)r� )rB �ir r r � <genexpr>y s z7QuotaWrapper._load_package_uids_data.<locals>.<genexpr>) rc r� rU r r0 rF r� r/ r\ r )r r� r r r r k s z$QuotaWrapper._load_package_uids_datac C s | j r | j S y.g | _ tdd�}x|D ]}| j �|� q$W W n6 ttfk rp } ztdt|� ��W dd}~X Y nX y:tdd�}x*|�� D ] }x|D ]}| j �|� q�W q�W W n6 ttfk r� } ztdt|� ��W dd}~X Y nX t | j kr�| j �t � | j S )zm Retrives all (root and resellers) panel package list :return: List of package names T)r z %s. Can not get admin packagesNz!%s. Can not get reseller packages) ri r r� r r r r0 r r� r� )r Zlist_admin_packagesr r� Zdict_resellers_packages� packages_listr r r r� ~ s& " " z"QuotaWrapper._get_all_package_listc s\ xV� � � D ]J�y.tt� �fdd�| jdd� ��}|� �< W q tk rR w Y q X q W � S )zI Convert dict to tuples for passing to printing routines c s | � � | fS )Nr )r� )r� r� r r r� � r� z6QuotaWrapper._convert_data_to_tuples.<locals>.<lambda>r� N)r� �tupler r` r� )r r� �entryr )r� r� r rs � s z$QuotaWrapper._convert_data_to_tuplesc st i }d}� j }tjdg}t|�}t�dtj�}�x0|�� D �]"}|�d��r|sRq:|� � }t |�dkrp� � |�}|d dd� } | dkr�q:yZ||kr�� � |�}||kr� � � �| �|�r�ttt� fd d �t|d d� ����|| < W n tttfk �r w:Y nX q:|�d ��r*||�d �d� �� }q:d|kr:|�|�} | r:� j�tttdd � | ���� q:W |�� �� � |S )zZ Gets current quota settings from repqouta utility for further processing Nz-unaz2(block|inode)\sgrace\stime:?\s(\d[\w:]+)(?:;|$|\s)�#� r r� r| c s � j | d | d fS )Nr r� )r` )r� )r r r r� � r� z3QuotaWrapper._load_current_quotas.<locals>.<lambda>r� z***z/devZgracec S s | d � � | d fS )Nr r� )�lower)r� r r r r� � r� ) re rK rS r r� r� � IGNORECASE� splitlinesr= r� r� �#_remove_redundant_fields_from_input�_find_unknown_device�_is_home_devicer� r/ rU r r� r� � IndexErrorr$ �find�strip�findallrX r� � _add_default) r r� r� �devicesr� r� Zgrace_regex_pattern�line�partsr9 �foundr )r r r� � s> . "z!QuotaWrapper._load_current_quotasc s: |d d� }t �d�� |�� fdd�|dd � D �� |S )Nr� z^\d+$c s g | ]}� � |�r|�qS r )r� )rB �field)�is_digit_patternr r rD � s zDQuotaWrapper._remove_redundant_fields_from_input.<locals>.<listcomp>)r� r� r� )r r Zstripped_partsr )r r r � s z0QuotaWrapper._remove_redundant_fields_from_inputc C sX t | j�dkr,| j�tdd� t�� D ��� y | j| S tk rR t|��Y nX d S )Nr c s s | ]}t |j�|jfV qd S )N)r0 rA r@ )rB r r r r r � s z.QuotaWrapper._fetch_homedir.<locals>.<genexpr>)r� r] r� r/ �pwd�getpwallr� r$ )r r9 r r r r� � s zQuotaWrapper._fetch_homedirc C s� i }t tj�}t�d�}x�|D ]�}|�d�r.q |�|�}|d }d|d i}xP|D ]H}|�d�sj|�d�r~|�d�d |d < qR|�d �rR|�d�d |d < qRW ||kr�|| �|� q |g||< q W |�� t |�dkr�t � �|S ) aj Gets mounted filesystems list and picks ones with quota on Example of returned data structure: {'/dev/mapper/VolGroup-lv_root': [ {'mountpoint': '/', 'quota_file': 'quota.user', 'quota_type': 'vfsv0'}, {'mountpoint': '/var', 'quota_file': 'quota.user', 'quota_type': 'vfsv0'} ], '/dev/mapper/VolGroup-lv_root2': [ {'mountpoint': '/', 'quota_file': 'quota.user', 'quota_type': 'vfsv0'}, {'mountpoint': '/var', 'quota_file': 'quota.user', 'quota_type': 'vfsv0'} ] } z |,zrootfs /r � mountpointr� z usrquota=zusruota=�=Z quota_filezjqfmt=Z quota_type) �openrK rR r� r� r= r� r� rn r� r7 ) r r Zproc_mounts_streamZ split_pattr � line_splitedr� �mountpoint_dataZline_splited_elementr r r rd � s* z QuotaWrapper._load_quota_devicesc s� i }zVy$�� � ttj�}tj|dd�}W n, ttfk rV t�j �t � �j d�iS X W d�� � X t�j �}xN|D ]F� � d |krv|�� d t tt� �fdd�tt�j �����i� P qvW |s�t�j �t � �j d�iS |S )zJ For non-privileged user we outputting data from the file �,)� delimiterr| Nr c s �j | � | d fS )Nr� )r` )r� )�rowr r r r� s z/QuotaWrapper._load_user_cache.<locals>.<lambda>)r� r rK r� r� �readerr �IOErrorr0 rc r/ �fromkeysr` r� r� rU r �ranger� )r r� �foZcvs_inr9 r )r% r r r� s$ z QuotaWrapper._load_user_cachec C sX dd� }g }x8t |�D ],\}}x"|D ]}|d }|�||f� q$W qW |j|dd� |S )z� return list tuple ('mountpoin tpath', 'device') reverse sorted by deep mountpoint path [('/mountpoint_path/path', '/device'), ('/mountpoint_path', '/device')] c S s$ | d dkrd}n| d � d�}|S )Nr r< )�count)Zdevice_mountpointZ deep_pathr r r �sort_by_deep_path$ s zBQuotaWrapper._get_mountpoint_device_map.<locals>.sort_by_deep_pathr T)r� �reverse)r r� �sort)r r r, Zmountpoint_device_mapr� Zmountpoint_data_listr" Zmountpoint_pathr r r rf s z'QuotaWrapper._get_mountpoint_device_mapc C sB dd� }|t j�|��}x$| jD ]\}}|�||��r |S q W dS )z3 Returns device user homedir is on c S s | r| d dkr| d7 } | S )Nr; r< r )r> r r r � _add_slash6 s z1QuotaWrapper._get_home_device.<locals>._add_slashN)ra r> �dirnamerg r= )r �homer/ r0 Zmounpoint_pathr� r r r r� 2 s z QuotaWrapper._get_home_devicec C s | � |�|kS )z; Checks if a device is user homedir device )r� )r r1 r� r r r r ? s z QuotaWrapper._is_home_devicec C s~ ydt �|�t }t �|�t �|�f}x<| j�� D ].}t �|�t }|t �|�t �|�fkr0|S q0W W n tk rx |S X d S )N)ra �statr �major�minorre r� r )r r� �devZ dev_to_findZcurrent_devicer r r r E s z!QuotaWrapper._find_unknown_devicec s� ddddg� yB| j �dd��d�\}}|dkr2d}|dkr>d}� �||g� W n* tjtjfk rx � �ddg� Y nX dttt � fdd�t | j ����iS )zw Insert 'default' quota. Calls only from _load_current_quotas, after parsing repquota's output r� r| r� r� z-1c s | d � | d fS )Nr� r r )r� )r� r r r� ` r� z+QuotaWrapper._add_default.<locals>.<lambda>) r_ r� r� r� r3 r� r� r/ rU r r� r` )r r� r� r )r� r r P s zQuotaWrapper._add_defaultc C s� |dkr d}|dkrd}| � |||�\}}|dks:|dkrf|dksV|dkrf| j�d�rf| j�d|� n0| j�d�s~| j�d� | j�d|d||f � | �� dS )z# Saves user limits Nr| r� z%s:%s)r� r_ r� r� � add_section�setr� )r r9 r� r� r r r r� b s z QuotaWrapper._save_user_limitsc C s� |dkr d}|dkrd}| � |||�\}}|dks:|dkrf|dksV|dkrf| j�d�rf| j�d|� n0| j�d�s~| j�d� | j�d|d||f � | �� | �|� dS )z& Saves package limits Nr| r� z%s:%s)r� r_ r� r� r6 r7 r� � _copy_package_limits_to_cpanel)r r r� r� r r r r� t s z!QuotaWrapper._save_package_limitsc C s t �� s dS d|� �}t|�}t|�dkr.dS | �|�\}}|dkrP|dkrPdS | j|dd�| }x$|�� D ]\}}|dkrld||< qlW | �|�dd �|�d d �f|d |d f�r�dS x6d D ].} d t | � d t || � d } |� | � q�W t ||d� dS )zV Copy package quota limits from cl-quotas.dat to cpanel packages data Nz/var/cpanel/packages/r T)rz r� z-1rO r| rP )rO rP �lve_r r� �w) rH Z is_cpanelr r� �_parse_cpanel_package_datarx r� r� r� r0 r� r ) r r � package_path�cpanel_package_lines�old_cpanel_dataZmodified_cpanel_linesZ quotas_datar� rp Z limit_typeZ limit_stringr r r r8 � s* z+QuotaWrapper._copy_package_limits_to_cpanelc C s� | dd� }i }xx| D ]p}|� d�rp|�� �d�}|d �dd��� }|d dkr^|d ||< |dkrp|�|� |� d �rd |krd S qW ||fS ) z� Process cpanel_package_lines - get values of all old lve_ limits and remove lines with limits that would be changed Nr9 r r r� r� �DEFAULT)rO rP Z_PACKAGE_EXTENSIONS�lve)NN)r= r r� �replace�remove)r= Z cpanel_package_lines_modifiedr> r � line_partsZ limit_namer r r r; � s z'QuotaWrapper._parse_cpanel_package_datac C sx |dkr<|dkr<y| j �||� W ql tjk r8 Y qlX n0| j �|�sT| j �|� | j �||d||f � | �� dS )z& Saves data to a file r| z%s:%sN)r_ r� r3 r� r� r6 r7 r� )r r� r� r� Z item_typer r r � _save_data� s zQuotaWrapper._save_datac C sp t j�|�}y(tjd|d�\}}t �|d�}|| _|S ttfk rj t j� |�r^t � |� t d��Y nX dS )zH Open temporary file for writing and return file object Zlvetmp_)�prefix�dirr: zCould not save dataN) ra r> r0 �tempfile�mkstemp�fdopen�_tmpr' r rj �unlinkr6 )r �filepathr> �fd� temp_pathr� r r r r� � s z QuotaWrapper._prepare_writerc C sH y.t tB tB tB }t�| j|� t�||� W n tk rB Y nX dS )z0 Routines after writing to file N) r r r r ra �renamerJ �chmodr )r r> �maskr r r r� � s zQuotaWrapper._end_writerc C s: | � d� | �tj�}| j�|� | �tj� | �� dS )z7 Actual place of saving data to a file TN)r� r� rK r� r_ �writer� r� )r r� r r r r� � s zQuotaWrapper._write_datac C s� |r dt _t jd kr~ytt jd�t _W n ttfk rF td��Y nX yt� t j� � tj � W n tk r| td��Y nX d S )NT�rz Can't open lock file for readingzCan't get lock) rK � LOCK_WRITErm r � LOCK_FILEr' r r6 �fcntl�flock�fileno�LOCK_EX)r rR r r r r� � s z QuotaWrapper._get_global_lockc C s$ t js t jd k r t j�� d t _d S )N)rK rT rm rn )r r r r r� � s zQuotaWrapper._release_lock)N)F)NNTTFF)NNTTF)NN)NN)NN)NN)NN)NN)NN)NN)N)NF)N)N)F)Jr! r" r# r2 rR rT rS ZGETPACKSr� r� rm rU rT r � staticmethodrQ rl rr ru rv ry r{ r r� r� r� r� r� r� r� r^ r� r~ r� r� r� r� r� r� r� r� r� rt rx r� r� r� r� r� r� r r r� rs r� r r� rd r� rf r� r r r r� r� r8 r; rD r� r� r� r� r� r r r r rK � s� " r S - % $ ' rK )?� __future__r r r ra r� r r� �clcommonr � configparserr3 rG rV r2 r r r r r �builtinsr � future.utilsr Z clcommon.cpapi.cpapiexceptionsr r �clcommon.utilsr r r r �clcommon.clpwdr � collectionsr �clcommon.cpapir r r Zclcommon.clquotar rH r� �detectZis_dar� r� r r r$ r& r' r) r6 r7 r8 r? rF �boolrJ �objectrK r r r r �<module> sF