<0ׄ d dl Z d dlmZmZmZ d dlmZ d dlmZm Z d dl
m Z d dl m
Z
mZ G d de Z G d d
e Z G d d e Z G d
d ZdS ) N)DictOptionalTuple)FormattedException)ExternalProgramFailed run_command) exec_utility)is_quota_activeis_quota_supportedc e Zd ZdS )ClQuotaExceptionN)__name__
__module__ __qualname__ py/cllimits/clquota_lib.pyr
r
s Dr r
c $ e Zd ZdZd fd Z xZS )UnexpectedClQuotaErrorzU
Raised when we get output from cl-quota
that cannot be parsed properly.
Nc Z dd|dd}t | d S )NzUnexpected response got from %(util)s. Raw response from cl-quota: '%(raw_response)s'.Please, contact CloudLinux support for help in resolving this issue.Quota)util raw_responsemessagecontextsuper__init__)selfr broken_exc_message __class__s r r z UnexpectedClQuotaError.__init__ sF *p 3:<)X)X Z Z +,,,,,r Nr r r __doc__r
__classcell__r" s @r r r sG - - - - - - - - - -r r c " e Zd ZdZ fdZ xZS )QuotaDisabledExceptionz=
Raised when quotas are disabled in system globally.
c X dddid}t | d S )N%(util)s is disabledr r r r )r disabled_exc_messager" s r r z QuotaDisabledException.__init__( s: +A,2G+< > >
-.....r r$ r' s @r r) r) $ sB / / / / / / / / /r r) c r e Zd ZdZdZd Zd Zd Zd ZddZ d Z
d
Z d Z d Z
d
Zd Zd Zd Zd Zd ZdS )
ClQuotaLibz/usr/bin/cl-quotaz/usr/sbin/repquotac > d | _ d | _ d | _ d | _ d S r# )_inode_user_limits_inode_package_dict_is_clquota_present_is_clquota_activatedr s r r zClQuotaLib.__init__2 s, IM IM #' %)"""r c \ | j t | j | j | _ | j S )zb
Get quota presence flag
:return: True/False - quotas present/not present
)r2 r _CL_QUOTA_PATH_REPQUOTA_PATHr4 s r is_clquota_presentz ClQuotaLib.is_clquota_present< s.
# +'9$:MtOb'c'cD $''r c \ | j t | j | j | _ | j S )zg
Get quota activated flag
:return: True/False - quotas activeted/not activeted
)r3 r
r6 r7 r4 s r is_clquota_activatedz ClQuotaLib.is_clquota_activatedE s.
% -)89LdNa)b)bD &))r c | t | }|| j v r
| j | S | j d S )z
Retrive inodes limits by uid
:param user_id: Supplied uid
:return: cortege (soft_limit, hard_limit). (None, None) if governor not present or error
0)_load_users_infostrr0 )r user_iduids r get_inodes_limits_by_uidz#ClQuotaLib.get_inodes_limits_by_uidN sK
'll $) ) )*3/ /&s++r Fc B | t |t r|dk rdd|id}t | |dk rdgdz }n|dk rdgdz }nt | t
t fv rnRt | d dk r| d }nd d
|id}t | |D ]V} |dk r#t | }|dk rt -# t $ r }d |dd d}t | |d
}~ww xY w| j
d| d|d d|d g}|r| d t | d
S # t $ r"}t t | |d
}~ww xY w)z
Set inodes limits for user uid
:param: int uid: user id
:param: list limits: new inodes limits
:param: bool force: save limits if even they are equal to defaults
:return: None
r z*User id '%(uid)s' isn't a positive integerr@ r unlimited default,z$Limits %(limits)s aren't acceptable.limitszFLimit value '%(limit)s' isn't a positive integer or string %(default)s)limitrE Nz
--user-id=
--soft-limit=
--hard-limit= z--force)r=
isinstanceintr
typetuplelistlensplit
ValueErrorr6 appendr r r> )r r@ rG force exc_messagerH ecmds r set_user_inodes_limitsz!ClQuotaLib.set_user_inodes_limits[ s
#s## 0sQww&R',cl4 4K";// / [ !]Q&FF
y
[1_FFF||t},,V\\#&&''1,, c***P+3V*< > > &{333
?
?? )) #E
199",,, .! ? ? ?.v8=).T.T#V #VK*;77Q>? "!C!!*vay***vay**, " JJy ! ! ! 2 $ 2 2 2"3q66** 1 2s0 )C??
D& D!!D&!E2 2
F < FF c 2 | |d dS )zq
Set unlimited inodes limits for user uid
:param: int uid: user id
:return: None
rC NrY r r@ s r set_user_inodes_limits_unlimitedz+ClQuotaLib.set_user_inodes_limits_unlimited s
##C55555r c 2 | |d dS )zo
Set default inodes limits for user uid
:param: int uid: user id
:return: None
rE Nr[ r\ s r reset_user_inodes_limitsz#ClQuotaLib.reset_user_inodes_limits s
##C33333r c | j dS | r| st | dg }n# t
$ r d| _ w xY wi | _ | D ]C\ }}t |d d t |d d f| j |< Dd| j vr&d| _ dd| j dd d
}t | dS ) zF
Loads users info from cl-quota
:return: None
N--jsonrK rD r< z'There is no %(what)s found in %(where)szdefault settingsz output)whatwherer ) r0 r8 r: r) _get_quotasr
itemsrM r6 )r datar@ rG rV s r r= zClQuotaLib._load_users_info s7 " . F&&(( +0I0I0K0K +(** * ##XJ//DD &*D # #% ::<< R RKC,/q ! ,=,=s6!9Q M:n--DD:& M M M(nEEE1 L M +(** * >Y & &" > 88Ir22 88I..$ $
G} s / A AAc | r| sdS t | j dg dS )zV
Reset inodes limits for all users to package limits
:return:
Nz--sync)r8 r: r r6 r4 s r reset_inodes_limitsz ClQuotaLib.reset_inodes_limits sK
&&(( 0I0I0K0K FT(8*55555r c | j dS | r| st | ddg }n# t
$ r d| _ w xY wi | _ | D ]^\ }}|d d }|dk rd}nt | }|d d }|dk rd}nt | }||d| j |< _dS ) zI
Loads packages info from cl-quota
:return: None
Nra z--all-package-limitsrK -r rD )softhard)r1 r8 r: r) rd r
re rM )r packagespackagerG
soft_limit
hard_limits r _load_packages_infoz ClQuotaLib._load_packages_info s, # / F&&(( +0I0I0K0K +(** * ''3I(JKKHH '+D $ $& '~~ / / Y YOGV 1 JS
__
1 JS
__
9CZ0X0XD $W - - Y Ys A A,c p | |s| j S || j v r|| j | iS i S )aY
Get inodes limits for supplied reseller and package
:param package_name_arg: Package name. Only if reseller name is provided. If None - all packages
:return: Packages limits dictionary: {package_name: { "soft": 100000, "hard": 200000 } }
If package with supplied name not found, dictionary will be empty
)r| r1 )r package_name_args r get_reseller_package_limitsz&ClQuotaLib.get_reseller_package_limits sN
""" ,+ + t 7 7 7$d&>?O&PQ Q r c $ d|||rd| dndddS )Nz3%(what)s set error for package=%(package)s%(error)sz [] )rb ry errorr r )r param package_nameerrs r __set_errorzClQuotaLib.__set_error s< P$)l47%?[#[[[[R A AB B Br c dddid}| r| st | | d }t | dk rt dddid d | d
|d d |d
g}t | j | \ }}|d k s|r$t | d|| dS )z
Set inodes limits for package
Called from cloudlinux-packages set
:param package_name: Package name
:param limits_to_set: Limits to set: soft_limit,hard_limit
:return: None
r+ r r r rF rD z%(util)s argument errorzcl-quota
--package=rI r rJ rK
Inodes limitsN)r8 r: r
rR rQ r r6 _ClQuotaLib__set_error)r r
limits_to_setr, rG cl_quota_cmdret_codestdouts r set_reseller_package_limitsz&ClQuotaLib.set_reseller_package_limits s% ,B,2G+< > > &&(( 90I0I0K0K 9"#788 8 $$S)) v;;! "/H06
/C$E $E F F
F 4\333q 333q 335 ((;\JJ& q==F="4#3#3O\SY#Z#Z[[ [ =r c ddi}| r| st | t | dk rt ddi t | ddhk s;t | ddhz
}t ddd | id
d | g}d|v r | d |d d|v r | d
|d t | j | \ }}|dk s|r$t | d|| dS )z
Set inodes limits for package
Called from cl-syncpkgs
:param package_name: Package name
:param limits: Limits to set: {"inodes_soft":1000, "inodes_hard":2000}
:return: None
r zQuota is disabledr z Unspecified limits for cl-quota inodes_soft inodes_hardz-Incorrect limits for cl-quota: %(bad_limits)s
bad_limitsz, r r rI rJ r N)
r8 r: r
rQ setjoinrT r r6 r )r r rG r, r r r r s r set_package_limits_independentlyz+ClQuotaLib.set_package_limits_independently s !*+> ? &&(( 90I0I0K0K 9"#788 8 v;;! "I/P#QRR R6{{}m <<< V
}'EEJ"/^0S>S/T$V $V W W
W3\334 F " " G}0E G G H H H F " " G}0E G G H H H'(;\JJ& q==F="4#3#3O\SY#Z#Z[[ [ =r N)F)r r r r6 r7 r r8 r: rA rY r] r_ r= rd rs r| r r r r r r r r. r. . s (N)N* * *( ( (* * * , , ,.2 .2 .2 .2`6 6 64 4 40 0 04 $6 6 6 Y Y Y> B B B
\ \ \6\ \ \ \ \r r. )rk typingr r r clcommon.clexceptionr clcommon.utilsr r cllimits.libr cllimits.lib.utilsr
r r
r r) r. r r r r sI ( ( ( ( ( ( ( ( ( ( 3 3 3 3 3 3 = = = = = = = = % % % % % % B B B B B B B B )
-
-
-
-
- -
-
-
-/ / / / / - / / /J\ J\ J\ J\ J\ J\ J\ J\ J\ J\r