Current File : //opt/alt/python37/lib/python3.7/site-packages/ssa/internal/__pycache__/mailer.cpython-37.pyc
B
r�b � @ s� d Z ddlZddlZddlZddlZddlmZmZ ddlm Z ddl
m Z ddl m
Z
ddlmZmZ dd lmZ dd
lmZ dd lmZmZ eed �d
d�ZG dd� d�ZdS )z8
This module contains SSA Mailer class and helper tools
� N)� ConfigParser� SectionProxy)� EmailMessage)� gethostname)�is_cl_solo_edition)� Environment�FileSystemLoader� )�mail_template_location)�SSAMailerError)�
duration_cast� format_date)� report_data�returnc C sf t tt�ddd�}t|jd< |�d�}|j| d tdd�d�}|�d�}|jt| d �|d
�}||fS ) zJ
Render an HTML table for e-mail report and a full e-mail message
T)�loaderZ trim_blocksZ
lstrip_blocksr zreport_table.html�domains)�skip_jwt_check)r Zis_soloz mail.html�date)r � report_table) r r r
r �filtersZ get_templateZrenderr r
)r �envZtable_templater Z
mail_template�mail� r �D/opt/alt/python37/lib/python3.7/site-packages/ssa/internal/mailer.py�render_report_table s
r c @ s� e Zd ZdZdd� Zeed�dd��Zeed�dd��Z e
ee d �d
d ��Z e
d d
�dd�Ze
d d
�dd�Ze
d d�dd�Zdeed d�dd�Zeee
d�dd�Zd S ) �Mailerz.
Class contains SSA e-mail send logic
c C s t �d�| _d | _d S )NZmailer)�logging� getLogger�loggerZ_sender)�selfr r r �__init__6 s zMailer.__init__)r c C s dS )z+
Local mail server address
)� localhostr )r r r r � mail_server: s zMailer.mail_serverc C s dt � � �S )z%
'From' mail address
zslow-site-analyzer@)r )r r r r �senderA s z
Mailer.sender)�namer c C sP t � d| � d�}tj�|�r8tdd�}|�|� |d S td| � dt � ���dS )zF
Get preformatted data for e-mail by name of template
�/z.iniN)�
interpolation�datazFailed to find template z in )r
�os�path�existsr �readr )r$ Ztmpl�configr r r �
read_templateH s
zMailer.read_templateN)�messager c
C s� | j �d� y4tj| j� � }|�|�}| j �d|� W dQ R X W nl tjk
rv } ztdt|�� ���W dd}~X Y n8 t t
fk
r� } ztdt|�� ���W dd}~X Y nX dS )z=
Send preformatted e-mail via localhost SMTP
zTry to send via smtpzSend result: %sNzsmtp mailing failed: zsmtp connection failed: ) r �info�smtplibZSMTPr" Z send_messageZ
SMTPExceptionr �str�ConnectionError�OSError)r r. �server�result�er r r �
_smtp_sendU s
"zMailer._smtp_sendc
C st | j �d� dddg}y tj||�� dddd� W n< tttjfk
rn } ztdt |�� ���W dd}~X Y nX dS ) z?
Send preformatted e-mail via sendmail utility
z Try to send via sendmail utilityz/usr/sbin/sendmailz-tz-oiT)�input�capture_output�text�checkz sendmail utility failed with N)
r r/ �
subprocess�run� as_stringr3 �
ValueError�CalledProcessErrorr r1 )r r. �cmdr6 r r r �
_console_sendc s
zMailer._console_send)r r c
C s� y| � |� W n� tk
r� } zh| j�t|�� y| �|� W nD tk
r� } z&| j�t|�� | j�d|d � W dd}~X Y nX W dd}~X Y nX dS )zm
Try to send mail via localhost smtp server,
if fails -- try to use sendmail utility
z3Both smtp and sendmail failed to send message to %s�ToN)r7 r r �errorr1 rB �critical)r r r6 r r r �_sendq s z Mailer._send�
ssa_report)� recipient�templater c K s | � | j||f|�� dS )z{
Create a message from given template, including additional kwargs,
and send it to given recipient
N)rF �_message)r rH rI �kwargsr r r �send_simple_mail� s zMailer.send_simple_mailc K sp | � |�}t� }|d | |d< | j|d< ||d< |�|d | � |j|d | dd� | j�d|�� � |S ) z
Form a message
�subjectZSubjectZFromrC r: �html)�subtypezGenerated mail --> %s)r- r r# � set_content�add_alternativer r/ r> )r rH rI rK r' �msgr r r rJ � s
zMailer._message)rG )�__name__�
__module__� __qualname__�__doc__r �property�tupler" r1 r# � staticmethodr r- r r7 rB rF rL rJ r r r r r 1 s r ) rV r r( r0 r< � configparserr r �
email.messager �socketr �clcommon.lib.cleditionr Zjinja2r r � constantsr
�
exceptionsr �utilsr r
�dictrX r r r r r r �<module>
s
Mini Shell