관리-도구
편집 파일: send_message.cpython-311.pyc
� IMg� � �� � d dl Z d dlZd dlZd dlZd dlmZmZ d dlm Z d dl m Z d dl m Z d dlmZ d dlmZmZ d dlmZ d d lmZmZmZ e e� � Z G d � d ee� � Z G d � d e� � Z dS )� N)�ABC�abstractmethod)� getLogger)�Optional)�Core)�Message)�IndependentAgentIDAPI�IAIDTokenError)�ServerJSONEncoder)�API�APIError� APITokenErrorc �R � e Zd ZdZedefd�� � Zdeddfd�Zdede ddfd �Z dS ) �BaseSendMessageAPIz /api/v2/send-message/{method}�returnc � � K � d S �N� )�self�message_method�headers� post_datas �\/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/api/server/send_message.py� _send_requestz BaseSendMessageAPI._send_request s � � � � �� �resultNc �� � d|vr"t d� |� � � � �|d dk r5t d� |� d� � � � � � �d S )N�statusz unexpected server response: {!r}�okzserver error: {}�msg)r �format�get)r r s r �check_responsez!BaseSendMessageAPI.check_response sk � � �6� !� !��=�D�D�V�L�L�M�M� M� �(� �t� #� #��-�4�4�V�Z�Z��5F�5F�G�G�H�H� H� $� #r �methodr c � � K � t j � � � d {V ��}n$# t $ r}t d|� �� � �d }~ww xY wd|d�}| � |||� � � d {V ��}| � |� � d S )NzIAID token error occurred zapplication/json)z Content-TypezX-Auth)r � get_tokenr r r r# )r r$ r �token�er r s r � send_dataz BaseSendMessageAPI.send_data" s� � � � � B�/�9�;�;�;�;�;�;�;�;�E�E�� � B� B� B� � @�Q� @� @�A�A� A����� B���� /�� � �� �)�)�&�'�9�E�E�E�E�E�E�E�E�� ���F�#�#�#�#�#s � � ?�:�?) �__name__� __module__� __qualname__�URLr �dictr r# �str�bytesr) r r r r r s� � � � � � � )�C�� �� � � � �^� �I�T� I�d� I� I� I� I� $�c� $�e� $�� $� $� $� $� $� $r r c � � e Zd Zej Zddedefd�Zdeddfd�Zde e ddfd �Z d e ddfd �Z d � Z d eddfd�ZdS )�SendMessageAPIN�rpm_ver�base_urlc �z � || _ || _ d| _ d | _ i | _ |r || _ d S | j | _ d S )N� )� _executorr3 � product_name� server_id�licenser4 � _BASE_URL)r r3 r4 �executors r �__init__zSendMessageAPI.__init__2 sF � �!��� �� � ��� ����� � � +� $�D�M�M�M� �N�D�M�M�Mr r8 r c � � || _ d S r )r8 )r r8 s r �set_product_namez SendMessageAPI.set_product_name= s � � (����r r9 c � � || _ d S r )r9 )r r9 s r � set_server_idz SendMessageAPI.set_server_id@ s � �"����r r: c � � || _ d S r )r: )r r: s r � set_licensezSendMessageAPI.set_licenseC s � � �� � � r c � �� K � t j � | j | j � |�� � z ||d�� � }| � || j �� � � d {V ��S )N)r$ �POST)�datar r$ )r< )�urllib�request�Requestr4 r- r! � async_requestr7 )r r r r rH s r r z SendMessageAPI._send_requestF st � � � ��.�(�(� �M�D�H�O�O�>�O� B� B� B���� )� � �� �'�'��$�.�'�I�I�I�I�I�I�I�I�Ir �messagec � � K � d|vrt j � � |d< |j | j |j | j | j d�}t j |t �� � � � � }| � |j |� � � d {V �� d S )N� timestamp)�payloadr3 � message_idr9 �name)�cls) �timerN r3 rO r9 r8 �json�dumpsr �encoder) r$ )r rK � data2sendr s r � send_messagezSendMessageAPI.send_messageO s� � � � � �g� %� %�#'�9�;�;�G�K� � ���|�!�,� � ��%� � � � �J�y�.?�@�@�@�G�G�I�I� ��n�n�W�^�Y�7�7�7�7�7�7�7�7�7�7�7r )NN)r* r+ r, r �DEFAULT_SOCKET_TIMEOUT�_SOCKET_TIMEOUTr/ r= r? r rA r. rC r r rW r r r r2 r2 / s� � � � � � ��1�O� +� +�� +�s� +� +� +� +�)�S� )�T� )� )� )� )�#�x��}� #�� #� #� #� #� �4� �D� � � � �J� J� J�8�'� 8�d� 8� 8� 8� 8� 8� 8r r2 ) rS rR � urllib.errorrG �urllib.request�abcr r �loggingr �typingr � defence360agent.contracts.configr �"defence360agent.contracts.messagesr � defence360agent.internals.iaidr r �defence360agent.utils.jsonr �defence360agent.api.serverr r r r* �loggerr r2 r r r �<module>re sY �� � � � � � � � � � � � � � � � � #� #� #� #� #� #� #� #� � � � � � � � � � � � � 1� 1� 1� 1� 1� 1� 6� 6� 6� 6� 6� 6�� � � � � � � � 9� 8� 8� 8� 8� 8� C� C� C� C� C� C� C� C� C� C� ��8� � ��$� $� $� $� $��c� $� $� $�4.8� .8� .8� .8� .8�'� .8� .8� .8� .8� .8r