fj d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dl m
Z
d dlm Z d dl m
Z
d dlmZmZ d dlmZ d d lmZ d
d lmZ d
d lmZ d
d
lmZm Z d
dl m Z d
dl m Z de fdZ! G d d Z"dS ) N)Any)Schema SchemaError)wraps)Callable)get_user_auth_key) userdomainscpusers)
find_wp_paths)gettext )
get_client)pullzone_info_schema) XRayAPIError XRayError) timestamp)get_xray_exec_userfuncc dt dt dt dt fdt fd }|S )N
account_iddomainwebsiteusernamec < t }|r|}nU|sSt t D ]}t | }|| k r|} n t t d t
| }|D ]}|d | dd dd k rQt t |d } | dd }
|
| vrt d| d
S t t d ) z1
If user owns domain and website
z;Passed account_id does not belong to any user on the serverr zhttps:// zhttp:// /z(Passed website is not found, all sites: z5Passed domain cannot be found in list of user domainsN) r listr
r r _r replacer ) r r r r
proxy_useruserauth_keydomains_docroot_list domain_itemwordpress_listwp_sites U/opt/cloudlinux/venv/lib64/python3.11/site-packages/xray/adviser/awp_provision_api.pyvalidatez%validate_users_data..validate sH ())
f!HH
f OO f fD 0 6 6H :--#' . $A&c$d$deee *8 4 4 / X XK1~
B!?!?!G!G SU!V!VVV!%mKN&C&C!D!D!//#r22.00#$_~$_$_``` W A U V VWW W c | d }| d } ||d |d | | i |}|S )Nr r r r )get)argskwargsr r responser r) s r( wrapperz$validate_users_data..wrapper> s^ ZZ --
::j))VH -vi/@(KKK4(((r* )strr )r r0 r) s ` @r( validate_users_datar2 st XS X# X Xs X X X X@ 4[[ [ Nr* c e Zd ZdZd ZddZedefd Zde de
de fd Z e dd
Z
e dd Ze dd Zdd
ZddZdS )AWPProvisionAPIz Awp provision service handlersc t t j d | _ t d } | | _ d S )Nawp_provision_api
awp_provision)logging getLoggerloggerr awp_client_api)selfawp_client_objects r( __init__zAWPProvisionAPI.__init__L s: '(;<< & 7 7 / / 1 1r* Nc t j d }|re t j t | j }t | S # t $ r"}t t d |d }~ww xY w|s? t | S # t $ r"}t t d |d }~ww xY w|S )N XRAYEXEC_UIDzAuth file was not found!) osenvironr, pwdgetpwuidintpw_namer Exceptionr r )r< r r r! es r( _account_idzAWPProvisionAPI._account_idR s Z^^N33
I
I <J88@(222
I
I
I"1%?#@#@AAqH
I I
I(222
I
I
I"1%?#@#@AAqH
Is/ 4A
B" A??B
B
C# C Creturnc dt d}| r| | t j t t | S )Nsuccess)resultr )r updatejsondumpsdictsorteditems)r. initials r( r/ zAWPProvisionAPI.responsed sU &Y[[AA # NN6 " " "z$vgmmoo 6 677888r* dataschemac | | S # t $ rf}| j d| |j d p |j d }t
t d t | z d}~ww xY w)z&Validate given data using given schemaz#Failed to validate API response: %szMalformed API response: %sN) r) r r: errorerrorsautosr r r1 )r< rU rV rH msgs r( _validatezAWPProvisionAPI._validatek s K??4(( ( K K K K CT J J J(2,-!'"+C q!= > >S IJJ J Ks
BA!BBc | || }| j ||| }| |t }| | S )N)rU rV rU )rI r; get_create_pullzoner] r r/ )r< r r r r rU r/ s r( get_pullzonez AWPProvisionAPI.get_pullzonet sb %%j(;;
"66z67SS>>t)= " ? ?}}(}+++r* c | || }| j ||| }| | S Nr_ )rI r; remove_pullzoner/ r< r r r r r/ s r( rd z AWPProvisionAPI.remove_pullzone| E %%j(;;
&66z67SS}}(}+++r* c | || }| j ||| }| | S rc )rI r; purge_cdn_cacher/ re s r( rh z AWPProvisionAPI.purge_cdn_cache rf r* c t j d r| | g}n|p| || g}| j | }| | S )Nr@ )r r_ )rA rB r, rI r; sync_accountr/ r< r r r/ s r( rj z AWPProvisionAPI.sync_account s{
:>>. ) ) P **H*==>JJ#O(8(8X(N(N'OJ&33J??}}(}+++r* c | || }| j | }| | S rc )rI r; get_usager/ rk s r( rm zAWPProvisionAPI.get_usage sA %%j(;;
&00<<}}(}+++r* )NN)N)__name__
__module__ __qualname____doc__r> rI staticmethodr1 r/ r r r] r2 ra rd rh rj rm r* r( r4 r4 I s ((2 2 2 $ 9c 9 9 9 \9 Kc K6 Kc K K K K , , , , , , , ,
, , , ,
, , , ,, , , , , ,r* r4 )#r8 rO rA rC typingr rV r r functoolsr r !xray.internal.clwpos_safe_importsr clcommon.cpapir r
clcommon.clwpos_libr xrayr r apiclientr apiclient.schemasr internal.exceptionsr r internal.utilsr internal.user_plugin_utilsr r2 r4 rs r* r( r s
& & & & & & & & ? ? ? ? ? ? / / / / / / / / - - - - - - " " " " " " 4 4 4 4 4 4 9 9 9 9 9 9 9 9 & & & & & & ; ; ; ; ; ;*h * * * *XJ, J, J, J, J, J, J, J, J, J,r*