Current File : //opt/alt/python37/lib/python3.7/site-packages/wmt/common/__pycache__/report.cpython-37.pyc
B
��wa� � @ s� d dl Z d dlmZmZ d dlmZ d dlmZ d dlm Z d dl
m Z m Z eG dd� d��Z
eG d d
� d
��ZeG d d � d ��Zd
d� Zdd� Zdd� ZdS )� N)� dataclass�asdict)�func)�literal_column)�cfg)� ScrapeResult�
session_scopec @ s> e Zd ZU eed< eed< eed< eed< eed< dd� ZdS ) �
SummaryReport� count_all�count_successful� count_failed� count_undone� average_timec G s | j | j| j| jt| jd �gS )Ni� )r
r r r
�intr )�self�args� r �B/opt/alt/python37/lib/python3.7/site-packages/wmt/common/report.py� to_template s
zSummaryReport.to_templateN)�__name__�
__module__� __qualname__r �__annotations__�floatr r r r r r s
r c @ s. e Zd ZU eed< eed< eed< dd� ZdS )� ErrorReport�code� count_errors�urlc C s0 t | j�}|dkr&d| j� d|� d�}|| jgS )N�htmlz <a href="z">z</a>)�
url_to_domainr r )r � alternativer r r r r # s
zErrorReport.to_templateN)r r r �strr r r r r r r r s
r c @ s& e Zd ZU eed< eed< dd� ZdS )�DurationReportr r c C s8 t | j�}|dkr&d| j� d|� d�}|t| jd �gS )Nr z <a href="z">z</a>i� )r r r r )r r r r r r r 0 s
zDurationReport.to_templateN)r r r r! r r r r r r r r" + s
r" c C s d}t �|d| �S )Nz http(s)?://� )�re�sub)r �patternr r r r 9 s r c C s� t | ��� }|�tjtjt�� �d���tj |ktj |ktjdktj
dk�� tjtj�� � }|�t�
|jj�t�|jj�|jj�� |jj��� }|�tjt�tj��d�t�� ��tj |ktj |ktjdk�� tj��td��� ��� }|�t��tj |ktj |ktj
dk��� }W d Q R X dd� |D �}dd� |D �}d d� |D �}d
d� |D �} td d � |D ��}
td
d � |D ��} dd� |D �} | �s�dnttdt| � t| � ��}
t|
| | |
| ||
d�}||| d�S )N� err_count�� Tr Fc S s& g | ] \}}}t �|�s|||f�qS r )r �is_domain_ignored)�.0r r �countr r r �
<listcomp>h s z#generate_report.<locals>.<listcomp>c S s& g | ] \}}}t �|�s|||f�qS r )r r) )r* r r+ r r r r r, k s c S s g | ]\}}}t |||d ��qS ))r r r )r )r* r r r r r r r, n s c S s* g | ]"\}}}t |tt|d ��d��qS )i� )r r )r" r �round)r* r r �_r r r r, q s c s s | ]\}}}|V qd S )Nr )r* r r. Z
success_countr r r � <genexpr>t s z"generate_report.<locals>.<genexpr>c s s | ]\}}}|V qd S )Nr )r* r. Z errors_countr r r r r/ u s c S s g | ] }|d �qS )� r )r* �itemr r r r, w s r i� )r
r r r
r )�summary_report� error_report�duration_report)r Zqueryr Z
response_codeZwebsiter r+ Zlabel�filterZ create_dateZ is_finishedZgroup_by�subqueryZ group_concat�c�sumr' �allZavgZresponse_time_msZorder_byr Zdescr r- �lenr )ZengineZ
start_dateZend_dateZsessionr6 Z error_statsZ
success_statsZcount_unsuccessfulr3 r4 Zsuccessful_requests_countZerror_requests_countZaveragesZ
average_countr2 r r r �generate_report> sV
& r; c C s0 t | d �dd� | d D �dd� | d D �d�S )Nr2 c S s g | ] }t |��qS r )r )r* r1 r r r r, � s z report_dict.<locals>.<listcomp>r3 c S s g | ] }t |��qS r )r )r* r1 r r r r, � s r4 )r2 r3 r4 )r )Zreportr r r � report_dict� s
r< )r$ Z dataclassesr r Z
sqlalchemyr Zsqlalchemy.sql.expressionr Z
wmt.commonr Zwmt.dbr r r r r" r r; r< r r r r �<module> s G
Mini Shell