관리-도구
편집 파일: awp_provision_api.cpython-311.pyc
� b��f� � �� � 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 fd�Z! G d� d� � Z"dS )� N)�Any)�Schema� SchemaError)�wraps)�Callable)�get_user_auth_key)� userdomains�cpusers)� find_wp_paths)�gettext� )� get_client)�pullzone_info_schema)� XRayAPIError� XRayError)� timestamp)�get_xray_exec_user�funcc � � �� dt dt dt dt fd��t � � � � �fd�� � }|S )N� account_id�domain�website�usernamec �< � 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_user�user�auth_key�domains_docroot_list� domain_item�wordpress_list�wp_sites �U/opt/cloudlinux/venv/lib64/python3.11/site-packages/xray/adviser/awp_provision_api.py�validatez%validate_users_data.<locals>.validate sH � � (�)�)� � � f�!�H�H� � f� ����O�O� f� f�D� 0�� 6� 6�H� �:�-�-�#'�� �� .� $�A�&c�$d�$d�e�e�e� *�8� 4� 4� �/� X� X�K��1�~�� � � �B�!?�!?�!G�!G� �SU�!V�!V�V�V�!%�m�K��N�&C�&C�!D�!D��!�/�/�#�r�2�2���.�0�0�#�$_�~�$_�$_�`�`�`���� W� �A� U� V� V�W�W� W� c � �� |� d� � }|� d� � } �||d |d |� � �| i |��}|S )Nr r r r )�get)�args�kwargsr r �responser r) s ��r( �wrapperz$validate_users_data.<locals>.wrapper> s^ �� ��Z�Z� �-�-� ��:�:�j�)�)�����V�H� -�v�i�/@�(�K�K�K��4��(��(�(���r* )�strr )r r0 r) s ` @r( �validate_users_datar2 st ��� � X�S� X�#� X�� X�s� X� X� X� X�@ �4�[�[�� � � � � �[�� �Nr* c � � e Zd ZdZd� Zdd�Zedefd�� � Zde de de fd �Z e dd �� � Z e dd �� � Ze dd �� � Zdd �Zdd�ZdS )�AWPProvisionAPIz Awp provision service handlersc �t � t j d� � | _ t d� � } |� � | _ d S )N�awp_provision_api� awp_provision)�logging� getLogger�loggerr �awp_client_api)�self�awp_client_objects r( �__init__zAWPProvisionAPI.__init__L s: � � �'�(;�<�<�� � &�� 7� 7�� /� /� 1� 1����r* Nc � � |s� t j � d� � }|r&t j t |� � � � j }t |� � S # t $ r"}t t d� � � � |�d }~ww xY w|S )N� XRAYEXEC_UIDzAuth file was not found!) �os�environr, �pwd�getpwuid�int�pw_namer � Exceptionr r )r< r r r! �es r( � _account_idzAWPProvisionAPI._account_idR s� � �� I� I� �Z�^�^�N� ;� ;� � � E� "�|�C� �O�O� <� <� D�H�(��2�2�2�� � I� I� I�"�1�%?�#@�#@�A�A�q�H����� I�����s �AA � B�$ B�B�returnc �� � dt � � d�}| r|� | � � t j t t |� � � � � � � � � S )N�success)�resultr )r �update�json�dumps�dict�sorted�items)r. �initials r( r/ zAWPProvisionAPI.response] sU � � &�Y�[�[�A�A�� � #� �N�N�6� "� "� "��z�$�v�g�m�m�o�o� 6� 6�7�7�8�8�8r* �data�schemac � � |� |� � 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: %s���zMalformed API response: %sN) r) r r: �error�errors�autosr r r1 )r< rU rV rH �msgs r( � _validatezAWPProvisionAPI._validated s� � � K��?�?�4�(�(� (��� K� K� K� �K� � � C�T� J� J� J��(�2�,�-�!�'�"�+�C� �q�!=� >� >��S��� I�J�J� J����� K���s � � B�A!B�Bc �� � | � ||� � }| 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_pullzonem sb � ��%�%�j�(�;�;� ��"�6�6�z�6�7�S�S���>�>�t�)=� "� ?� ?���}�}�(�}�+�+�+r* c � � | � ||� � }| j � |||� � }| � |�� � S �Nr_ )rI r; �remove_pullzoner/ �r< r r r r r/ s r( rd z AWPProvisionAPI.remove_pullzoneu �E � ��%�%�j�(�;�;� ��&�6�6�z�6�7�S�S���}�}�(�}�+�+�+r* c � � | � ||� � }| j � |||� � }| � |�� � S rc )rI r; �purge_cdn_cacher/ re s r( rh z AWPProvisionAPI.purge_cdn_cache{ rf r* c � � |p| � ||� � g}| j � |� � }| � |�� � S rc )rI r; � sync_accountr/ �r< r r r/ s r( rj z AWPProvisionAPI.sync_account� sI � � �K�D�$4�$4�Z��$J�$J�#K� ��&�3�3�J�?�?���}�}�(�}�+�+�+r* c � � | � ||� � }| j � |� � }| � |�� � S rc )rI r; � get_usager/ rk s r( rm zAWPProvisionAPI.get_usage� sA � ��%�%�j�(�;�;� ��&�0�0��<�<���}�}�(�}�+�+�+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� � � � � �9�c� 9� 9� 9� �\�9� K�c� K�6� K�c� K� K� K� K� �,� ,� ,� ��,� �,� ,� ,� ��,� �,� ,� ,� ��,� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,r* r4 )#r8 rO rA rC �typingr rV r r � functoolsr r � clwpos.papir �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( �<module>r s �� ���� � � � � � � � � � � � � � � � � � � &� &� &� &� &� &� &� &� � � � � � � � � � � � � )� )� )� )� )� )� /� /� /� /� /� /� /� /� -� -� -� -� -� -� � � � � � � "� "� "� "� "� "� 4� 4� 4� 4� 4� 4� 9� 9� 9� 9� 9� 9� 9� 9� &� &� &� &� &� &� ;� ;� ;� ;� ;� ;�*�h� *� *� *� *�X@,� @,� @,� @,� @,� @,� @,� @,� @,� @,r*