0x1998 - MANAGER
Düzenlenen Dosya: analyst_cleanup.cpython-311.pyc
� ���i� � �N � d dl Zd dlmZmZ d dlmZmZmZ G d� de� � ZdS )� N)�Model�instance)�datetime�timezone� timedeltac � � e Zd ZdZ G d� d� � Z ej � � Z ej d�� � Z ej d�� � Z ej d�� � Z ej d ej ej � � �� � Z ej dd ej d� � g� � � Z ej d ej ej � � �� � Zed � � � Zedd �� � Zedd�� � Zededz fd�� � Zed� � � Z ed� � � Z dS )�AnalystCleanupRequestz Model for storing analyst cleanup requests. Tracks request details and status for each cleanup request submitted. c � � e Zd Zej ZdZdS )�AnalystCleanupRequest.Meta�analyst_cleanup_requestsN)�__name__� __module__� __qualname__r �db�database�db_table� � �Z/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/model/analyst_cleanup.py�Metar s � � � � � ��;��-���r r F)�null)r �default�pendingz/status in ('pending','in_progress','completed'))r r � constraintsc �2 � | � |||�� � S )z Create a new cleanup request)�username� zendesk_id� ticket_link)�create)�clsr r r s r �create_requestz$AnalystCleanupRequest.create_request" s& � � �z�z� �*�+� � � � r �2 r c � � | � � � � | j |k � � � | j � � � � � � |� � � |� � S )z$Get all requests for a specific user)�select�wherer �order_by� created_at�desc�limit�offset)r r r) r* s r �get_user_requestsz'AnalystCleanupRequest.get_user_requests) sY � � �J�J�L�L� �U�3�<�8�+� ,� ,� �X�c�n�)�)�+�+� ,� ,� �U�5�\�\� �V�F�^�^� r c �� � | � � � � | j � � � � � � |� � � |� � S )z Get all requests for a sever)r$ r& r' r( r) r* )r r) r* s r �get_all_requestsz&AnalystCleanupRequest.get_all_requests4 sE � � �J�J�L�L� �X�c�n�)�)�+�+� ,� ,� �U�5�\�\� �V�F�^�^� r �returnNc � � | � � � � | j |k | j � ddg� � z � � � d� � � � � }|r|j ndS )z� Gets user requests for a user and checks if there are requests with [pending | in_progress] state. If found, returns ticket_link, otherwise returns None r � in_progress� N)r$ r% r �status�in_r) �firstr )r r �active_requests r �get_active_request_linkz-AnalystCleanupRequest.get_active_request_link> ss � � �J�J�L�L� �U�� ��)��:�>�>�9�m�"<�=�=�?�� � �U�1�X�X� �%�'�'� � .<�E�~�)�)��Er c � � | � ||�� � � | j |k � � � � � S )z Update the status of a request)r2 � last_updated)�updater% r �execute)r r � new_statusr8 s r � update_statusz#AnalystCleanupRequest.update_statusP s9 � � �J�J�j�|�J� D� D� �U�3�>�Z�/� 0� 0� �W�Y�Y� r c � � t j t j � � t d�� � z }t � t j t j t j t j � � � t j � ddg� � t j dk t j |k z z � � S )z� Returns a query to fetch active cleanup requests and recently completed requests for the specified users. � )�daysr r0 � completed) r �nowr �utcr r r$ r r r2 r8 r% r3 )r �three_days_agos r �get_all_relevant_requestsz/AnalystCleanupRequest.get_all_relevant_requestsY s� � � "� �h�l�3�3�i�Q�6G�6G�6G�G��$�+�+� !� *� !� ,� !� (� !� .� � � �%� "� )� -� -�y�-�.H� I� I�&�-��<�(�5� �G�I� � � � r )r" r ) r r r �__doc__r �pw� AutoField�id� CharFieldr r � TextFieldr �TimestampFieldr rA r rB r' �Checkr2 r8 � classmethodr! r+ r- �strr6 r< rD r r r r r s� � � � � � �� � .� .� .� .� .� .� .� .� �� � � �B��r�|��'�'�'�H� �� �5�)�)�)�J� �"�,�E�*�*�*�K�"��"� �L�H�L�� � 6� 6�� � �J� �R�\� �� �B�H�F� G� G� �� � �F� %�2�$� �L�H�L�� � 6� 6�� � �L� � � � �[� � � � � � �[� � � � � � �[� � �F�#��*� F� F� F� �[�F�" � � � �[� � � � � �[� � � r r ) �peeweerF �defence360agent.modelr r r r r r r r r �<module>rQ s� �� � � � � 1� 1� 1� 1� 1� 1� 1� 1� 2� 2� 2� 2� 2� 2� 2� 2� 2� 2�f � f � f � f � f �E� f � f � f � f � f r
geri dön