read ini e send mail
This commit is contained in:
@@ -4,8 +4,11 @@ import sys
|
||||
import os
|
||||
import re
|
||||
import logging
|
||||
import argparse
|
||||
|
||||
from smtplib import SMTP_SSL as SMTP # SMTP_SSL port 465
|
||||
from email.mime.text import MIMEText
|
||||
from configparser import ConfigParser
|
||||
|
||||
from asebat.timefmt import timestamp_fmt as ts
|
||||
from asebat.timefmt import date_refmt as df
|
||||
|
||||
@@ -14,6 +17,8 @@ from pyftpdlib.servers import FTPServer
|
||||
from pyftpdlib.authorizers import UnixAuthorizer
|
||||
from pyftpdlib.filesystems import UnixFilesystem
|
||||
|
||||
config = {}
|
||||
|
||||
|
||||
class ASEHandler(FTPHandler):
|
||||
def on_file_received(self, file):
|
||||
@@ -60,25 +65,49 @@ class ASEHandler(FTPHandler):
|
||||
logging.info("PID {:>5} >> {} moved into {}.".format(os.getpid(), filenameExt, newFilename))
|
||||
except OSError:
|
||||
logging.error("PID {:>5} >> Error to move {} into {}.".format(os.getpid(), filenameExt, newFilename))
|
||||
send_mail("Error", "OS error move " + filenameExt + " to " + newFilename)
|
||||
|
||||
def on_incomplete_file_received(self, file):
|
||||
# remove partially uploaded files
|
||||
os.remove(file)
|
||||
|
||||
|
||||
def read_config():
|
||||
c = ConfigParser()
|
||||
c.read(['/etc/ase/ftpcsvreceiver.ini',
|
||||
'/home/alex/devel/python/ftpcsvreceiver.ini'])
|
||||
|
||||
config["firstport"] = c.getint('ftpserver', 'firstport')
|
||||
config["logfilename"] = c.get('ftpserver', 'logfilename')
|
||||
config["proxyaddr"] = c.get('ftpserver', 'proxyaddr')
|
||||
config["smtphost"] = c.get('mailserver', 'hostname')
|
||||
config["smtpport"] = c.getint('mailserver', 'port')
|
||||
config["sender"] = c.get('mailserver', 'sender')
|
||||
config["password"] = c.get('mailserver', 'password')
|
||||
config["receivers"] = c.get('mailserver', 'receivers')
|
||||
config["message"] = c.get('mailserver', 'message')
|
||||
config["subject"] = c.get('mailserver', 'subject')
|
||||
config["debuglevel"] = c.getint('mailserver', 'debug')
|
||||
|
||||
def send_mail(sev, msg):
|
||||
msg = MIMEText(config["message"] + " " + msg)
|
||||
msg['Subject'] = config["subject"] + " " + sev
|
||||
msg['From'] = config["sender"]
|
||||
msg['To'] = config["receivers"]
|
||||
|
||||
conn = SMTP(config["smtphost"])
|
||||
conn.set_debuglevel(config["debuglevel"])
|
||||
conn.login(config["sender"], config["password"])
|
||||
try:
|
||||
conn.sendmail(config["sender"], config["receivers"], msg.as_string())
|
||||
except:
|
||||
logging.info("PID {:>5} >> Mail failed: {}.".format(os.getpid(), "CUSTOM_ERROR"))
|
||||
finally:
|
||||
conn.quit()
|
||||
|
||||
def main():
|
||||
|
||||
config =ConfigParser()
|
||||
config.read('/etc/ase/ftpcsv.receiver.ini')
|
||||
firstport = config.get('ftpserver', 'firstport')
|
||||
logfilename = config.get('ftpserver', 'logfilename')
|
||||
proxyaddr = config.get('ftpserver', 'proxyaddr')
|
||||
|
||||
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("-l", "--logfilename", help="log file", required=True)
|
||||
parser.add_argument("-p", "--port", help="passive port range start", type=int, required=True)
|
||||
parser.add_argument("-x", "--proxy", help="proxy address", required=True)
|
||||
args = parser.parse_args()
|
||||
read_config()
|
||||
|
||||
try:
|
||||
authorizer = UnixAuthorizer(
|
||||
@@ -86,18 +115,18 @@ def main():
|
||||
handler = ASEHandler
|
||||
handler.authorizer = authorizer
|
||||
handler.abstracted_fs = UnixFilesystem
|
||||
handler.masquerade_address = args.proxy
|
||||
_range = list(range(args.port, args.port + 20))
|
||||
handler.masquerade_address = config["proxyaddr"]
|
||||
_range = list(range(config["firstport"], config["firstport"] + 20))
|
||||
handler.passive_ports = _range
|
||||
|
||||
logging.basicConfig(format='%(asctime)s %(message)s',
|
||||
filename="/var/log/" + args.logfilename, level=logging.INFO)
|
||||
filename="/var/log/" + config["logfilename"], level=logging.INFO)
|
||||
|
||||
server = FTPServer(('0.0.0.0', 21), handler)
|
||||
|
||||
server.serve_forever()
|
||||
except KeyboardInterrupt:
|
||||
logging.info("PID {:>5} >> Error: {}.".format(os.getpid(), "Shutdown requested...exiting"))
|
||||
logging.info("PID {:>5} >> Info: {}.".format(os.getpid(), "Shutdown requested...exiting"))
|
||||
except Exception:
|
||||
print("{} - PID {:>5} >> Error: {}.".format(ts.timestamp("log"), os.getpid(), sys.exc_info()[1]))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user