4e L N d dl Zd dlmZmZ d dlmZmZmZ G d de ZdS ) N)Modelinstance)datetimetimezone 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.Metaanalyst_cleanup_requestsN)__name__
__module__ __qualname__r dbdatabasedb_table Z/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/model/analyst_cleanup.pyMetar s ;-r r F)null)r defaultpendingz/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& zz *+
r 2 r c | | j |k | j | | S )z$Get all requests for a specific user)selectwherer order_by
created_atdesclimitoffset)r r r) r* s r get_user_requestsz'AnalystCleanupRequest.get_user_requests) sY
JJLL
U3<8+
,
,
Xcn))++
,
,
U5\\
VF^^
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
JJLL
Xcn))++
,
,
U5\\
VF^^
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 statusin_r) firstr )r r active_requests r get_active_request_linkz-AnalystCleanupRequest.get_active_request_link> ss
JJLL
U ):>>9m"<==? U1XX
%'' .<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
JJj|J D D
U3>Z/
0
0
WYY
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 " hl33iQ6G6G6GG$++ ! * ! , ! ( ! .
%
"
)
-
-y-.H
I
I&-<(5 GI
r )r" r ) r
r r __doc__r pw AutoFieldid CharFieldr r TextFieldr TimestampFieldr rA r rB r' Checkr2 r8 classmethodr! r+ r- strr6 r< rD r r r r r s
. . . . . . . .
Br|'''H 5)))J ",E***K""
LHL 6 6 J R\
BHF G G
F %2$
LHL 6 6 L
[
[
[
F#* F F F [F"
[
[
r r ) peeweerF defence360agent.modelr r r r r r r r r rQ s 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2f
f
f
f
f
E f
f
f
f
f
r