fjB d Z ddlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z d d
l
mZ d d lmZ d d lmZmZ d d
lmZ d dlmZmZmZmZmZm Z d dl m Z G d de
Z dS )z>
This module contains X Ray continuous tracing implementation
N) defaultdict)List)gettext )ContinuousCommon)Mailer )
get_client)regex) XRayError XRayAPIError) url_split) read_sys_id prev_datedate_of_timestampget_formatted_dateget_html_formatted_linksget_text_formatted_links)initialize_managerc e Zd ZdZ fdZd ZdefdZddZe de
de
fd Z de
ddfd
Z de
e fd Zdefd Zdefd
ZddZ xZS )ContinuousTracingz+
Continuous tracing implementation
c t t }t d } || | _ t | | _ t | _ d S )N
auto_tasks) system_id) super__init__r r
clientr manager_instancer mail_sender)selfr
client_object __class__s N/opt/cloudlinux/venv/lib64/python3.11/site-packages/xray/continuous/tracing.pyr zContinuousTracing.__init__ se
MM "<00
#mi888 29 = = !88 c ~ | | | d S )N)clear_zombie_auto_tasks
start_tracing
generate_mail)r s r# __call__zContinuousTracing.__call__( s> $$&&& r$ returnc | j }nP# t $ rC}| j ddt | i t
}Y d}~nd}~ww xY w|S )z%
Get auto tasks list
zFailed to get auto taskserrextraN)r get_tasksr
loggererrorstrlist)r taskses r# current_auto_tasksz$ContinuousTracing.current_auto_tasks- s K))++EE K 8%*CFFO
5
5
5FFEEEEEE s
A)9A$$A)Nc L d | D }|D ]} || dk r| j | | j | D# t $ r2}| j dt | Y d}~{d}~ww xY wdS )z6
Stop and complete running auto tasks
c b i | ],}|j r#|j d k r|j dk |j |j -S ) completedz
*autotracing*) auto_taskstatususertask_id).0ts r#
z=ContinuousTracing.clear_zombie_auto_tasks..= sM = = =a k =./h+.E.E"#&O";"; i #<";";r$ runningz%sN)r6 r stopcompleter r0 r1 r2 )r auto_running_taskst_idr5 s r# r& z)ContinuousTracing.clear_zombie_auto_tasks9 s = = " 5 5 7 7 = = = ' 0 0D
0%d+y88)..t444%..t4444
0
0
0 !!$A////////
0 0 0s A A%%
B!/(B B!domainc | d d}t j t | s t t d | |S )z<
Construct URL for auto tracing: {domain}/*
/z/*z invalid url)striprematchr
ValueError_)rF us r# _urlzContinuousTracing._urlI sO
||C $ $ $xq!! 2 Q} - -q11 1r$ c < | j | dxx dz
cc< dS )z;
Increment execution_count of given domain
execution_countr N) tracing_conf)r rF s r# executedzContinuousTracing.executedS s/
&!"3444944444r$ c d}| D ]}|j dk r | j | |j | |j \# t $ r5}| j
|dt | i Y d}~d}~wt $ r6}| j
|t | |d Y d}~d}~ww xY w|
S )z8
Start auto tasks with status 'running'
zStart auto task failedrA )urlr, r- N)r, continuous_task)get_continuous_tasksr; r
start_autorO original_urlrS rF rL r0 r1 r2 r dump_tracing_configuration)r err_textrV r5 s r# r' z ContinuousTracing.start_tracingY sm , # 8 8 : : : :O %22
:)44 IIo&B C C 5 E E E MM/"89999 " = = =K%%h-2CFFO & = = = = = = = = ) ) )K%%h14Q=L-( -( & ) ) ) ) ) ) ) )) 3 ..000s# 3A33
C0=+B--
C0:,C++C0c t }| D ]5} |j || |j < &# t
$ r Y 2w xY w|S )z[
Map e-mail addresses to urls in existing continuous tracing configuration
)dictrW emailrO rY rL )r url_mail_mapr? s r# retrieve_mail_addrsz%ContinuousTracing.retrieve_mail_addrsp sp vv **,, A
:;' TYYq~ 6 677
s "A
AAc h dt dt fdfd| D S )z6
Select previous date completed tasks
tsr* c d t | t k S # t $ r Y dS w xY w)zO
Verify if task timestamp corresponds to previous date
F)r r TypeError)rb s r#
check_datez:ContinuousTracing.filter_tasks_by_date..check_date s@
(,, ;;
uu
s !
//c J g | ] }|j d k
|j | S )r9 )r; starttime)r> r? re s r#
z:ContinuousTracing.filter_tasks_by_date.. sD ) ) )a-.X-D-D
1;'' .E -D-D-Dr$ )intboolr6 )r re s @r# filter_tasks_by_datez&ContinuousTracing.filter_tasks_by_date| s\
3 4 ) ) ) )''))) ) ) )r$ c
| j d t t }| }| }|D ]y}| |j }|r[|| t |j d |j
i | j d|j | z| D ]J\ }}| j
|dt t! | t# | K| j d dS )zI
Generate e-mails with links and send them to recipients
zGenerating e-mails startedr z"Selected task %s for mailing to %sreport)templatedatelinks
html_linkszE-mails sending finishedN)r0 infor r3 r` rk getrU appendr shared_linkr= itemsr send_mailr r r )r
email_confurl_mail_mappingyesterday_taskstaskmailkvs r# r( z ContinuousTracing.generate_mail sc
5666 &&
3355 3355# 5 5D#''11D
54 '' tx((+T-=>@ @ @ !E!% t5 5 5 $$&& / /DAq & &q8,>,@,@-Ea-H-H2J+,3. 3.
'
/
/
/
/
344444r$ )r* N)__name__
__module__ __qualname____doc__r r) r3 r6 r& staticmethodr2 rO rS r r] r' r` rk r(
__classcell__)r" s @r# r r s9 $ $ $ $ $
D
0 0 0 0 S S \:s :t : : : : 1tDz 1 1 1 1.
T
)d ) ) ) )&5 5 5 5 5 5 5 5r$ r ) r rJ collectionsr typingr xrayr rM commonr mailerr apiclientr
console_utils.validationsr internal.exceptionsr r
internal.typesr internal.utilsr r r r r r managerr r r$ r# r ss
# # # # # # $ $ $ $ $ $ " " " " " " - - - - - - 9 9 9 9 9 9 9 9 & & & & & &K K K K K K K K K K K K K K K K ( ( ( ( ( (M5 M5 M5 M5 M5( M5 M5 M5 M5 M5r$