File: //opt/alt/python37/lib/python3.7/site-packages/xray/continuous/__pycache__/mailer.cpython-37.pyc
B
�V�d � @ s� d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZ d dl m
Z
d dl m Z ddl
mZmZ ddlmZ G d d
� d
�ZdS ) � N)� ConfigParser� SectionProxy)� EmailMessage)� gethostname)�Optional� )�mail_template_location�mail_scripts_location)�XRayMailerErrorc @ s� e Zd ZdZdd� Zeed�dd��Zeed�dd��Z ed�d d
�Z
e e d�d d �Z e e d�d
d�Z
ed eed�dd��Zedd�dd�Zedd�dd�Zedd�dd�Zd eedd �d d �ZdS )!�Mailerz0
Class contains X-Ray e-mail send logic
c C s t �d�| _d | _d S )N�mailer)�logging� getLogger�logger�_sender)�self� r �G/opt/alt/python37/lib/python3.7/site-packages/xray/continuous/mailer.py�__init__ s zMailer.__init__)�returnc C s dS )z+
Local mail server address
)� localhostr )r r r r � mail_server s zMailer.mail_serverc C s | j dkr| �� | _ | j S )zG
Retrieve 'From' mail address if it is not already set
N)r �retrieve_mail_sender)r r r r �sender$ s
z
Mailer.senderc C s$ dt � � �}| �� }|dk r |S |S )zC
'From' address (control panel admin or dummy one)
zxray.continuous@N)r � admin_email)r Z
dummy_mailZ
admin_mailr r r r - s zMailer.retrieve_mail_senderc
C s� | � � }|dk r�t� d|� d�}y tj|gdddd�}|j�� S tjk
rv } z| j�d|t |�� W dd}~X Y n> t
t tj fk
r� } z| j�d|t |�� W dd}~X Y nX dS )z<
Try to retrieve control panel admin e-mail
N�/Z_emailT)�capture_output�text�checkz% script failed with: %sz Failed to run script %s with: %s)
�get_control_panelr �
subprocess�run�stdout�strip�CalledProcessErrorr �error�str�OSError�
ValueError�SubprocessError)r �panel�get_email_script�p�er r r r 5 s
zMailer.admin_emailc
C s� y t jddgdddd�j�� S t jtfk
rV } z| j�dt|�� W dd}~X Y n< t t
t j fk
r� } z| j�dt|�� W dd}~X Y nX dS )z(
Get control panel name
Zcldetectz--detect-cp-nameT)r r r z cldetect utility failed with %sNz&Failed to run cldetect utility with %s) r r! r" r# r$ �AttributeErrorr r% r& r' r( r) )r r- r r r r G s zMailer.get_control_panel�greeting)�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
r z.iniN)�
interpolation�datazFailed to find template z in )r �os�path�existsr �readr
)r0 �tmpl�configr r r �
read_templateW 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�smtplib�SMTPr � send_messageZ
SMTPExceptionr
r& �ConnectionErrorr' )r r: �server�resultr- r r r �
_smtp_sendd s
"zMailer._smtp_sendc
C sr | j �d� dddg}y tj||�� dddd� W n: ttjfk
rl } 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)�inputr r r z sendmail utility failed with N) r r; r r! � as_stringr' r$ r
r& )r r: �cmdr- r r r �
_console_sendr s
zMailer._console_send)�mailr 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)rB r
r r% r& rF �critical)r rG r- r r r �_send� s z Mailer._send)� recipient�templater c K sv | � |�}t� }|d |d< | j|d< ||d< |�|d | � |j|d | dd� | j�d|�� � | �|� d S ) N�subject�Subject�FromrH r �html)�subtypezGenerated mail --> %s) r9 r r � set_content�add_alternativer r; rD rJ )r rK rL �kwargsr2 �msgr r r � send_mail� s
zMailer.send_mail)r/ )r/ )�__name__�
__module__� __qualname__�__doc__r �property�tupler r& r r r r r � staticmethodr r9 r rB rF rJ rV r r r r r s r )r
r3 r< r � configparserr r �
email.messager �socketr �typingr Zinternal.constantsr r �internal.exceptionsr
r r r r r �<module> s