Current File : //proc/self/root/opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyc
�
i�B^c @` s� d d l m Z d d l m Z d d l m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l m Z m
Z
m Z m Z m Z d d l m Z d d l m Z d d l m Z d d l 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 d
l# m$ Z$ d Z% d e& f d � � YZ' d e& f d � � YZ( d e& f d � � YZ) d e& f d � � YZ* d e& f d � � YZ+ d e& f d � � YZ, d � Z- d � Z. d e/ f d � � YZ0 d S( i ( t print_function( t division( t absolute_importN( t S_IRUSRt S_IWUSRt S_IRGRPt S_IROTHt ST_DEV( t map( t iteritems( t CPAPIExternalProgramFailed( t
check_commandt run_commandt ExternalProgramFailed( t ClPwd( t defaultdict( t
list_userst admin_packagest resellers_packages( t check_quota_enabledt defaultt NoSuchPackageExceptionc B` s e Z d � Z RS( c C` s t j | d | f � d S( Ns No such package (%s)( t Exceptiont __init__( t selft package( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR # s ( t __name__t
__module__R ( ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR " s t NoSuchUserExceptionc B` s e Z d � Z RS( c C` s t j | d | f � d S( Ns No such user (%s)( R R ( R t user( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR ( s ( R R R ( ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR ' s t InsufficientPrivilegesExceptionc B` s e Z d � Z RS( c C` s t j | d � d S( Ns Insufficient privileges( R R ( R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR - s ( R R R ( ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR , s t IncorrectLimitFormatExceptionc B` s e Z d � Z RS( c C` s t j | d | f � d S( Ns Incorrect limit format (%s)( R R ( R t limit( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR 2 s ( R R R ( ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR 1 s t GeneralExceptionc B` s e Z d � Z RS( c C` s t j | | � d S( N( R R ( R t message( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR 7 s ( R R R ( ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR! 6 s t QuotaDisabledExceptionc B` s e Z d d d d � Z RS( c C` s_ d } | r | d | 7} n | r4 | d | 7} n | rK | d | 7} n t j | | � d S( Ns Quota disableds for user id %ss (home directory %s)s ; %s( R R ( R t uidt homedirR" t all_msg( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR < s N( R R t NoneR ( ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR# ; s c C` s[ | d d k r | d 7} n d
} | d k r3 t Sx! | D] } | j | � r: t Sq: Wd S( sd
>>> _is_sys_path('/home/username')
False
>>> _is_sys_path('/var/davecot')
True
i����t /s /root/s /usr/s /var/s /sbin/s /dev/s /bin/s /srv/s /sys/s /etc/ntp/N( s /root/s /usr/s /var/s /sbin/s /dev/s /bin/s /srv/s /sys/s /etc/ntp/( t Truet
startswith( t patht sys_path_t path_( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt _is_sys_pathG s
c C` sL t � } | j � } g | D]& } t | | j � s | | j ^ q } | S( s)
Return no system users uid list
( R t
get_user_dictR. t pw_dirt pw_uid( t cl_pwdt pw_dictt usrt users_uid( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt _get_users_listX s 3t QuotaWrapperc B` s� e Z d Z d Z d Z d Z d Z d Z d Z dA Z
e d Z e Z
d � Z d � Z d
� Z d � Z d � Z d
� Z dA d � Z d � Z e d � Z dA dA e e e e d � Z dA dA e e e d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z e! d � � Z" dA dA d � Z# dA dA d � Z$ dA dA d � Z% dA dA d � Z& dA dA d � Z' dA dA d � Z( dA dA d! � Z) dA dA d" � Z* dA d# � Z+ dA e d$ � Z, d% � Z- d& � Z. d' � Z/ dA d( � Z0 d) � Z1 dA d* � Z2 d+ � Z3 d, � Z4 d- � Z5 d. � Z6 d/ � Z7 d0 � Z8 d1 � Z9 d2 � Z: d3 � Z; d4 � Z< d5 � Z= d6 � Z> d7 � Z? d8 � Z@ d9 � ZA d: � ZB d; � ZC d< � ZD d= � ZE d> � ZF e d? � ZG d@ � ZH RS(B s4
Base quota class for inode quotas handling
s /proc/mountss /usr/sbin/setquotas /usr/sbin/repquotas% /usr/bin/getcontrolpaneluserspackagess /etc/container/cl-quotas.dats /etc/container/cl-quotas.caches .lockc C` s� t t j � t t j � t t j � t � | _ d | _ i | _ i | _
i | _ i | _ i | _
i | _ | j � | _ d d d d d d g | _ t j � | _ | j � | _ | j | j � | _ d | _ d | _ d S( Nt
bytes_usedt
bytes_softt
bytes_hardt inodes_usedt inodes_softt inodes_hard( R R7 t PROC_MOUNTSt REPQUOTAt SETQUOTAt listt _quota_enabled_listR' t _panel_presentt _gracet _quotat
_device_quotat _package_to_uids_mapt _uid_to_packages_mapt _uid_to_homedir_mapt _get_saved_data_handlert _dht _fieldst ost geteuidt _euidt _load_quota_devicest _devicest _get_mountpoint_device_mapt _mountpoint_device_mappedt _device_user_mapt _all_package_list( R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR s s$
c C` s | S( N( ( R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt __enter__� s c C` s | j j � d S( N( t LOCK_FDt close( R t typet valuet traceback( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt __exit__� s c C` s | j | j | � � S( s9
Returns user limits converted to tuples
( t _convert_data_to_tuplest _get_current_quotas( R R$ ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt get_user_limits� s c C` s | j | j � � S( s=
Returns all user limits converted to tuples
( R] R^ ( R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt get_all_users_limits� s c C` s | j | j d | � � S( s�
: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)
t packname( R] t _get_package_quotas( R R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt get_package_limits� s c C` s | j | j d | d t � � S( sY
Returns all packages limits converted to tuples (called only from main)
Ra t all_packages( R] Rb R) ( R R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt get_all_packages_limits� s c C` s- | d k r d S| d k r d S| j | � S( s�
Preprocessed passed limit: 'default' --> '0', 'unlimited' --> -1, else calls _check_limit
:param limit:
:return:
R t 0t unlimiteds -1( Rg s -1( t _check_limit( R R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt _preprocess_limit� s
c C` sC i } i } | j j d � r� t | j j d � � d k r� | j � } x� | j j d � D] } | r� | | k r� | j j d | � q[ n | j j d | � j d � } t | � d k r� q[ n | d | d f | | <q[ W| r� | j � q� n | j
� x> | j j � D]- } | | k r1| | | | <qd | | <qW| S( s�
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) }
t packagesi t :i i Rf ( Rf Rf (
RK t has_sectiont lent itemst _get_all_package_listt optionst
remove_optiont gett splitt _write_datat _get_package_to_users_mapRG t keys( R t clean_dead_packagest package_limits_dictt db_packagest list_of_packagesR t package_limits( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt _get_all_packages_with_limits� s( 0
c
C` s� | r# | d k s | d k r# d S| j � | } | } | j | � } | j | � } | d k r| d k rx d } n | d k r� d } n | j j d � r� | j j d d � r� | j j d d � j d � }
| p� |
d } | p� |
d } n | p� d } | p
d } | j d | d | � | r�| d k r>d } n | d k rS| } n | d k rhd } n | d k r}| } n | j d d d | d | � n | j
� | _ | j � }
xX |
D]P } |
| \ } } | d k s�| d k r�| j | | | | | d | �q�q�Wd S| rC| j
d | d | d | � \ } } n d \ } } | | } } | d k st| d k r}d } n | d k s�| d k r�d } n | j | � } | | } | d
| k s�| d | k s�| rC| j | � d } | t k r| j j d � rX| j j d | � rX| j j d | � j d � }
|
d } |
d } n d } d } | pm| } | py| } n | } | } t j d | | d | d | | | j | j | � � g } t | � | rC| d k r�d } n | d k r | } n | d k rd } n | d k r*| } n | j | | | � qCn | d k r[| d k ss| d k r�| d k r�| j | | | � n d S( s=
Sets limits for users
:return: None
NRf s -1t usersRk i i t softt hardRg R$ t
only_storeR= R< s -uR9 R: ( s -1Rg ( s -1Rg ( Rf Rf ( Rf s -1( Rf s -1( R' t _check_adminRi RK Rl t
has_optionRr Rs t _apply_to_all_if_not_sett _save_user_limitsRJ R| t set_package_limitt _combine_user_limitsR^ t _get_uid_to_packages_mapt DEFAULT_PACKAGER7 R@ t _get_home_devicet _fetch_homedirR ( R R$ R~ R t savet synct
force_saveR� t soft_argt hard_argt limitst
cache_softt
cache_hardRx R t p_softt p_hardt user_combine_softt user_combine_hardt user_softt user_hardt saved_quotast cachedt user_packaget
soft_limitt
hard_limitt cmd( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt set_user_limit� s�
'
&'
& '
0c C` s� | t k r d S| r3 | d k s/ | d k r3 d S| j � | j | � } | j | � } g } | r� | j | | | � \ } } n | r� | | j d t � k r� | | j � k r� | j | | | � d Sy | j | � Wn t k
r� | r� d Sn X| j
� } | j � } x�| j � D]�}
x| j | � D] } | | |
k rRq6n | j
d | d | d | � } | s|q6n | d k r�d }
n | d p�d }
| d k r�d } n | d p�d } ye | j | | d
| | d f |
| f � s*| j d | | | d
| | d |
| f � n Wq6t k
r>q6Xq6Wt | � d k r[q n d j | � d } | r�|
| j k r�d | j |
<n | j |
| | j |
<n" t j d |
g } t | d | �g } q W| r�| j | | | � n d S( s?
Sets limits for package
:rtype : None
NRd R$ R~ R s -1Rf i i R< R= s %s %s %s %s %sR9 R: s
t s -but std_in( R� R' R� Ri t! _get_saved_package_limits_if_noneRb R) Ru t _save_package_limitsR t _get_device_user_mapR^ Rv t _combine_package_limitst limits_are_equalt appendt KeyErrorRm t joinRF R7 R@ R ( R R R~ R R� R� R� R� t device_user_mapR� t deviceR$ t dataR� R� R� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� L sp
c C` s~ | j � | j t � } x@ | j � D]2 } | | \ } } | j | | | t d t �q& Wi | _ | j j d � r7x� | j j d � D]� } yo | j
| � | j j d | � j d � } t
| � d k r� w� n | \ } } | j | | | t d t �Wq� t k
r%| j j d | � q� Xq� W| j � n x@ | j j � D]/ } t j d | g } t | d | j | �qGWd S( sN
Read limits from file and applies them to packages and users
R� R} Rk i s -buR� N( R� R| R) Rv R� t FalseRE RK Rl Rp R� Rr Rs Rm R� R Rq Rt RF R7 R@ R ( R R{ R R~ R R$ R� R� R� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt synchronize� s*
c ` s� | j � g } | j � � xX t t � j � � d t �D]8 � | j � g t t � � f d � | j � � � q; W| j t
� | j t j
� } t j | d t j �} | j | � | j t j
� | j � d S( sF
Caches the limits to non-privileged user to see them
t keyc ` s � � | S( N( ( t x( t current_quotast k( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt <lambda>� R� t quotingN( R� R^ t sortedRA Rv t intR� R RL t _get_global_lockR) t _prepare_writerR7 t CACHEFILEt csvt writert
QUOTE_MINIMALt writerowst _end_writert
_release_lock( R t
cache_contentt file_handlert csv_out( ( R� R� sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt save_user_cache� s
%6
c C` s: | j d k r3 d t d d g � j � k | _ n | j S( s6
Return True if control panel present
t Unknowns /usr/bin/cldetects --detect-cp-nameonlyN( RC R' R t rstrip( R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt _check_present_panel� s $c C` s | j d k r t � � n d S( s2
Raise exception if no admin user
i N( RO R ( R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s c C` s@ | j t � t j � } t | _ | j t j � | j � | S( s:
Gets ConfigParser handler for future use
( R� R) t ConfigParsert strt optionxformt readR7 t DATAFILER� ( R t dh( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyRJ � s
c C` s� | j d k r | j Si } g } xT | j � D]F } y) | j | j | j | � � | f � Wq/ t k
rt q/ q/ Xq/ WxH | D]@ } | d | k r� g | | d <n | | d j | d � q� W| | _ | j S( sF
Returns dictionary mapping devices to lists of users
i i N( RT R' t _get_list_of_uidsR� R� R� R� ( R t devices_mapt device_user_pairsR$ t pair( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s )
c C` s\ | d k s | d k r | St j d � } | j | � } | sO t | � � n | j d � S( Ns -1s (\d+)i ( R' t ret compilet searchR t group( R R t
limit_patternt
pattern_match( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyRh � s c C` s | | k r t St S( s�
Compare tuples
:param limits1: tuple
:param limits2: tuple
:return: True of tuple 1 equal to tuple 2
( R) R� ( t limits1t limits2( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s c
C` sg g } | j � } | j � } xB| j � D]4} t j d | g } x� | j � D]� } | | | k sP t | j | � k r� qP n | j d | d | d | � } | s� qP n | j | | d | | d f | � sP | j
d | | | d | | d | d
pd | d pd f � qP qP Wt | � d
k r6q+ n d
j | � d
} t
| d | �g } q+ Wd S( sr
Applies limits to all users if no other (user or package) ones has
not been set for them
s -buR$ R~ R R< R= s %s %s %s %s %sR9 R: i Rf i s
R� N( R� R^ Rv R7 R@ R� R� R� t _combine_default_limitsR� R� Rm R� R (
R R~ R R� R� R� R� R� R$ R� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� s. % (
c C` s� | j d | d | d | � \ } } d \ } } x; | j | � D]* } | j d | d | d | � \ } } q@ W| j d d d | d | � \ } }
| d k r� d } n |
d k r� d }
n | |
f S( sc
Determines user limits taking into account
saved package and default ones
R$ R~ R R Rf N( NN( t _get_saved_user_limits_if_noneR' R� t* _get_saved_package_limits_if_none_or_unlimt' _get_saved_user_limits_if_none_or_unlim( R R$ R~ R t u_softt u_hardR� R� R t d_softt d_hard( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� & s $ ($ c C` sn | j d | � \ } } | r( | r( d S| d k r= | } n | d k rR | } n | j d d d | d | � S( sc
Determines package limits taking into account
saved user and default ones
R$ Rf R~ R ( N( R� R' R� ( R R$ R~ R t t_softt t_hard( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� 8 s c C` sv | j d | � \ } } | d k r- | } n | d k rB | } n | d k rW d } n | d k rl d } n | | f S( s�
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
R$ s -1Rf N( R� R' ( R R$ R~ R R� R� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� F s c C` s� yj | j j d | � j d � \ } } | d k rH | d k rH | } n | d k ri | d k ri | } n Wn t j t j f k
r� n X| j | � } | j | � } | | f S( s?
Retrives saved user limits if none has passed
R} Rk Rf N( RK Rr Rs R' R� t NoSectionErrort
NoOptionErrorRh ( R R$ R~ R R� R� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� X s $
c C` s� y� | j j d | � j d � \ } } | d k r< d } n | d k rQ d } n | d k si | d k r~ | d k r~ | } n | d k s� | d k r� | d k r� | } n Wn t j t j f k
r� n X| j | � } | j | � } | | f S( sN
Applies saved user limits if none or unlimit has been passed
R} Rk s -1Rf N( RK Rr Rs R' R� R� R� Rh ( R R$ R~ R R� R� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� h s $ $ $
c C` s� | t k r� yj | j j d | � j d � \ } } | d k rT | d k rT | } n | d k ru | d k ru | } n Wq� t j t j f k
r� q� Xn | j | � } | j | � } | | f S( sA
Applies saved package limits if none has passed
Rj Rk Rf N( R� RK Rr Rs R' R� R� R� Rh ( R R R~ R t pack_softt pack_hard( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� } s $
c C` s� | t k r� y� | j j d | � j d � \ } } | d k sK | d k r` | d k r` | } n | d k sx | d k r� | d k r� | } n Wq� t j t j f k
r� q� Xn | j | � } | j | � } | | f S( sQ
Applies saved package limits if none or unlimit has been passed
Rj Rk Rf N( R� RK Rr Rs R' R� R� R� Rh ( R R R~ R R� R� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s $$ $
c C` s� | j d k r | j � S| j s4 | j � | _ n | r� y i | j | | 6SWq� t k
r| | j | � t | � � q� Xn | j S( s�
Retrieves current quotas.
If euid == 0, use data from repquota utility, else from /etc/container/cl-quotas.cache file
i ( RO t _load_user_cacheRE t _load_current_quotasR� t _check_if_quota_enabledR ( R R$ ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR^ � s
c C` s� i } | r | j � } n | j � } x| D]} d g } y� | d k rs | j j d d � j d � \ } } n$ | j j d | � j d � \ } } | j | � } | j | � } | d k r� d } n | d k r� d } n | j | | g � Wn0 t j t j f k
r%| j d d g � n X| j
| j | | � � q. W| ry i | | | 6SWqt k
r{t
| � � qXn | S( sa
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'}
t -R R} Rf Rk Rj s -1( Ro t _get_list_of_packagesRK Rr Rs Rh t extendR� R� R� t updatet _populateR� R ( R Ra Rd t qRz R t valuesR~ R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyRb � s4
'$
c ` s6 i t t t � f d � t | j d � � � � | 6S( Nc ` s | d � | d f S( Ni i ( ( R� ( R� ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � R� i ( t dictRA R t enumerateRL ( R t itemR� ( ( R� sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s c C` s t | j � j � � S( N( RA Ru Rv ( R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s c C` s t | j � j � � S( N( RA R� Rv ( R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s c C` sZ | j s | j � | _ n | rS y | j | SWqS t k
rO t | � � qS Xn | j S( N( RG t _load_package_uids_dataR� R ( R R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyRu � s
c C` si | | j k r d S| j | � } t d | � } | rU t d | d | d | � � n | j j | � d S( NR+ R$ R% R" ( RB R� R R# R� ( R R$ t home_dirt quota_disabled_message( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s c C` sZ | j s | j � | _ n | rS y | j | SWqS t k
rO t | � � qS Xn | j S( N( RH R� RG R� R ( R R$ ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s
c C` sA y
t � } Wn/ t t f k
r> } t d t | � � � n Xt t � } t t � | _ xU t | � D]G \ } } t | � } | d } | | j | � | j | j | � qg Wy t
d t � } Wn/ t t f k
r� } t d t | � � � n Xx* | D]" } | | k rq� n g | | <q� Wt | k r=g | t <n | S( s�
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']}
s %s. Can not get usersR t raise_excs %s. Can not get admin packages(
R t OSErrorR
R
R� R RA RH R R� R R) R� ( R t users_packagest et packages_usersR$ t uid_datat s_uidR t
admin_pkgs( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt _get_packages_uids_from_cpapi s,
c C` sl i } | j d k r | S| j � sb t t t t � � � | t <t d � | t D� � | _ | S| j � S( s�
Gets map of packages and users
:rtype dict
:return Dictionary with data. Example:
{'default': ['1038', '1043', '1046'], 'res1_pack1': ['1044'], 'pack1': ['1042']}
i c s` s | ] } | t f Vq d S( N( R� ( t .0t i( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pys <genexpr>8 s (
RO R� RA R R� R6 R� R� RH R� ( R Rj ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� * s c C` sg | j r | j Sy^ g | _ t d t � } x? | D]7 } t | t � rY | j d � } n | j j | � q2 WWn/ t t f k
r� } t d t
| � � � n Xyl t d t � } xV | j � D]H } x? | D]7 } t | t � r� | j d � } n | j j | � q� Wq� WWn/ t t f k
r=} t d t
| � � � n Xt
| j k r`| j j t
� n | j S( sm
Retrives all (root and resellers) panel package list
:return: List of package names
R� s utf-8s %s. Can not get admin packagess! %s. Can not get reseller packages( RU R R) t
isinstancet unicodet encodeR� R� R
R
R� R R� R� ( R t list_admin_packagesR R� t dict_resellers_packagest
packages_list( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyRo = s.
c ` sh xa � j � D]S � y6 t t � � f d � | j d � � } | � � <Wq
t k
r_ q
q
Xq
W� S( sI
Convert dict to tuples for passing to printing routines
c ` s | � � | f S( N( ( R� ( R� R� ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� c R� i ( Rv t tupleR RL R� ( R R� t entry( ( R� R� sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR] ] s
c ` s� i } d } � j } t j d g } t | � } t j d t j � } x�| j � D]�} | j d � rd| ss qR n | j
� } t | � d k r� � j | � } n | d d } | d k r� qR n y~ | | k r� � j
| � } n | | k r@� j � j | � | � r@t t t � f d � t | d � � � � | | <n Wq�t t t f k
r`qR q�XqR | j d
� r�| | j d � j � } qR d | k rR | j | � }
|
r�� j j t t t d
� |
� � � � q�qR qR W| j � j � � | S( sZ
Gets current quota settings from repqouta utility for further processing
s -unas2 (block|inode)\sgrace\stime:?\s(\d[\w:]+)(?:;|$|\s)t #i i i Rf c ` s � j | d | d f S( Ni i ( RL ( R� ( R ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � R� i s ***s /devt gracec S` s | d j � | d f S( Ni i ( t lower( R� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � R� N( R' RQ R7 R? R R� R� t
IGNORECASEt
splitlinesR* Rs Rm t# _remove_redundant_fields_from_inputt _find_unknown_devicet _is_home_deviceR� R� RA R R� R� t
IndexErrorR t findt stript findallRD R� t _add_default( R R� R� t devicesR� R� t grace_regex_patternt linet partsR$ t found( ( R sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� j s> '9
2c C` sP | d } t j d � } | j g | d D] } | j | � r* | ^ q* � | S( Ni s ^\d+$( R� R� R� R� ( R R t stripped_partst is_digit_patternt field( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR � s
-c C` sq t | j � d k r> | j j t d � t j � D� � � n y | j | SWn t k
rl t | � � n Xd S( Ni c s` s' | ] } t | j � | j f Vq d S( N( R� R1 R0 ( R R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pys <genexpr>� s ( Rm RI R� R� t pwdt getpwallR� R ( R R$ ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s
c C` sE i } t t j � } t j d � } x� | D]� } | j d � rF q+ n | j | � } | d } i | d d 6} xo | D]g } | j d � s� | j d � r� | j d � d | d <qw | j d
� rw | j d � d | d <qw qw W| | k r| | j | � q+ | g | | <q+ W| j � t | � d k rAt
d d
� � n | S( sj
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'}
]
}
s |,s rootfs /i i t
mountpoints usrquota=s usruota=t =t
quota_files jqfmt=t
quota_types all userss all filesystems( t openR7 R> R� R� R* Rs R� RX Rm R# ( R R t proc_mounts_streamt
split_pattR t line_splitedR� t mountpoint_datat line_splited_element( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyRP � s*
c
` s/ i } y2 � j � t t j � } t j | d d �} WnA t t f k
r{ � j � i t j
� j d � t � j
� 6SX� j � t � j
� } xj | D]b � � d | k r� | j i t t t � � f d � t t � j � � � � � � d 6� Pq� q� W| s+i t j
� j d � t � j
� 6S| S( sJ
For non-privileged user we outputting data from the file
t delimitert ,Rf i c ` s � j | � | d f S( Ni ( RL ( R� ( t rowR ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � R� ( R� R% R7 R� R� t readerR� t IOErrorR� R� t fromkeysRL R� RO R� RA R t rangeRm ( R R� t fot cvs_inR$ ( ( R- R sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s$
$
6#c C` st d � } g } xH t | � D]: \ } } x+ | D]# } | d } | j | | f � q/ Wq W| j d | d t � | S( s�
return list tuple ('mountpoin tpath', 'device') reverse sorted by deep mountpoint path
[('/mountpoint_path/path', '/device'), ('/mountpoint_path', '/device')]
c S` s0 | d d k r d } n | d j d � } | S( Ni R( ( t count( t device_mountpointt deep_path( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt sort_by_deep_path� s R! R� t reverse( R R� t sortR) ( R R R7 t mountpoint_device_mapR� t mountpoint_data_listR) t mountpoint_path( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyRR � s
c C` sX d � } | t j j | � � } x0 | j D]% \ } } | j | | � � r+ | Sq+ Wd S( s3
Returns device user homedir is on
c S` s' | r# | d d k r# | d 7} n | S( Ni����R( ( ( R+ ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt
_add_slash� s
N( RM R+ t dirnameRS R* ( R t homeR= R> t mounpoint_pathR� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s
c C` s | j | � | k S( s;
Checks if a device is user homedir device
( R� ( R R? R� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR s c C` s� y� t j | � t } t j | � t j | � f } xR | j j � D]A } t j | � t } | t j | � t j | � f k rD | SqD WWn t k
r� | SXd S( N( RM t statR t majort minorRQ Rv R� ( R R� t devt dev_to_findt current_device( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR s $
c ` s� d d d d g � ye | j j d d � j d � \ } } | d k rN d } n | d k rc d } n � j | | g � Wn0 t j t j f k
r� � j d d g � n Xi t t t � f d � t
| j � � � � d 6S( sw
Insert 'default' quota.
Calls only from _load_current_quotas, after parsing repquota's output
R� Rf R} Rk s -1c ` s | d � | d f S( Ni i ( ( R� ( R� ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� ! R� ( RK Rr Rs R� R� R� R� R� RA R R� RL ( R R� R� ( ( R� sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR s $ c C` s� | d k r d } n | d k r* d } n | j | | | � \ } } | d k s] | d k r� | d k s� | d k r� | j j d � r� | j j d | � nE | j j d � s� | j j d � n | j j d | d | | f � | j � d S( s#
Saves user limits
Rf R} s %s:%sN( R' R� RK Rl Rq t add_sectiont setRt ( R R$ R~ R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� # s * c C` s� | d k r d } n | d k r* d } n | j | | | � \ } } | d k s] | d k r� | d k s� | d k r� | j j d � r� | j j d | � nE | j j d � s� | j j d � n | j j d | d | | f � | j � d S( s&
Saves package limits
Rf Rj s %s:%sN( R' R� RK Rl Rq RG RH Rt ( R R R~ R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� 5 s 0 c C` s� | d k rI | d k rI y | j j | | � Wq� t j k
rE q� XnE | j j | � sn | j j | � n | j j | | d | | f � | j � d S( s&
Saves data to a file
Rf s %s:%sN( RK Rq R� R� Rl RG RH Rt ( R R~ R R� t item_type( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt
_save_dataG s c C` s� t j j | � } yA t j d d d | � \ } } t j | d � } | | _ | SWnE t t f k
r� t j j | � r� t j
| � n t d � � n Xd S( sH
Open temporaty file for writing and return file object
t prefixt lvetmp_t dirt wbs Could not save dataN( RM R+ R> t tempfilet mkstempt fdopent _tmpR/ R� t existst unlinkR! ( R t filepathR+ t fdt temp_pathR� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� V s c C` sQ y9 t t Bt Bt B} t j | j | � t j | | � Wn t k
rL n Xd S( s0
Routines after writing to file
N( R R R R RM t renameRR t chmodR� ( R R+ t mask( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� e s
c C` sM | j t � | j t j � } | j j | � | j t j � | j � d S( s7
Actual place of saving data to a file
N( R� R) R� R7 R� RK t writeR� R� ( R R� ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyRt p s
c C` s� | r t t _ n t j d k r� y t t j d � t _ Wn# t t f k
r_ t d � � n Xy t
j t j j � t
j
� Wq� t k
r� t d � � q� Xn d S( Nt rs Can't open lock file for readings Can't get lock( R) R7 t
LOCK_WRITERW R' R% t LOCK_FILER/ R� R! t fcntlt flockt filenot LOCK_EX( R R[ ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� | s
c C` s6 t j r2 t j d k r2 t j j � d t _ n d S( N( R7 R] RW R' RX ( R ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR� � s
N(I R R t __doc__R> R@ R? t GETPACKSR� R� R' RW R^ R� R] R RV R\ R_ R` Rc Re Ri R| R) R� R� R� R� R� R� RJ R� Rh t staticmethodR� R� R� R� R� R� R� R� R� R^ Rb R� R� R� Ru R� R� R� R� Ro R] R� R R� RP R� RR R� R R R R� R� RJ R� R� Rt R� R� ( ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyR7 d s�
" rH -
%
$ &
(1 t
__future__R R R RM R� R R� R� RO R_ RA R R R R R t builtinsR t future.utilsR t clcommon.cpapi.cpapiexceptionsR
t clcommon.utilsR R R
t clcommon.clpwdR t collectionsR t clcommon.cpapiR R R t clcommon.clquotaR R� R R R R R R! R# R. R6 t objectR7 ( ( ( sA /opt/alt/python27/lib/python2.7/site-packages/clquota/__init__.pyt <module> s8 (
Mini Shell