�
=�Dg� � � � d dl Zd dlZd dlZd dlZd dlZd dlmZm Z d dl
m Z d dl m
Z
d dlmZ dZ G d� de� � Z G d � d
� � ZdS ) � N)�base� multipart)�
SysLogHandler)�Optional)�errorzAleksandr Shyshatskyc � � e Zd ZdS )�MailSendFailedN)�__name__�
__module__� __qualname__� � �t/builddir/build/BUILD/imunify360-venv-2.4.0/opt/imunify360/venv/lib/python3.11/site-packages/clcommon/mail_helper.pyr r s � � � � � ��Dr r c � � e Zd ZdZd� Zd dej dee fd�Z d dee fd�Z
dede j fd �Z
d dee fd
�ZdS ) �
MailHelperz6Common class to send emails via local smtp or sendmailc �� � t j j s�t j � d� � r�t
dt j �� � }t j � |� � t j d�� � }|�
|� � |� t j � � nt j
� � t j � � | _ d S )Nz/dev/log)�address�facilityz3mail_helper: (%(name)s) [%(levelname)s] %(message)s)�fmt)�logging�root�handlers�os�path�existsr �
LOG_DAEMON�
addHandler� Formatter� setFormatter�setLevel�ERROR� basicConfig� getLogger�_log)�self�SYS_LOG�
formatter_syss r �__init__zMailHelper.__init__ s� � ��|�$� &��w�~�~�j�)�)�
&�'�
�]�E]�^�^�^��� �'�'��0�0�0� '� 1�6k� l� l� l�
��$�$�]�3�3�3�� � �� �/�/�/�/��#�%�%�%��%�'�'�� � � r N�msg�
encoding_namec �& � |r(|� � � � |� � }n|� � � }g d�}t j |t j t j t j d�� � 5 }|� |� � d }|j dk rt j |j ||� � � ddd� � dS # 1 swxY w Y dS # t t j f$ r"}t t |� � � � |�d}~ww xY w)z�
Alternative function that should be called
when we cannot send email via smtp
:type msg: email.mime.base.MIMEBase
:type encoding_name: encoding name to encode message text
)z/usr/sbin/sendmailz-tz-oiT)�stdin�stdout�stderr�textr N) � as_string�encode�
subprocess�Popen�PIPE� communicate�
returncode�CalledProcessError�OSErrorr �str)r% r) r* � s_message�cmd�procr- �exs r �_sendmail_consolez MailHelper._sendmail_console* sh � � 2� �
,� �M�M�O�O� 2� 2�=� A� A� � � �M�M�O�O� �5�5�5�C��!�� �o�!��!��� � � �
V� � �)�)�)�4�4�Q�7���?�a�'�'�$�7����f�U�U�U� (�
V�
V�
V�
V�
V�
V�
V�
V�
V�
V�
V�
V����
V�
V�
V�
V�
V�
V�� ��6�7� 2� 2� 2� ��R���)�)�r� 1����� 2���s= �A9C �;AC �> C � C�C �C�C �D�. D � Dc �x � t j d� � }|r(|� � � � |� � }n|� � � }|� |||� � |� � � dS # t j t f$ r"}t t |� � � � |�d}~ww xY w)z�
Send email using smtp protocol;
:type from_addr: str
:type to_addrs: List[str]
:type msg: email.mime.base.MIMEBase
:type encoding_name: encoding name to encode message text
� localhostN)
�smtplib�SMTPr0 r1 �sendmail�quit�
SMTPException� socketerrorr r9 )r% � from_addr�to_addrsr) r* �
smtp_instancer: r= s r �_sendmail_smtpzMailHelper._sendmail_smtpE s� � � 2� #�L�� 5� 5�M� �
,� �M�M�O�O� 2� 2�=� A� A� � � �M�M�O�O� � � "� "�9�h� � B� B� B� � � � � � � � ���%�{�3� 2� 2� 2� ��R���)�)�r� 1����� 2���s �A=B �B9� B4�4B9rG c �� � |d s_t j � |� d� � d �� � |d< | j � d|d � � dS dS )z{Add Message-ID to message if it is missing
:type from_addr: str
:type msg: multipart.MIMEMultipart
z
Message-ID�@���)�domainzAdded Message-ID: %sN)�email�utils�
make_msgid�splitr$ �debug)r% rG r) s r �_add_message_idzMailHelper._add_message_idX so � �
�<� � G� %� � 6� 6�i�o�o�c�>R�>R�SU�>V� 6� W� W�C� � � �I�O�O� 2�C� �4E� F� F� F� F� F� G� Gr c �� � | j � d|||� � | � ||� � | � ||||� � dS # t $ r�}| j � d� � | � |� � nQ# t $ rD | j � d� � | j � t |� � � � Y n
w xY wY d}~dS Y d}~dS d}~ww xY w)a0
Try to send message using local smtp;
If that can't be done - try to use /usr/sbin/sendmail;
:type from_addr: str
:type to_addrs: List[str]
:type msg: multipart.MIMEMultipart
:type encoding_name: Optional[str] encoding name to encode message text
z+Trying to send mail with params: %s, %s, %sz-Smtp send failed, trying to send via sendmailzcCannot send email neither via smtp nor using sendmail; Please, check your mail server configurationN) r$ rS rT rJ r r> � exceptionr r9 )r% rG rH r) r* �smtp_exceptions r rC zMailHelper.sendmaila s4 � �
� ���E�y�RZ�\_�`�`�`� � � �Y��,�,�,� 5� � � � �8�S�-� H� H� H� H� H�� � 5� 5� 5� �I�O�O� K� L� L� L�
5��&�&�s�+�+�+�+��!�
5�
5�
5�� �#�#� %S� T� T� T�� ����N� 3� 3�4�4�4�4�4�
5���� ,�+�+�+�+�+� 5�4�4�4�4�4����� 5���s< �A �
C,�C'�4B
� C'�
A C�C'�C�C'�'C,)N)r
r r �__doc__r( � mime_base�MIMEBaser r9 r> rJ r �
MIMEMultipartrT rC r
r r r r s� � � � � � �@�@�
(�
(�
(�2� 2�Y�%7� 2��QT�
� 2� 2� 2� 2�62� 2�h�s�m� 2� 2� 2� 2�&G�� G�9�3J� G� G� G� G�5� 5���
� 5� 5� 5� 5� 5� 5r r )� email.utilsrO r r rA r2 �
email.mimer rY r �logging.handlersr �typingr �_socketr rF �
__author__� Exceptionr r r
r r �<module>rc s� �� � � � � ���� � � � � ���� � � � � 3� 3� 3� 3� 3� 3� 3� 3� *� *� *� *� *� *� � � � � � � (� (� (� (� (� (�
#�
� � � � � �Y� � � �Z5� Z5� Z5� Z5� Z5� Z5� Z5� Z5� Z5� Z5r