Edit File: lookup.cpython-311.pyc
� �*vh� � �" � d dl Z d dlZd dlmZ d dlmZ d dlmZ ddlm 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 e fz Zee j fd�Zd� ZdS )� N)�Any)�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 �requestr/ �userr3 �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_TYPESr7 �formatr )r �name�attrr3 � 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�ROOTr7 )r rI s r �reset_rpc_handlersz Endpoints.reset_rpc_handlersd s7 � � #�+�X�]�;� .� .�I�+-�C� �i� (� (� .� .r )r1 N)r r r �__doc__r �AV_IM360�SCOPE�setrE r rN rM r7 r r � classmethodr+ r0 r r? rK rO � __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 rP r rM rN rE r r r rW rW k s& � � � � � �?�?� %� .�� � >���r rW c �" � e Zd ZdZej hZdS )� RootEndpointsz'Endpoints available only for root user.N)r r r rP r rN rE r r r rY rY q s � � � � � �1�1� %�]�O���r rY c �" � e Zd ZdZej hZdS )�UserOnlyEndpointsz,Endpoints available only for non root users.N)r r r rP r rM rE 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_wrapperr] s r �wrapsrd � s- � � � �� ���� � � � r c � � � � fd�}|S )z4Mark a function as processing RPC calls for command.c �4 �� t | t �� � | S r! )�setattrr# )�funcr3 s �r � decoratorzbind.<locals>.decorator� s �� ���i��)�)�)�� r r )r3 ri s ` r �bindrj � s$ �� �� � � � � �r )ra r& �typingr � defence360agent.contracts.configr �defence360agent.utilsr � exceptionsr r# � Exceptionr r r rW rY r[ �WRAPPER_ASSIGNMENTS�LOOKUP_ASSIGNMENTS�WRAPPER_UPDATESrd rj r r r �<module>rs s� �� � � � � ���� � � � � � � 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
Back to File Manager