PKCZE __init__.pynu[# coding=utf-8 # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT from .lve_storage import ( LimitsDataStorage, DEFAULTS, OWN_SOURCE, PACKAGE_SOURCE, RESELLER_SOURCE, GLOBAL_SOURCE, BUILTIN_SOURCE ) __all__ = ( 'LimitsDataStorage', 'DEFAULTS', 'OWN_SOURCE', 'PACKAGE_SOURCE', 'RESELLER_SOURCE', 'GLOBAL_SOURCE', 'BUILTIN_SOURCE' ) PKCZ'K*__pycache__/__init__.cpython-311.opt-1.pycnu[ g.ddlmZmZmZmZmZmZmZdZdS))LimitsDataStorageDEFAULTS OWN_SOURCEPACKAGE_SOURCERESELLER_SOURCE GLOBAL_SOURCEBUILTIN_SOURCEN) lve_storagerrrrrrr __all__py/cllimitslib_v2/__init__.pyrsj  r PKCZ5$__pycache__/__init__.cpython-311.pycnu[ g.ddlmZmZmZmZmZmZmZdZdS))LimitsDataStorageDEFAULTS OWN_SOURCEPACKAGE_SOURCERESELLER_SOURCE GLOBAL_SOURCEBUILTIN_SOURCEN) lve_storagerrrrrrr __all__N/opt/cloudlinux/venv/lib64/python3.11/site-packages/cllimitslib_v2/__init__.pyrsj  r PKCZ5Ϊ@@-__pycache__/lve_storage.cpython-311.opt-1.pycnu[ g|A(ddlmZddlmZmZmZmZmZmZddl m Z ddl m Z m Z ddlmZmZmZmZmZddlmZmZmZmZmZedgdZed gd Zed gd Zed ddgZdZdZdZ dZ!dZ"dZ#GddZ$dS)) namedtuple)DictIteratorListOptionalTupleUnion)ClPwd) NotSupported resellers)DEFAULT_PROVIDER EMPTY_LIMITSInheritedLimitsLimitsXmlConfigReader)DEFAULT_PACKAGE LVE_DEFAULT LVP_DEFAULTpanelpackagesdictpaneluserslistPackagenameproviderlimitsResellerrdefaultsrUseridpackagerreseller is_panel_user IneritedLimitvaluesourceDEFAULTSownr"r#globalbuiltinceZdZdZeeddZeeedZ ddZ e dZ dZ e dZd Zd Zd Zd Zed ZedZedZedZdZdZdZdZdZdZdS)LimitsDataStoragea This class is designed to merge data from xml config & control panel and provide user-friendy interface to work with lve limits (packages, resellers, defaults) What this class can do: - process information from xml config and control panel - re-pack structures for easy data access What this class should not do: - set limits in kernel - know anything about kernel limits Nrrct|_|p t|_d|_d|_d|_d|_d|_dSN) r _pwdr _xml_source_panel_resellers_list _packages_users _resellers _defaults)self_xml_config_readers py/cllimitslib_v2/lve_storage.py__init__zLimitsDataStorage.__init__@sHGG -B1B1B%)" c(|dkrd}n|dkrd}|S)z2 Convert mem to vmem and reversed vmemmem) limit_names r9_limit_name_convertz%LimitsDataStorage._limit_name_convertIs,   JJ 5 Jr;cv|j, t|_n#t$r g|_YnwxYw|jS)z Get list of names of panel resellers. If cpapi panel plugin don't support that method, we return empty list )r2r r r7s r9_get_panel_resellersz&LimitsDataStorage._get_panel_resellersWsS  % - 0-6[[** 0 0 0-/*** 0))s 11c|jduS)z@ Not activated reseller doesn't have any limits N)r)r#s r9_is_active_resellerz%LimitsDataStorage._is_active_resellerfs  ,,r;c#JK|jD]}|VdSr/)r1 users_lve_idsr7uids r9_list_uids_from_ve_cfgz(LimitsDataStorage._list_uids_from_ve_cfgns9#1133  CIIII  r;c||jvr |j|St||j|j||jdS)ax This method will return User by uid. If we will pass uid of nonexistent user, method return User without any limits, with with DEFAULT_RESELLER_OBJ and with DEFAULT_PKG_OBJ. This trick is needed for validating of limits for nonexistent user in case of migrating limits and entities from another server. Fr )usersrDEFAULT_PKG_OBJr1get_user_limitsDEFAULT_RESELLER_OBJrIs r9get_user_by_uidz!LimitsDataStorage.get_user_by_uidssZ $*  :c? "(#33C88.     r;c||f|jvr|j||fSt|f|jvr|jt|fSt||j|j|tS)a' This method is needed because of lvectl bug "by design" our api designed only for situations when user's owner is the owner of the package that contains that user but sometimes user can be in admin's package. We will return entity Package without any limits and with DEFAULT_RESELLER_OBJ if we request name of nonexistent package. This trick is needed for validating of limits for nonexistent package in case of migrating limits and entities from another server r)packagesr rrPr1get_package_limits)r7rr#s r9 get_package_by_name_and_resellerz2LimitsDataStorage.get_package_by_name_and_resellers d t} , ,=(D!12 2 d #t} 4 4=!14!78 8.#66     r;c<||jvr |j|S|jSr/)r rPr7rs r9get_reseller_by_namez&LimitsDataStorage.get_reseller_by_names% 4> ! !>$' '((r;c|jfi|_|D]$}|j||j|<%|j|jt <|jSr/)r6rDr1get_reseller_default_limitsrr(rWs r9rzLimitsDataStorage.defaultsst > !DN1133 Z Z'+'7'S'STX'Y'Yt$$'+'7'@'@'B'BDN8 $~r;c |jzi|_tD]R\}}|D]J}t||||j|||j||f<KS|jS)Nr)r3ritemsrrXr1rT)r7rprovider_packages package_names r9rSzLimitsDataStorage.packagess > !DN/@/B/B/H/H/J/J  ++$5L?F)!%!:!:8!D!D#/BB($  @@@DNHl#;<<~r;c |ji|_tD]a\}}}t|||||||j|d|j|<b|D]I}||jvr>t||j|j||j d|j|<J|jS)NT)r!r#r"rr$Fr ) r4rrrXrUr1rOrKrNrP)r7rJr#r"s r9rMzLimitsDataStorage.userss ; DK*8*:*:  &Xw#'!66x@@ AA'8TT+;;C@@"& $$$ C  2244  dk))'+ $ 4#/??DD!%!:&+ (((DK${r;c|jhi|_|D]L}t||j||j||j|<M|jS)Nr)r5rDrr1rZget_reseller_limits)r7 reseller_names r9r zLimitsDataStorage.resellerss ? " DO!%!:!:!zELimitsDataStorage.get_packages_with_custom_limits..s<   )S'~)gn .L.L .L.L.Lr;)rSr\rCs r9get_packages_with_custom_limitsz1LimitsDataStorage.get_packages_with_custom_limitss1  -1]-@-@-B-B    r;cHd|jDS)NcVi|]&\}}|j |jtk|j#||'Sr/)rrr$)rerJusers r9rgzBLimitsDataStorage.get_users_with_custom_limits..sD   ##t{&4;,+F+F4K]+F +F+F+Fr;)rMr\rCs r9get_users_with_custom_limitsz.LimitsDataStorage.get_users_with_custom_limitss1  '+z'7'7'9'9    r;c\|jj}|jt}|jpt}|D]\}}|t||d}t||d}t| |} |t|t||<e|4| |jst|t||<t| t||<t|t||<t!di|S)zA Method will return inherited limits for package Nr&r'r?)rrr(rr_asdictr\getattrrrAInheritedLimitRESELLER_SOURCErF GLOBAL_SOURCEBUILTIN_SOURCE OWN_SOURCEr) r7r"reseller_defaultsglobal_defaultspackage_own_limitsrfr&reseller_default_valueglobal_default_valuebuiltin_default_values r9rTz$LimitsDataStorage.get_package_limitssc $,5-1%n< EEGG,2244  JC})01BC)N)N&'.T'J'J$(3D4L4LS4Q4Q(R%)5.<4.///&s++*5d>V>VW^Wg>h>h5.<2,///&s++ /=3-///&s++ +9%+++"3''44!3444r;c|jt}|jj}|jj}|jpt }|D]\}}|t||d}t||d} t||d} t| |} |t|t||<v| t| t||<| 4||jst| t||<t| t ||<t|t"||<t%di|S)z> Method will return inherited limits for user Nrnr?)rr(r#r"rrror\rprrArqPACKAGE_SOURCErrrFrsrtrur) r7rkrwrvpackage_limits user_limitsrfr& package_valueryrzr{s r9rOz!LimitsDataStorage.get_user_limits"s -1 M2,{2l;;== %++--" " JC} 'T B B )01BC)N)N&'.T'J'J$(3D4L4LS4Q4Q(R% ,'5+-(((K$$,7'54.(((K$$*5d>V>VW[Wd>e>e5'52,(((K$$ (63-(((K$$ $2%$$$ C  -----r;c t||stttdtttdtttdtttdtttdtttdtttdtttd t S|j}|D]Z\}}|:t||}t|t||<At|t||<[td i|S) z Method will return limits equal builtin for non-activated reseller and inherited limits for activated reseller cpurnncpuior>pmemnprocepiops)rrrr=rrrrNr?) rFrrqrrtrror\rAru)r7r#reseller_limitsrfr&r{s r9raz%LimitsDataStorage.get_reseller_limitsQs''11" ""%e,)$%f-)"%d+)$%e,)$%f-)%%g.)"%d+)$%f-);!!!! F#/1133)//11  JC}(3D4L4LS4Q4Q(R%'5/)((($$ (6%((($$11111r;c|jt}|tkr|}nK||st dit jS|j}|D]Z\}}|:t||}t|t||<At|t||<[t di|S)z Method will return empty inherited defaults limits for non-activated reseller or inherited defaults limits for activated reseller. For global defaults method will return inherited limits Nrnr?) rr(rorFrrr\rrArqrtru)r7r#rwrvrfr&r{s r9get_defaults_limitsz%LimitsDataStorage.get_defaults_limitss-1 x   / 7 7 9 9  ++H55 A&@@)=)?)?@@@ ( 1 9 9 ; ; +1133  JC}(3D4L4LS4Q4Q(R%)7/)***!#&& *8%***!#&&33!2333r;r/)__name__ __module__ __qualname____doc__rr rPrrrNr: staticmethodrArDrFrKrQrUrXpropertyrrSrMr rhrlrTrOrarr?r;r9r-r- s   $8 g %O   \  * * *--\-    ,   6))) XX X0  X       &5&5&5P-.-.-.^;2;2;2z44444r;r-N)% collectionsrtypingrrrrrr clcommonr clcommon.cpapir r clveconfigr rrrr lvectllibrrrrrrrrrqr(rur}rrrsrtr-r?r;r9rs#"""""????????????????22222222______________bbbbbbbbbbbbbb *Y > > > ? ? :j"@"@"@ A Az&RRRSSOgx-@AA   K4K4K4K4K4K4K4K4K4K4r;PKCZ8@@'__pycache__/lve_storage.cpython-311.pycnu[ g|A(ddlmZddlmZmZmZmZmZmZddl m Z ddl m Z m Z ddlmZmZmZmZmZddlmZmZmZmZmZedgdZed gd Zed gd Zed ddgZdZdZdZ dZ!dZ"dZ#GddZ$dS)) namedtuple)DictIteratorListOptionalTupleUnion)ClPwd) NotSupported resellers)DEFAULT_PROVIDER EMPTY_LIMITSInheritedLimitsLimitsXmlConfigReader)DEFAULT_PACKAGE LVE_DEFAULT LVP_DEFAULTpanelpackagesdictpaneluserslistPackagenameproviderlimitsResellerrdefaultsrUseridpackagerreseller is_panel_user IneritedLimitvaluesourceDEFAULTSownr"r#globalbuiltinceZdZdZeeddZeeedZ ddZ e dZ dZ e dZd Zd Zd Zd Zed ZedZedZedZdZdZdZdZdZdZdS)LimitsDataStoragea This class is designed to merge data from xml config & control panel and provide user-friendy interface to work with lve limits (packages, resellers, defaults) What this class can do: - process information from xml config and control panel - re-pack structures for easy data access What this class should not do: - set limits in kernel - know anything about kernel limits Nrrct|_|p t|_d|_d|_d|_d|_d|_dSN) r _pwdr _xml_source_panel_resellers_list _packages_users _resellers _defaults)self_xml_config_readers Q/opt/cloudlinux/venv/lib64/python3.11/site-packages/cllimitslib_v2/lve_storage.py__init__zLimitsDataStorage.__init__@sHGG -B1B1B%)" c(|dkrd}n|dkrd}|S)z2 Convert mem to vmem and reversed vmemmem) limit_names r9_limit_name_convertz%LimitsDataStorage._limit_name_convertIs,   JJ 5 Jr;cv|j, t|_n#t$r g|_YnwxYw|jS)z Get list of names of panel resellers. If cpapi panel plugin don't support that method, we return empty list )r2r r r7s r9_get_panel_resellersz&LimitsDataStorage._get_panel_resellersWsS  % - 0-6[[** 0 0 0-/*** 0))s 11c|jduS)z@ Not activated reseller doesn't have any limits N)r)r#s r9_is_active_resellerz%LimitsDataStorage._is_active_resellerfs  ,,r;c#JK|jD]}|VdSr/)r1 users_lve_idsr7uids r9_list_uids_from_ve_cfgz(LimitsDataStorage._list_uids_from_ve_cfgns9#1133  CIIII  r;c||jvr |j|St||j|j||jdS)ax This method will return User by uid. If we will pass uid of nonexistent user, method return User without any limits, with with DEFAULT_RESELLER_OBJ and with DEFAULT_PKG_OBJ. This trick is needed for validating of limits for nonexistent user in case of migrating limits and entities from another server. Fr )usersrDEFAULT_PKG_OBJr1get_user_limitsDEFAULT_RESELLER_OBJrIs r9get_user_by_uidz!LimitsDataStorage.get_user_by_uidssZ $*  :c? "(#33C88.     r;c||f|jvr|j||fSt|f|jvr|jt|fSt||j|j|tS)a' This method is needed because of lvectl bug "by design" our api designed only for situations when user's owner is the owner of the package that contains that user but sometimes user can be in admin's package. We will return entity Package without any limits and with DEFAULT_RESELLER_OBJ if we request name of nonexistent package. This trick is needed for validating of limits for nonexistent package in case of migrating limits and entities from another server r)packagesr rrPr1get_package_limits)r7rr#s r9 get_package_by_name_and_resellerz2LimitsDataStorage.get_package_by_name_and_resellers d t} , ,=(D!12 2 d #t} 4 4=!14!78 8.#66     r;c<||jvr |j|S|jSr/)r rPr7rs r9get_reseller_by_namez&LimitsDataStorage.get_reseller_by_names% 4> ! !>$' '((r;c|jfi|_|D]$}|j||j|<%|j|jt <|jSr/)r6rDr1get_reseller_default_limitsrr(rWs r9rzLimitsDataStorage.defaultsst > !DN1133 Z Z'+'7'S'STX'Y'Yt$$'+'7'@'@'B'BDN8 $~r;c |jzi|_tD]R\}}|D]J}t||||j|||j||f<KS|jS)Nr)r3ritemsrrXr1rT)r7rprovider_packages package_names r9rSzLimitsDataStorage.packagess > !DN/@/B/B/H/H/J/J  ++$5L?F)!%!:!:8!D!D#/BB($  @@@DNHl#;<<~r;c |ji|_tD]a\}}}t|||||||j|d|j|<b|D]I}||jvr>t||j|j||j d|j|<J|jS)NT)r!r#r"rr$Fr ) r4rrrXrUr1rOrKrNrP)r7rJr#r"s r9rMzLimitsDataStorage.userss ; DK*8*:*:  &Xw#'!66x@@ AA'8TT+;;C@@"& $$$ C  2244  dk))'+ $ 4#/??DD!%!:&+ (((DK${r;c|jhi|_|D]L}t||j||j||j|<M|jS)Nr)r5rDrr1rZget_reseller_limits)r7 reseller_names r9r zLimitsDataStorage.resellerss ? " DO!%!:!:!zELimitsDataStorage.get_packages_with_custom_limits..s<   )S'~)gn .L.L .L.L.Lr;)rSr\rCs r9get_packages_with_custom_limitsz1LimitsDataStorage.get_packages_with_custom_limitss1  -1]-@-@-B-B    r;cHd|jDS)NcVi|]&\}}|j |jtk|j#||'Sr/)rrr$)rerJusers r9rgzBLimitsDataStorage.get_users_with_custom_limits..sD   ##t{&4;,+F+F4K]+F +F+F+Fr;)rMr\rCs r9get_users_with_custom_limitsz.LimitsDataStorage.get_users_with_custom_limitss1  '+z'7'7'9'9    r;c\|jj}|jt}|jpt}|D]\}}|t||d}t||d}t| |} |t|t||<e|4| |jst|t||<t| t||<t|t||<t!di|S)zA Method will return inherited limits for package Nr&r'r?)rrr(rr_asdictr\getattrrrAInheritedLimitRESELLER_SOURCErF GLOBAL_SOURCEBUILTIN_SOURCE OWN_SOURCEr) r7r"reseller_defaultsglobal_defaultspackage_own_limitsrfr&reseller_default_valueglobal_default_valuebuiltin_default_values r9rTz$LimitsDataStorage.get_package_limitssc $,5-1%n< EEGG,2244  JC})01BC)N)N&'.T'J'J$(3D4L4LS4Q4Q(R%)5.<4.///&s++*5d>V>VW^Wg>h>h5.<2,///&s++ /=3-///&s++ +9%+++"3''44!3444r;c|jt}|jj}|jj}|jpt }|D]\}}|t||d}t||d} t||d} t| |} |t|t||<v| t| t||<| 4||jst| t||<t| t ||<t|t"||<t%di|S)z> Method will return inherited limits for user Nrnr?)rr(r#r"rrror\rprrArqPACKAGE_SOURCErrrFrsrtrur) r7rkrwrvpackage_limits user_limitsrfr& package_valueryrzr{s r9rOz!LimitsDataStorage.get_user_limits"s -1 M2,{2l;;== %++--" " JC} 'T B B )01BC)N)N&'.T'J'J$(3D4L4LS4Q4Q(R% ,'5+-(((K$$,7'54.(((K$$*5d>V>VW[Wd>e>e5'52,(((K$$ (63-(((K$$ $2%$$$ C  -----r;c t||stttdtttdtttdtttdtttdtttdtttdtttd t S|j}|D]Z\}}|:t||}t|t||<At|t||<[td i|S) z Method will return limits equal builtin for non-activated reseller and inherited limits for activated reseller cpurnncpuior>pmemnprocepiops)rrrr=rrrrNr?) rFrrqrrtrror\rAru)r7r#reseller_limitsrfr&r{s r9raz%LimitsDataStorage.get_reseller_limitsQs''11" ""%e,)$%f-)"%d+)$%e,)$%f-)%%g.)"%d+)$%f-);!!!! F#/1133)//11  JC}(3D4L4LS4Q4Q(R%'5/)((($$ (6%((($$11111r;c|jt}|tkr|}nK||st dit jS|j}|D]Z\}}|:t||}t|t||<At|t||<[t di|S)z Method will return empty inherited defaults limits for non-activated reseller or inherited defaults limits for activated reseller. For global defaults method will return inherited limits Nrnr?) rr(rorFrrr\rrArqrtru)r7r#rwrvrfr&r{s r9get_defaults_limitsz%LimitsDataStorage.get_defaults_limitss-1 x   / 7 7 9 9  ++H55 A&@@)=)?)?@@@ ( 1 9 9 ; ; +1133  JC}(3D4L4LS4Q4Q(R%)7/)***!#&& *8%***!#&&33!2333r;r/)__name__ __module__ __qualname____doc__rr rPrrrNr: staticmethodrArDrFrKrQrUrXpropertyrrSrMr rhrlrTrOrarr?r;r9r-r- s   $8 g %O   \  * * *--\-    ,   6))) XX X0  X       &5&5&5P-.-.-.^;2;2;2z44444r;r-N)% collectionsrtypingrrrrrr clcommonr clcommon.cpapir r clveconfigr rrrr lvectllibrrrrrrrrrqr(rur}rrrsrtr-r?r;r9rs#"""""????????????????22222222______________bbbbbbbbbbbbbb *Y > > > ? ? :j"@"@"@ A Az&RRRSSOgx-@AA   K4K4K4K4K4K4K4K4K4K4r;PKCZ L5|A|Alve_storage.pynu[# coding=utf-8 # # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT from collections import namedtuple from typing import Dict, Iterator, List, Optional, Tuple, Union # NOQA from clcommon import ClPwd from clcommon.cpapi import NotSupported, resellers from clveconfig import DEFAULT_PROVIDER, EMPTY_LIMITS, InheritedLimits, Limits, XmlConfigReader # NOQA from lvectllib import DEFAULT_PACKAGE, LVE_DEFAULT, LVP_DEFAULT, panelpackagesdict, paneluserslist Package = namedtuple('Package', ['name', 'provider', 'limits']) Reseller = namedtuple('Reseller', ['name', 'defaults', 'limits']) User = namedtuple('User', ['id', 'package', 'limits', 'reseller', 'is_panel_user']) InheritedLimit = namedtuple('IneritedLimit', ['value', 'source']) # Name of key for defaults limits DEFAULTS = 'DEFAULTS' # Names of source inheritance OWN_SOURCE = 'own' PACKAGE_SOURCE = 'package' RESELLER_SOURCE = 'reseller' GLOBAL_SOURCE = 'global' BUILTIN_SOURCE = 'builtin' class LimitsDataStorage: """ This class is designed to merge data from xml config & control panel and provide user-friendy interface to work with lve limits (packages, resellers, defaults) What this class can do: - process information from xml config and control panel - re-pack structures for easy data access What this class should not do: - set limits in kernel - know anything about kernel limits """ # If we can't find Reseller with defined name, # we return Reseller with default name (DEFAULT_PROVDER), # and without any limits (own and defaults) DEFAULT_RESELLER_OBJ = Reseller( name=DEFAULT_PROVIDER, defaults=None, limits=None, ) # hack for cPanel; for some reason we do not allow apply limits of 'default' package # instead, we use 'virtual' package named VE_DEFAULT without any limits # FIXME: LU-936 DEFAULT_PKG_OBJ = Package( name=DEFAULT_PACKAGE, provider=DEFAULT_RESELLER_OBJ, limits=None, ) def __init__(self, _xml_config_reader=None): self._pwd = ClPwd() self._xml_source = _xml_config_reader or XmlConfigReader() self._panel_resellers_list = None # type: List[Optional[str]] self._packages = None # type: Dict[Tuple[str, str], Package] self._users = None # type: Dict[int, User] self._resellers = None # type: Dict[str, Reseller] self._defaults = None # type: Dict[str, Limits] @staticmethod def _limit_name_convert(limit_name): # type: (str) -> str """ Convert mem to vmem and reversed """ if limit_name == 'vmem': limit_name = 'mem' elif limit_name == 'mem': limit_name = 'vmem' return limit_name def _get_panel_resellers(self): # type: () -> List[Optional[str]] """ Get list of names of panel resellers. If cpapi panel plugin don't support that method, we return empty list """ if self._panel_resellers_list is None: try: self._panel_resellers_list = resellers() except NotSupported: self._panel_resellers_list = [] return self._panel_resellers_list @staticmethod def _is_active_reseller(reseller): # type: (Reseller) -> bool """ Not activated reseller doesn't have any limits """ return reseller.defaults is not None def _list_uids_from_ve_cfg(self): # type: () -> Iterator[int] for uid in self._xml_source.users_lve_ids(): yield uid def get_user_by_uid(self, uid): # type: (int) -> User """ This method will return User by uid. If we will pass uid of nonexistent user, method return User without any limits, with with DEFAULT_RESELLER_OBJ and with DEFAULT_PKG_OBJ. This trick is needed for validating of limits for nonexistent user in case of migrating limits and entities from another server. """ if uid in self.users: return self.users[uid] return User( id=uid, package=self.DEFAULT_PKG_OBJ, limits=self._xml_source.get_user_limits(uid), reseller=self.DEFAULT_RESELLER_OBJ, is_panel_user=False, ) def get_package_by_name_and_reseller(self, name, reseller): # type: (str, str) -> Package """ This method is needed because of lvectl bug "by design" our api designed only for situations when user's owner is the owner of the package that contains that user but sometimes user can be in admin's package. We will return entity Package without any limits and with DEFAULT_RESELLER_OBJ if we request name of nonexistent package. This trick is needed for validating of limits for nonexistent package in case of migrating limits and entities from another server """ if (reseller, name) in self.packages: return self.packages[(reseller, name)] if (DEFAULT_PROVIDER, name) in self.packages: return self.packages[DEFAULT_PROVIDER, name] return Package( name=name, provider=self.DEFAULT_RESELLER_OBJ, limits=self._xml_source.get_package_limits( name, DEFAULT_PROVIDER, ) ) def get_reseller_by_name(self, name): # type: (str) -> Reseller if name in self.resellers: return self.resellers[name] return self.DEFAULT_RESELLER_OBJ @property def defaults(self): # type: () -> Dict[str, Limits] if self._defaults is None: self._defaults = {} for name in self._get_panel_resellers(): self._defaults[name] = self._xml_source.get_reseller_default_limits(name) self._defaults[DEFAULTS] = self._xml_source.defaults() return self._defaults @property def packages(self): # type: () -> Dict[Tuple[str, str], Optional[Package]] if self._packages is None: self._packages = {} for provider, provider_packages in panelpackagesdict().items(): for package_name in provider_packages: self._packages[(provider, package_name)] = Package( name=package_name, provider=self.get_reseller_by_name(provider), limits=self._xml_source.get_package_limits( package_name, provider, ) ) return self._packages @property def users(self): # type: () -> Dict[int, User] if self._users is None: self._users = {} for uid, reseller, package in paneluserslist(): self._users[uid] = User( id=uid, reseller=self.get_reseller_by_name(reseller), package=self.get_package_by_name_and_reseller(package, reseller), limits=self._xml_source.get_user_limits(uid), is_panel_user=True, ) for uid in self._list_uids_from_ve_cfg(): if uid not in self._users: self._users[uid] = User( id=uid, package=self.DEFAULT_PKG_OBJ, limits=self._xml_source.get_user_limits(uid), reseller=self.DEFAULT_RESELLER_OBJ, is_panel_user=False, ) return self._users @property def resellers(self): # type: () -> Dict[str, Reseller] if self._resellers is None: self._resellers = {} for reseller_name in self._get_panel_resellers(): self._resellers[reseller_name] = Reseller( name=reseller_name, defaults=self._xml_source.get_reseller_default_limits(reseller_name), limits=self._xml_source.get_reseller_limits(reseller_name) ) return self._resellers def get_packages_with_custom_limits(self): # type: () -> Dict[Tuple[str, str], Package] return { key: package for key, package in self.packages.items() if package.limits is not None and package.limits != EMPTY_LIMITS } def get_users_with_custom_limits(self): # type: () -> Dict[int, User] return { uid: user for uid, user in self.users.items() if user.limits is not None and user.limits != EMPTY_LIMITS and user.is_panel_user } # TODO: Come up with abstract mechanism of inheritance def get_package_limits(self, package): # type: (Package) -> InheritedLimits """ Method will return inherited limits for package """ reseller_defaults = package.provider.defaults # type: Limits global_defaults = self.defaults[DEFAULTS] # type: Limits package_own_limits = (package.limits or EMPTY_LIMITS)._asdict() for key, value in package_own_limits.items(): if value is None: reseller_default_value = getattr(reseller_defaults, key, None) global_default_value = getattr(global_defaults, key, None) builtin_default_value = LVE_DEFAULT[self._limit_name_convert(key)] if reseller_default_value is not None: package_own_limits[key] = InheritedLimit( value=reseller_default_value, source=RESELLER_SOURCE ) # Package's provider doesn't have Limits if it's deactivated # and package is inherited limits from global defaults. # Package isn't inherited from global defaults if # it's provider (reseller) is activated. elif global_default_value is not None and not self._is_active_reseller(package.provider): package_own_limits[key] = InheritedLimit( value=global_default_value, source=GLOBAL_SOURCE ) else: package_own_limits[key] = InheritedLimit( value=builtin_default_value, source=BUILTIN_SOURCE ) else: package_own_limits[key] = InheritedLimit( value=value, source=OWN_SOURCE ) return InheritedLimits(**package_own_limits) def get_user_limits(self, user): # type: (User) -> InheritedLimits """ Method will return inherited limits for user """ global_defaults = self.defaults[DEFAULTS] # type: Limits reseller_defaults = user.reseller.defaults # type: Limits package_limits = user.package.limits # type: Limits user_limits = (user.limits or EMPTY_LIMITS)._asdict() for key, value in user_limits.items(): if value is None: package_value = getattr(package_limits, key, None) reseller_default_value = getattr(reseller_defaults, key, None) global_default_value = getattr(global_defaults, key, None) builtin_default_value = LVE_DEFAULT[self._limit_name_convert(key)] if package_value is not None: user_limits[key] = InheritedLimit( value=package_value, source=PACKAGE_SOURCE ) elif reseller_default_value is not None: user_limits[key] = InheritedLimit( value=reseller_default_value, source=RESELLER_SOURCE ) # User's reseller doesn't have Limits if it's deactivated # and user is inherited limits from global defaults. # User isn't inherited from global defaults if # it's reseller is activated. elif global_default_value is not None and not self._is_active_reseller(user.reseller): user_limits[key] = InheritedLimit( value=global_default_value, source=GLOBAL_SOURCE ) else: user_limits[key] = InheritedLimit( value=builtin_default_value, source=BUILTIN_SOURCE ) else: user_limits[key] = InheritedLimit( value=value, source=OWN_SOURCE ) return InheritedLimits(**user_limits) def get_reseller_limits(self, reseller): # type: (Reseller) -> InheritedLimits """ Method will return limits equal builtin for non-activated reseller and inherited limits for activated reseller """ # We return limits equal builtin for non-activated reseller # while activation of reseller. # It for corner case when we check input limits at the moment of activation reseller if not self._is_active_reseller(reseller): return InheritedLimits( cpu=InheritedLimit( value=LVP_DEFAULT['cpu'], source=BUILTIN_SOURCE, ), ncpu=InheritedLimit( value=LVP_DEFAULT['ncpu'], source=BUILTIN_SOURCE, ), io=InheritedLimit( value=LVP_DEFAULT['io'], source=BUILTIN_SOURCE, ), vmem=InheritedLimit( value=LVP_DEFAULT['mem'], source=BUILTIN_SOURCE, ), pmem=InheritedLimit( value=LVP_DEFAULT['pmem'], source=BUILTIN_SOURCE, ), nproc=InheritedLimit( value=LVP_DEFAULT['nproc'], source=BUILTIN_SOURCE, ), ep=InheritedLimit( value=LVP_DEFAULT['ep'], source=BUILTIN_SOURCE, ), iops=InheritedLimit( value=LVP_DEFAULT['iops'], source=BUILTIN_SOURCE, ) ) reseller_limits = reseller.limits._asdict() for key, value in reseller_limits.items(): if value is None: builtin_default_value = LVP_DEFAULT[self._limit_name_convert(key)] reseller_limits[key] = InheritedLimit( value=builtin_default_value, source=BUILTIN_SOURCE ) else: reseller_limits[key] = InheritedLimit( value=value, source=OWN_SOURCE ) return InheritedLimits(**reseller_limits) def get_defaults_limits(self, reseller): # type: (Union[Reseller, DEFAULTS]) -> InheritedLimits """ Method will return empty inherited defaults limits for non-activated reseller or inherited defaults limits for activated reseller. For global defaults method will return inherited limits """ global_defaults = self.defaults[DEFAULTS] # type: Limits if reseller == DEFAULTS: reseller_defaults = global_defaults._asdict() else: # We return empty inherited limits for non-activated reseller, # because it has no any limits. if not self._is_active_reseller(reseller): return InheritedLimits(**EMPTY_LIMITS._asdict()) reseller_defaults = reseller.defaults._asdict() for key, value in reseller_defaults.items(): if value is None: builtin_default_value = LVE_DEFAULT[self._limit_name_convert(key)] reseller_defaults[key] = InheritedLimit( value=builtin_default_value, source=BUILTIN_SOURCE ) else: reseller_defaults[key] = InheritedLimit( value=value, source=OWN_SOURCE ) return InheritedLimits(**reseller_defaults) PKCZE __init__.pynu[PKCZ'K*__pycache__/__init__.cpython-311.opt-1.pycnu[PKCZ5$'__pycache__/__init__.cpython-311.pycnu[PKCZ5Ϊ@@-___pycache__/lve_storage.cpython-311.opt-1.pycnu[PKCZ8@@'fG__pycache__/lve_storage.cpython-311.pycnu[PKCZ L5|A|Alve_storage.pynu[PK;R