0x1998 - MANAGER
Düzenlenen Dosya: 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
geri dön