Wjs 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 R t }|r|}nU|sSt t D ]}t | }|| k r|} n t t d t
| }|D ]}|d | dd dd k r\t t |d } | d r
|dd n|}
|
| vrt d | dS 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:// /Nz(Passed website is not found, all sites: z5Passed domain cannot be found in list of user domains)
r listr
r r _r replacer
startswith) 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 s[ ())
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 *1););C)@)@M'!""++g.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..wrapperA s^ ZZ --
::j))VH -vi/@(KKK4(((r+ )strr )r r1 r* s ` @r) validate_users_datar3 st !XS !X# !X !Xs !X !X !X !XF 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__P 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_idV 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) r0 zAWPProvisionAPI.responseh 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 r2 )r= rV rW rI msgs r) _validatezAWPProvisionAPI._validateo s K??4(( ( K K K K CT J J J(2,-!'"+C q!= > >S IJJ J Ks
BA!BBc | || }| j ||| }| |t }| | S )N)rV rW rV )rJ r< get_create_pullzoner^ r r0 )r= r r r r rV r0 s r) get_pullzonez AWPProvisionAPI.get_pullzonex s[ %%j(;;
"66z67SS>>t4H>II}}(}+++r+ c | || }| j ||| }| | S Nr` )rJ r< remove_pullzoner0 r= r r r r r0 s r) re z AWPProvisionAPI.remove_pullzone E %%j(;;
&66z67SS}}(}+++r+ c | || }| j ||| }| | S rd )rJ r< purge_cdn_cacher0 rf s r) ri z AWPProvisionAPI.purge_cdn_cache rg r+ c t j d r| | g}n|p| || g}| j | }| | S )NrA )r r` )rB rC r- rJ r< sync_accountr0 r= r r r0 s r) rk z AWPProvisionAPI.sync_account s{
:>>. ) ) P **H*==>JJ#O(8(8X(N(N'OJ&33J??}}(}+++r+ c | || }| j | }| | S rd )rJ r< get_usager0 rl s r) rn zAWPProvisionAPI.get_usage sA %%j(;;
&00<<}}(}+++r+ )NN)N)__name__
__module__ __qualname____doc__r? rJ staticmethodr2 r0 r r r^ r3 rb re ri rk rn r+ r) r5 r5 M s ((2 2 2 $ 9c 9 9 9 \9 Kc K6 Kc K K K K , , , , , , , ,
, , , ,
, , , ,, , , , , ,r+ r5 )#r9 rP rB rD typingr rW 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 r3 r5 rt r+ r) r s
& & & & & & & & ? ? ? ? ? ? / / / / / / / / - - - - - - " " " " " " 4 4 4 4 4 4 9 9 9 9 9 9 9 9 & & & & & & ; ; ; ; ; ;-h - - - -`I, I, I, I, I, I, I, I, I, I,r+