qj d dl Zd dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
d dl m Z d dl
mZmZ d dlmZ d dlmZ ej e Z ed Zd
Zd Z G d d
e
ZdS ) N)datetime timedelta)API)ANTIVIRUS_MODE)IndependentAgentIDAPIIAIDTokenError)run_in_executor_decorator) parse_params
)minutesno_agent_tokenc P t j | }nK# t t t j j f$ r'}t d| i cY d }~S d }~ww xY wt |t r|S t dt | j i S )Nz#Cannot decode API response body: %sz&API response body is %s, not an object)
jsonload
ValueErrorOSErrorhttpclient
HTTPExceptionloggerwarning
isinstancedicttype__name__)responsebodyes _/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/api/server/analyst_cleanup.py _json_objectr s y"" !: ; )H)J)J#J#J#J#J#J#JK - G --g66666666F#)CJx &.lnnCJ{ #M
55 s
AC% %
C32C3c | | }| dd S # t $ r&}t d| Y d}~dS d}~ww xY w)&Execute the actual request in executorr# Fz&Failed to check cleanup permission: %sNT)r1 get Exceptionr errorr8 r1 r# r s r r7 z AnalystCleanupAPI._check_allowedQ so [[))F::h.. . LLA1 E E E
44444
s *-
A AA idsreturnc K t |t r d d |D }nt | }| j | j }d|i} t j t || dt j d{V id }n0# t $ r#}t d | d}~ww xY w| | d{V S )
a
Retrieve tickets from Zendesk API using the show_many endpoint
Args:
ids (list or str): List of ticket IDs or comma-separated string of IDs
Returns:
list: List of dictionaries with 'id', 'status', and 'updated_at' fields
,c 3 4 K | ]}t | V d S )N)str).0_ids r z0AnalystCleanupAPI.get_tickets..m s( 7 7Cs3xx 7 7 7 7 7 7 r& rA r( Nr) r* z&Failed to get IAID token for tickets: )r listjoinrF SHOW_MANY_URL_TEMPLATEr4 r5 r0 r1 r2 r
r r6 r r r? _execute_get_tickets)r8 rA ids_strurlparamsr1 r s r get_ticketsz AnalystCleanupAPI.get_tickets` s, c4 hh 7 73 7 7 777GG#hhG (//S]/CC! n,, VS))!)>)H)J)J#J#J#J#J#J#JK - GG
LLE!EE F F F --g666666666s ,A B6 6
C# C C#c g } | | }| dg D ]T}| | d | d | d d U||c S # t $ r'}t d| Y d}~nd}~ww xY w |S # |c cY S xY w)z2Execute the actual get_tickets request in executorticketsidstatus
updated_at)rU rV rW zFailed to get tickets: N)r1 r= appendr> r r? )r8 r1 simplified_ticketsr# ticketr s r rN z&AnalystCleanupAPI._execute_get_tickets s &[[))F !**Y33
")) $jj . ."(**X"6"6&,jj &>&> & & % % % 8 8 8 LL6166 7 7 7 7 7 7 7 7 8 7% %% % % % % % % % %s* BB
B: B50C 5B::C Cc K t j | j | j t
j d{V ddt j d|i
d }| | d{V }|S # t $ r t d i cY S w xY w) z'Check if email is registered in Zendeskr& Napplication/jsonr( z Content-Typecustomer_emailPOSTr+ datar, zGot IAIDTokenError)r0 r1 r2 IS_REGISTERED_URL_TEMPLATEr4 r5 r r6 r dumpsencode_register_statusr r r? )r8 emailr1 r# s r check_registeredz"AnalystCleanupAPI.check_registered s n,,.553=5II$9$C$E$E E E E E E E$6 Z!15 9::AACC - G //88888888FM LL- . . .III s BB" "&C
C c | | }|S # t $ r'}t d| i cY d}~S d}~ww xY w)r< z&Failed to check email registration: %sN)r1 r> r r? r@ s r re z"AnalystCleanupAPI._register_status s` [[))FM LLA1 E E EIIIIII s
A
AA
A
c K t j d{V }n# t $ r ddt ifcY S w xY wt j | j | j |ddt j |||t rdndd
d
}| | d{V S ) zAsk the backend to open a support ticket.
Return an (HTTP status, response body) pair; the status is None when
the backend could not be reached at all.
Nmessager& r\ r]
pr_imunify_avpr_im360)rf subject descriptionproductr_ r` )r r6 r NO_AGENT_TOKENr0 r1 r2 CREATE_TICKET_URL_TEMPLATEr4 r5 r rc r rd _send_create_ticket)r8 rf rm rn tokenr1 s r
create_ticketz AnalystCleanupAPI.create_ticket s 5/9;;;;;;;;EE 5 5 5)^44 4 4 4 5 .(( * 1 1s} 1 E E 2 " &#.+9Iz fhh! )
$ ,,W555555555s 66c t j || j 5 }|j t | fcddd S # 1 swxY w Y dS # t j j $ r)}|j |j t | ni fcY d}~S d}~wt $ r)}t d| di fcY d}~S d}~ww xY w)r< )timeoutNz*Failed to reach create-ticket endpoint: %s)
r0 r1 urlopen_SOCKET_TIMEOUTrV r r? HTTPErrorcodefpr> r r )r8 r1 r r s r rr z%AnalystCleanupAPI._send_create_ticket s4 ''!4 (
? X(>(>>
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
? |% G G G6ad.><???BF F F F F F F NNG K K K8OOOOOO sQ &A A A AA AA C, B
C
C C;CC)r
__module__ __qualname__r3 rM rb rq r minr. classmethodr: r r7 rF r rR rN rg re rt rr rJ r r" r" ' si 8 ! K; < \ F
[. [ 7SE 7tf 7 7 7 [ 7@ & & [&, [& [ 6 6 [ 6> [ rJ r" ) http.clientr r urllib.errorr0 urllib.requestloggingr r defence360agent.api.serverr defence360agent.contracts.configr defence360agent.internals.iaidr r defence360agent.rpc_tools.utilsr defence360agent.utils.supportr
getLoggerr r r/ rp r r" r rJ r r s> ( ( ( ( ( ( ( ( * * * * * * ; ; ; ; ; ; F E E E E E 6 6 6 6 6 6 8 $ $ Ib ! ! ! " ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ rJ