관리-도구
편집 파일: lookup.cpython-311.opt-1.pyc
� IMg� � �2 � d dl Z d dlZd dlmZmZmZmZmZ d dlm Z d dl m Z ddl m Z dZ G d� d e� � Z G d � d e� � Z G d � d � � Z G d� de� � Z G d� de� � Z G d� de� � Ze j efz Zee j fd�Zd� ZdS )� N)�Any�Callable�Dict�Set�Tuple)�UserType)�Scope� )�RpcError� __rpc_commandc � � e Zd ZdS )�DuplicateHandlerErrorN��__name__� __module__� __qualname__� � �U/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/rpc_tools/lookup.pyr r � � � � � � ��Dr r c � � e Zd ZdS )�NotCoroutineErrorNr r r r r r r r r c �� � � e Zd ZdZej Z e� � Ze j i e j i iZ g Z � fd�Zed� � � Zd� Zee j fdefd�� � Zed d�� � Zed � � � Z� xZS ) � Endpointsz\Endpoints class implements registration and lookup for functions implementing RPC calls.c �n �� t � � j di |�� | j � | � � d S )Nr )�super�__init_subclass__� _subclasses�append)�cls�kwargs� __class__s �r r zEndpoints.__init_subclass__! s< �� �!����!�+�+�F�+�+�+� �� � �s�#�#�#�#�#r c �x � g }| j D ]/}t j |d� � � }|r|� |� � �0|S )Nc �. � t | t d � � S �N)�getattr� _RPC_MARK)�items r �<lambda>z0Endpoints.get_active_endpoints.<locals>.<lambda>+ s � �W�T�9�d�%C�%C� r )r �inspect� getmembersr )r �active_endpoints�subcls� rpc_handlerss r �get_active_endpointsz Endpoints.get_active_endpoints% s[ � � ���o� 0� 0�F�"�-��C�C� � �L� � 0� �'�'��/�/�/�� � r c � � || _ d S r% )�_sink)�self�sinks r �__init__zEndpoints.__init__1 s � ��� � � r �returnc � �. K � |d }t |� � }|| j | vr+t dd� |d � � z � � �| j | | \ }}t ||� � |� � } |di |d ��� d{V ��S )a: Find appropriate class and function within that class that implements processing for request based on supplied 'command' within. Call that (async) function and return its result. If target class/function for given request['command'] is not found then RpcError exception is raised.�commandz&Endpoint not found for RPC method "%s"� �paramsNr )�tuple�_Endpoints__COMMAND_MAPr �joinr& ) r �requestr3 �userr7 �key� cls_handler� handler_name�handlers r �route_to_endpointzEndpoints.route_to_endpoint4 s� � � � � �)�$���G�n�n�� �c�'��-� -� -��8��(�(�7�9�-�.�.�/�� � � %(�$5�d�$;�C�$@�!� �\��+�+�d�+�+�\�:�:�� �W�1�1�w�x� 0�1�1�1�1�1�1�1�1�1r Nc � � t | � � D ]�}|� d� � r�t | |� � }t |t d� � }|��At j |� � st d� � �| j D ][}|| j | v r8d� ||| j | | |� � }t |� � �| |f| j | |< �\��dS )z{Registers RPC handlers for all functions within a class. Functions should be decorated with @bind('command', ...).�_NzMust be a coroutinez1Duplicate handlers for command {} ({}): {} and {}) �dir� startswithr& r' r* �iscoroutinefunctionr �APPLICABLE_USER_TYPESr; �formatr )r �name�attrr7 � user_type�msgs r �register_rpc_handlersz Endpoints.register_rpc_handlersH s � � ��H�H� D� D�D����s�#�#� ���3��%�%�D� �d�I�t�4�4�G�����.�t�4�4� ?�'�(=�>�>�>� � 6� D� D� ��c� /� � :�:�:�K� �� #� %� � -�i� 8�� A� � � � � 0��4�4�4�9<�d� ��!�)�,�W�5�5� D� D� Dr c �P � t j t j hD ] }i | j |< � dS )z)Clears all previously made registrations.N)r �NON_ROOT�ROOTr; )r rM s r �reset_rpc_handlersz Endpoints.reset_rpc_handlersd s7 � � #�+�X�]�;� .� .�I�+-�C� �i� (� (� .� .r )r5 N)r r r �__doc__r �AV_IM360�SCOPE�setrI r rR rQ r; r r � classmethodr/ r4 r rC rO rS � __classcell__)r" s @r r r s �� � � � � � � � �N�E� �C�E�E��� �r���2��M� �K�$� $� $� $� $� � � � �[� �� � � �9A� � 2� 2�3� 2� 2� 2� �[�2�& �D� D� D� �[�D�6 �.� .� �[�.� .� .� .� .r r c �. � e Zd ZdZej ej hZdS )�CommonEndpointsz5Endpoints available both for root and non root users.N)r r r rT r rQ rR rI r r r r[ r[ k s& � � � � � �?�?� %� .�� � >���r r[ c �" � e Zd ZdZej hZdS )� RootEndpointsz'Endpoints available only for root user.N)r r r rT r rR rI r r r r] r] q s � � � � � �1�1� %�]�O���r r] c �" � e Zd ZdZej hZdS )�UserOnlyEndpointsz,Endpoints available only for non root users.N)r r r rT r rQ rI r r r r_ r_ w s" � � � � � �6�6� %� .� /���r r_ c �F � t j t j | ||�� � S )z4Decorator replacing functools.wraps for rpc handlers��wrapped�assigned�updated)� functools�partial�update_wrapperra s r �wrapsrh � s- � � � �� ���� � � � r c � � � � fd�}|S )z4Mark a function as processing RPC calls for command.c �4 �� t | t �� � | S r% )�setattrr' )�funcr7 s �r � decoratorzbind.<locals>.decorator� s �� ���i��)�)�)�� r r )r7 rm s ` r �bindrn � s$ �� �� � � � � �r )re r* �typingr r r r r � defence360agent.contracts.configr �defence360agent.utilsr � exceptionsr r' � Exceptionr r r r[ r] r_ �WRAPPER_ASSIGNMENTS�LOOKUP_ASSIGNMENTS�WRAPPER_UPDATESrh rn r r r �<module>rw s� �� � � � � ���� 2� 2� 2� 2� 2� 2� 2� 2� 2� 2� 2� 2� 2� 2� 5� 5� 5� 5� 5� 5� '� '� '� '� '� '� � � � � � � � � � � � � �I� � � � � � � � � � � � �S.� S.� S.� S.� S.� S.� S.� S.�l?� ?� ?� ?� ?�i� ?� ?� ?� ,� ,� ,� ,� ,�I� ,� ,� ,� 0� 0� 0� 0� 0� � 0� 0� 0� �2�i�\�A� � )�)�2K� � � � �� � � � r