mirror of
https://github.com/Pardus-LiderAhenk/ahenk
synced 2024-12-12 11:32:16 +03:00
72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>
|
||
|
|
|
||
|
|
import smtplib
|
||
|
|
import os
|
||
|
|
from email.mime.multipart import MIMEMultipart
|
||
|
|
from email.mime.text import MIMEText
|
||
|
|
from email.mime.base import MIMEBase
|
||
|
|
from email.utils import formatdate
|
||
|
|
from email import encoders
|
||
|
|
from base.scope import Scope
|
||
|
|
|
||
|
|
|
||
|
|
class Mail:
|
||
|
|
def __init__(self):
|
||
|
|
scope = Scope().get_instance()
|
||
|
|
self.logger = scope.get_logger()
|
||
|
|
self.configuration_manager = scope.get_configuration_manager()
|
||
|
|
try:
|
||
|
|
self.smtp_host = self.configuration_manager.get('MAIL', 'smtp_host')
|
||
|
|
self.smtp_port = int(self.configuration_manager.get('MAIL', 'smtp_port'))
|
||
|
|
self.from_username = self.configuration_manager.get('MAIL', 'from_username')
|
||
|
|
self.from_password = self.configuration_manager.get('MAIL', 'from_password')
|
||
|
|
self.to_address = self.configuration_manager.get('MAIL', 'to_address')
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
self.logger.error(
|
||
|
|
'A problem occurred while reading mail server parameters from conf file. Error Message: {0}'.format(
|
||
|
|
e))
|
||
|
|
self.server = None
|
||
|
|
self.logger.debug('Mail service initialized.')
|
||
|
|
|
||
|
|
def connect(self):
|
||
|
|
self.logger.debug('Connecting to SMTP server')
|
||
|
|
self.server = smtplib.SMTP(self.smtp_host, self.smtp_port)
|
||
|
|
self.logger.debug('Setting debug level')
|
||
|
|
self.server.set_debuglevel(2)
|
||
|
|
self.logger.debug('Using tls if server supports')
|
||
|
|
self.server.starttls()
|
||
|
|
self.logger.debug('Loging in to smtp server as {0}'.format(self.from_username))
|
||
|
|
self.server.login(self.from_username, self.from_password)
|
||
|
|
|
||
|
|
def disconnect(self):
|
||
|
|
self.logger.debug('Disconnecting from mail server')
|
||
|
|
self.server.quit()
|
||
|
|
self.logger.debug('Disconnected')
|
||
|
|
|
||
|
|
def send_mail(self, subject, message, files=None):
|
||
|
|
|
||
|
|
if files is None:
|
||
|
|
files = []
|
||
|
|
|
||
|
|
msg = MIMEMultipart()
|
||
|
|
msg['Date'] = formatdate(localtime=True)
|
||
|
|
msg['Subject'] = subject
|
||
|
|
|
||
|
|
msg.attach(MIMEText(message))
|
||
|
|
|
||
|
|
# TODO files attachment max size
|
||
|
|
if files is not None:
|
||
|
|
for f in files:
|
||
|
|
part = MIMEBase('application', "octet-stream")
|
||
|
|
part.set_payload(open(f, "rb").read())
|
||
|
|
encoders.encode_base64(part)
|
||
|
|
part.add_header('Content-Disposition', 'attachment; filename="{0}"'.format(os.path.basename(f)))
|
||
|
|
msg.attach(part)
|
||
|
|
|
||
|
|
self.logger.debug('Sending mail to {0} {1}'.format(self.to_address, ' about {0}'.format(subject)))
|
||
|
|
self.server.sendmail(self.from_username, self.to_address, msg.as_string())
|
||
|
|
self.logger.debug('Mail was sent.')
|