File: //opt/alt/python37/lib/python3.7/site-packages/xray/continuous/__pycache__/tracing.cpython-37.pyc
B
�V�d! � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z dd l
m Z dd
l m
Z
dd lmZmZ dd lmZ dd
lmZmZmZmZmZmZ ddlmZ G dd� de�Z dS )z>
This module contains X Ray continuous tracing implementation
� N)� defaultdict)�List� )�ContinuousCommon)�Mailer� )�
get_client)�regex)� XRayError� XRayAPIError)� url_split)� read_sys_id� prev_date�date_of_timestamp�get_formatted_date�get_html_formatted_links�get_text_formatted_links)�initialize_managerc s� e Zd ZdZ� fdd�Zdd� Zed�dd�Zd d�d
d �Ze e
e
d �d
d��Z e
d d �dd�Z e
e d�dd�Zed�dd�Zed�dd�Zd d�dd�Z� ZS )�ContinuousTracingz+
Continuous tracing implementation
c s: t � �� t� }td�}||d�| _t|�| _t� | _d S )NZ
auto_tasks)� system_id) �super�__init__r
r �clientr �manager_instancer � mail_sender)�selfr Z
client_object)� __class__� �H/opt/alt/python37/lib/python3.7/site-packages/xray/continuous/tracing.pyr s
zContinuousTracing.__init__c C s | � � | �� | �� d S )N)�clear_zombie_auto_tasks�
start_tracing�
generate_mail)r r r r �__call__' s zContinuousTracing.__call__)�returnc
C sT y| j �� }W n@ tk
rN } z"| jjddt|�id� t� }W dd}~X Y nX |S )z%
Get auto tasks list
zFailed to get auto tasks�err)�extraN)r Z get_tasksr �logger�error�str�list)r Ztasks�er r r �current_auto_tasks, s z$ContinuousTracing.current_auto_tasksNc C s� dd� | � � D �}xj|D ]b}y(|| dkr6| j�|� | j�|� W q tk
rx } z| j�dt|�� W dd}~X Y qX qW dS )z6
Stop and complete running auto tasks
c S s0 i | ](}|j r|jd kr|jdkr|j|j�qS )� completedz
*autotracing*)Z auto_task�status�user�task_id)�.0�tr r r �
<dictcomp>< s z=ContinuousTracing.clear_zombie_auto_tasks.<locals>.<dictcomp>�runningz%sN)r+ r �stopZcompleter
r& r' r( )r Zauto_running_tasksZt_idr* r r r r 8 s
z)ContinuousTracing.clear_zombie_auto_tasks)�domainr# c C s* | � d�� d�}t�t|�s&td|��|S )z<
Construct URL for auto tracing: {domain}/*
�/z/*z invalid url)�strip�reZmatchr �
ValueError)r5 �ur r r �_urlH s
zContinuousTracing._urlc C s | j | d d7 < dS )z;
Increment execution_count of given domain
Zexecution_countr N)Z tracing_conf)r r5 r r r �executedR s zContinuousTracing.executedc C s� d}x�| � � D ]�}|jdkry| jj| �|j�d� W nt tk
rp } z | jj|dt |�id� W dd}~X Y q t
k
r� } z | jj|t |�|d�d� W dd}~X Y qX | � |j � qW | �
� S )z8
Start auto tasks with status 'running'
zStart auto task failedr3 )�urlr$ )r% N)r$ �continuous_task)�get_continuous_tasksr- r Z
start_autor; � original_urlr9 r& r' r( r
r< r5 Zdump_tracing_configuration)r Zerr_textr> r* r r r r X s
z ContinuousTracing.start_tracingc C sH t � }x<| �� D ]0}y|j|| �|j�< W q tk
r> Y qX qW |S )z[
Map e-mail addresses to urls in existing continuous tracing configuration
)�dictr? Zemailr; r@ r9 )r Z url_mail_mapr1 r r r �retrieve_mail_addrso s
z%ContinuousTracing.retrieve_mail_addrsc s& t td�dd�� � fdd�| �� D �S )z6
Select previous date completed tasks
)�tsr# c S s( yt | �t� kS tk
r" dS X dS )zO
Verify if task timestamp corresponds to previous date
FN)r r � TypeError)rC r r r �
check_date� s z:ContinuousTracing.filter_tasks_by_date.<locals>.check_datec s$ g | ] }|j d kr� |j�r|�qS )r, )r- Z starttime)r0 r1 )rE r r �
<listcomp>� s
z:ContinuousTracing.filter_tasks_by_date.<locals>.<listcomp>)�int�boolr+ )r r )rE r �filter_tasks_by_date{ s
z&ContinuousTracing.filter_tasks_by_datec C s� | j �d� tt�}| �� }| �� }xL|D ]D}|�|j�}|r*|| �t |j�d |j
i� | j �d|j |� q*W x4|� � D ](\}}| j
j|dt� t|�t|�d� q|W | 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 %sZreport)ZtemplateZdateZlinksZ
html_linkszE-mails sending finishedN)r& �infor r) rB rI �getr= �appendr Z shared_linkr/ �itemsr Z send_mailr r r )r Z
email_confZurl_mail_mappingZyesterday_tasksZtaskZmail�k�vr r r r! � s$
z ContinuousTracing.generate_mail)�__name__�
__module__� __qualname__�__doc__r r" r) r+ r � staticmethodr( r; r< r rA r rB rI r! �
__classcell__r r )r r r s r ) rS r8 Z collectionsr Ztypingr �commonr Zmailerr Z apiclientr Zconsole_utils.validationsr Zinternal.exceptionsr
r Zinternal.typesr Zinternal.utilsr
r r r r r Zmanagerr r r r r r �<module>
s