Qb* d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z
d dlm Z d dl m
Z
d dlmZ d dlmZmZ d dlmZ d dlmZ d d lmZ d d
lmZm Z d d l m Z d d l m!Z! d d
l"m#Z# d dl$m%Z%m&Z&m'Z' d dl(m)Z)m*Z* d dl+m,Z,m-Z- da.dZ/dZ0dZ1d Z2d Z3d Z4dCdZ5 G d de6 Z7e/fdZ8dDd Z9d Z:d Z; G d d e ZdEd$Z?dEd%Z@d& ZAd' ZBdFd)ZCdEd*ZDd+ ZEdDd,ZFdDd-ZGdDd.ZHdDd/ZId0 ZJd1 ZKd2 ZL G d3 d4eM ZNd5 ZOd6 ZPd7 ZQd8 ZRdEd9ZSd: ZTd;eUdeXeXe
d?eXeU dz d@eYdAeYd i }| D ]}| | |S )z
>>> merge_dicts({1: 2}, {3: 4})
{1: 2, 3: 4}
>>> merge_dicts({1: '1', 2: '2'}, {2: 'two', 3: 'three'})
{1: '1', 2: 'two', 3: 'three'}
)update)dicts
dic_mergeddic_s r merge_dictsr) = s5 J $ r c # j K t dt | | D ]}| |||z V dS )zS
:type chunk_length: int
:type _list: list
:rtype: Generator[list]
r N)rangelen)_list chunk_lengthis r
get_chunksr1 J sQ 1c%jj,
/
/ ( (Aa,&&'''''( (r c e Zd ZddZdS )LVEVersionError c B t | d| d S )NzCan't detect LVE version; ) Exception__init__)selfmessages r r7 zLVEVersionError.__init__U s' 4!Gg!G!GHHHHHr N)r4 )__name__
__module__ __qualname__r7 r r r r3 r3 T s. I I I I I Ir r3 c t | dd 5 }| d } t | d d cddd S # t t
t f$ r}t d| d | |d}~ww xY w# 1 swxY w Y dS # t t f$ r}t | |d}~ww xY w)
z0
:param str _lve_file:
:return int:
rutf-8encoding :r Nzerror parsing line 'z ' from file )
openreadintsplit TypeError
IndexError
ValueErrorr3 OSErrorIOError) _lve_filelve_file_stream lve_ver_stres r get_lve_versionrQ Y s9
(
)S7
3
3
3 )..q11K
;,,S11!455 z:6
%O;OOIOO
W ( ( ( a a'(sW B! B'A B! B6B BBBB! B B! !C2CCc z | p
t } | }n# t $ r d}Y nw xY w|pdS )a
Parse governor config file and return governor mode
:type governor: MySQLGovernor
:rtype: str
:return:
if file exists and correct: "on", "single", "off", "abusers" or "all"
if governor is not installed or error occurs: "none"
Nnone)r get_governor_moder )governor
governor_modes r rT rT k sY *=??H 2244
"F"s ' 66c t j d sdS t j t rt } nt } | S )z
:rtype: str|None
z/usr/sbin/db_governorN)ospathexistsGOVERNOR_CONFIGGOVERNOR_CONFIG_OLD)governor_config_paths r get_governor_config_pathr^ | sK
7>>1 2 2 t w~~o&& 3 . 2 r c d t }|dS t j | }| d D ]S} |j d j | k r*|j d j dk c S D# t $ r Y Pw xY wn # t t t f$ r Y dS w xY wdS )z3
:rtype: bool|None
:type username: str
NusernamemodeignoreF) r^ r parsegetElementsByTagName
attributesvaluelowerKeyErrorrL r r )usernamer] governor_configr` s r get_governor_ignore_for_userrl s
455#t
!-(<==#88@@ D
?6*0H<< ?628>>@@HLLLL =
Z . tt 5s; +B =B=B B
B B
BB B-,B-c e Zd ZdS )DomainExceptionN)r: r; r< r r r rn rn s Dr rn c F t } | S )z/
:rtype: (str, MySQLGovException|None)
)r get_governor_status)mysql_governors r rp rp s #__N - - / //r c j t j | S # t t t f$ r i cY S w xY w)z
Get dict[user, domain], empty if cannot obtain.
:param reseller_name: reseller's name
:rtype: dict[str, str|None]
)r
reseller_domainsr r AttributeError)
reseller_names r get_reseller_domainsrv sC %m444 & n E s 22Tc t j | | }|S # t t t t j j t j j t f$ r+}|r t t | |g cY d}~S d}~ww xY w)z
Get list of aliases for specified user's domain, or empty list.
:type username: str
:type domain: str
:type raise_exception: bool
:raises DomainException: if cannot obtain domains
:rtype: list[str]|None
N) r
useraliasesri rL rH cpapiexceptionsr NoPanelUserr rn str)rj domainraise_exception user_aliasesrP s r get_aliasesr s (6::
gy%*?*L ! -| = 1!#a&&))q 0 s =B A<6B<Bc t j | }d |D S # t t t t j j t j j t f$ r+}|r t t | |g cY d}~S d}~ww xY w)z
Get list of domains for specified user, or None.
:type username: str
:type raise_exception: bool
:raises DomainException: if cannot obtain domains
:rtype: list[str]|None
c g | ]
}|d S )r r ).0r| s r
z get_domains.. s 555fq 555r N) r
userdomainsri rL rH ry r rz r rn r{ )rj r} user_domainsrP s r get_domainsr s 6 (22 65 5555
gy%*?*L ! -| = 1!#a&&))q 0 s " =B
B?B
B
c Z t | }t | dk rdS |d S )z
:param username: user name string
:return:
if not supported in CP, or domain name is not set: None
if we can find it: domain name string
:rtype: str|None
:type username: str
r N)r r- )rj r s r
get_domainr s3 x((L
<A t ?r c | }| d | d| | dS )zP
Create database
:param db_engine:
:param db_name:
:return:
COMMITzCREATE DATABASE N)connectexecuteclose) db_enginedb_nameconns r
init_databaser sS DLLLL-G--...JJLLLLLr
tmp_unittest_c | |t t j dd z }t | | |S )z
Create database with random prefix in name
Using for unittests
:param db_engine:
:param str prefix:
:return str:
r )r{ uuiduuid4r )r prefixr s r init_database_rndr s: s4:<<((1--G)W%%% Nr c ( t | }|j dd }|r|j dk rt j | dS t j |j dz |j z }| }|rd|j v r|
d| d |
d d
d
}|
d d
}t t t |
d d
dgk r|
d| d n{|
d| d na|r_d|j v rV|
d }| D ],}|d |k r |
d|d
-|
d |
d| | dS dS )z
Drop database (support MySQL and PostgresSQL)
:param uri:
:param force: kill all connection and locked session before drop database
:return:
NsqliteTz://
postgresqlz?UPDATE pg_database SET datallowconn = 'false' where datname = ''zSHOW server_versionr . zLSELECT pg_terminate_backend(procpid) FROM pg_stat_activity WHERE datname = 'zHSELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = 'mysqlzSHOW PROCESSLIST zKILL r zDROP DATABASE )r rY schemerX unlinksqlalchemy_engine
create_enginenetlocr ra r fetchallstriprG listmaprF r ) uriforceuri_r r r pg_vercursorfetcheds r
drop_databaser s C==DimG ;( " " Ig 4%3DK%4G$+4UVV "" 7\Y^33 LLe[beee f f f\\"788AACCAFqIF\\^^))#..q1FCV\\# . .//00Aq699 vls v v vwwww rho r r rssss
7w).00\\"455F!??,,
7
71:((LL!5!5!5666 X /g//000
t9 r c d} t j | sdS t | d 5 }| cd d d S # 1 swxY w Y d S )Nz/etc/redhat-releaser4 r? r@ )rX rY rZ rD readline)rY release_files r
_release_liner 1 s D
7>>$ r
dW % % % ' $$&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's AA Ac 6 | t } | }d|v r| d dz }ng S || }g }| d D ]5} | t | &# t
$ r Y 2w xY w|S )a
Return CloudLinux version as list [major, minor]
or [] if can't obtain version
:param str release_line:
:return list: splited version
>>> cl_ver('CloudLinux Server release 5.11 (Vladislav Volkov)')
[5, 11]
>>> cl_ver('CloudLinux Server release 6.8 (Oleg Makarov)')
[6, 8]
>>> cl_ver('CloudLinux release 7.2 (Valeri Kubasov)')
[7, 2]
>>> cl_ver('Incorrect release line')
[]
Nreleaser r )r rG indexappendrF rJ ) release_linerelease_line_splitedver_indxver_strverss r cl_verr 9 s $ '--//((('--i881< "8,G
C
]]3
JJs1vv D Js &"B
BBc J t j d } fd|D S )N)cploginreseller)keylsc & g | ]
\ }}|k |S r r )r r reseller_r s r r z#_reseller_users..[ s( Y Y Y*8yCXCXGCXCXCXr )r
cpinfo)r cplogin_reseller_s ` r _reseller_usersr Y s2 +BCCC Y Y Y Y.? Y Y YYr c t t dt } || S # t j j $ r g cY S w xY w)Nreseller_users)getattrr
r ry CPAPIException)r r s r get_users_for_resellerr ^ sS U$4oFFN ~h''' / s
( A Ac t | dd 5 t j | | ddd n # 1 swxY w Y t j | d dS )z
The equivalent console touch command
:param str fname: path to file
:param None| tuple times: (atime, mtime) if None atime and mtime are default
r! r? r@ N )rD rX utimechmod)fnametimess r touchr f s
eS7 + + +
HUEs 599c 2 fd}t | | dS )z2
:type data: list
:type filename: str
c Z t j | }| dS )z:type csv_file: fileN)csvwriter writerows)csv_filer datas r csv_writez#atomic_write_csv..csv_writev s, H%% r N
_atomic_write)filenamer r s ` r atomic_write_csvr q s4
(I&&&&&r c 2 fd}t | | dS )z1
:type data: str
:type filename: str
c 2 | dS )z:type str_file: fileN)write)str_filer s r str_writez#atomic_write_str..str_write s t r Nr )r r r s ` r atomic_write_strr } s4
(I&&&&&r c d\ }}d}t j ddt j | 5 }|j } || | t | D ]W} t j | n/# t $ r }|}t j | Y d}~Pd}~ww xY w| ddd n # 1 swxY w Y t j
|d t j || dS )zF
:type write_function: (file) -> None
:type filename: str
)r g333333?NwF)deletedirr )tempfileNamedTemporaryFilerX rY dirnamera flushr, fsyncfilenorK timesleepr rename) r write_function retry_countretry_intervalexcf tmp_filename_rP s r r r so
#) K
C $S,1)+ )B)B
D
D
D GHv q {## A
+$$$
+
+
+
>********
+ I H\5!!!IlH%%%%%s; 8C5&B C
C'C=CCCC C c 8 e Zd ZddZed Zd Zd ZdS ) reboot_lockNc || _ | t | _ n|| _ t j | j t j t j z d | _ d S )Ni ) timeoutr _build_lock_filenamer rX rD O_WRONLYO_CREATlock)r8 r r s r r7 zreboot_lock.__init__ sN =$99;;DJJ DJGDJ bj(@%HH r c d} t j | st | t j | d S )Nz/var/lvez
lvestats.lock)rX rY rZ mkdir_pjoin)
lock_file_dirs r r z reboot_lock._build_lock_filename sA "
w~~m,, # M " " "w||M?;;;r c n t j } t j | j t j t j z d S # t $ r`}|j t j k r | j %t j |z
| j k rY d }~d S t j
d Y d }~nd }~ww xY w)NTg?) r fcntlflockr LOCK_EXLOCK_NBrL errnoEAGAINr r )r8
start_timerP s r __enter__zreboot_lock.__enter__ s Y[[
$
$ DIu}u}'DEEE
$
$
$7el**|/DIKK*4Lt|4[4[ JsOOOOOOOO
$ $s 1A
B3rV s
( ( ( ( ( ( ! ! ! ! ! ! ) ) ) ) ) ) ) ) ( ( ( ( ( ( 2 2 2 2 2 2 # # # # # # . . . . . . . . / / / / / / # # # # # # ( ( ( ( ( (
: 9 9 9 9 9 9 9 M M M M M M M M
69
( ( ( (I I I I Ii I I I
' ( ( ( ($# # # #" , ( 0 0 0 (6 6 6 6&
% % % %P' ' ' @Z Z Z Z
' ' '' ' '& & &8& & & & && & & &R # # #
( ( ( ( K# K$ K K K K 04())* $tCy/ I , "% $' 03 r