alex/python
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ok con classe

Browse Source
This commit is contained in:
Alessandro Battilani
2020-04-26 19:08:50 +02:00
parent d7b0f79597
commit d5f5f1a5da
1 changed files with 43 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
FtpCsvReceiver.py Normal file → Executable file
View File

@@ -20,45 +20,44 @@ from pyftpdlib.servers import FTPServer
from pyftpdlib.authorizers import UnixAuthorizer from pyftpdlib.authorizers import UnixAuthorizer
from pyftpdlib.filesystems import UnixFilesystem from pyftpdlib.filesystems import UnixFilesystem
config = {}
class config():
def __init__(self):
c = ConfigParser()
c.read([
"/etc/ase/ftpcsvreceiver.ini",
"./ftpcsvreceiver.ini"
])
def read_config(): self.firstport = c.getint("ftpserver", "firstport")
c = ConfigParser() self.logfilename = c.get("ftpserver", "logfilename")
c.read([ self.proxyaddr = c.get("ftpserver", "proxyaddr")
"/etc/ase/ftpcsvreceiver.ini",
"./ftpcsvreceiver.ini"
])
config["firstport"] = c.getint("ftpserver", "firstport") self.smtphost = c.get("mailserver", "hostname")
config["logfilename"] = c.get("ftpserver", "logfilename") self.smtpport = c.getint("mailserver", "port")
config["proxyaddr"] = c.get("ftpserver", "proxyaddr") self.sender = c.get("mailserver", "sender")
self.password = c.get("mailserver", "password")
self.receivers = c.get("mailserver", "receivers")
self.message = c.get("mailserver", "message")
self.subject = c.get("mailserver", "subject")
self.debuglevel = c.getint("mailserver", "debug")
config["smtphost"] = c.get("mailserver", "hostname") self.mqhost = c.get("mqserver", "hostname")
config["smtpport"] = c.getint("mailserver", "port") self.mquser = c.get("mqserver", "user")
config["sender"] = c.get("mailserver", "sender") self.mqpass = c.get("mqserver", "password")
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")
config["mqhost"] = c.get("mqserver", "hostname") self.csvfs = c.get("csvfs", "path")
config["mquser"] = c.get("mqserver", "user")
config["mqpass"] = c.get("mqserver", "password")
config["csvfs"] = c.get("csvfs", "path")
def send_mail(sev, msg): def send_mail(sev, msg, cfg):
msg = MIMEText(config["message"] + "\n" + msg) msg = MIMEText(cfg.message + "\n" + msg)
msg["Subject"] = config["subject"] + " " + sev msg["Subject"] = cfg.subject + " " + sev
msg["From"] = config["sender"] msg["From"] = cfg.sender
msg["To"] = config["receivers"] msg["To"] = cfg.receivers
conn = SMTP(host=config["smtphost"], port=config["smtpport"], local_hostname=None, timeout=5, source_address=None ) conn = SMTP(host=cfg.smtphost, port=cfg.smtpport, local_hostname=None, timeout=5, source_address=None )
conn.set_debuglevel(config["debuglevel"]) conn.set_debuglevel(cfg.debuglevel)
try: try:
conn.login(config["sender"], config["password"]) conn.login(cfg.sender, cfg.password)
conn.sendmail(config["sender"], config["receivers"], msg.as_string()) conn.sendmail(cfg.sender, cfg.receivers, msg.as_string())
except SMTPAuthenticationError: except SMTPAuthenticationError:
logging.error("PID {:>5} >> Mail failed: {}.".format( logging.error("PID {:>5} >> Mail failed: {}.".format(
os.getpid(), "SMTP authentication error")) os.getpid(), "SMTP authentication error"))
@@ -69,9 +68,9 @@ def send_mail(sev, msg):
conn.quit() conn.quit()
def write_queue(msg): def write_queue(msg, cfg):
parameters = pika.URLParameters('amqp://' + config["mquser"] + ':' + parameters = pika.URLParameters('amqp://' + cfg.mquser + ':' +
config["mqpass"] + '@' + config["mqhost"] + cfg.mqpass + '@' + cfg.mqhost +
':5672/%2F') ':5672/%2F')
connection = pika.BlockingConnection(parameters) connection = pika.BlockingConnection(parameters)
channel = connection.channel() channel = connection.channel()
@@ -90,6 +89,7 @@ def write_queue(msg):
class ASEHandler(FTPHandler): class ASEHandler(FTPHandler):
def on_file_received(self, file): def on_file_received(self, file):
cfg = self.cfg
path, filenameExt = os.path.split(file) path, filenameExt = os.path.split(file)
filename, fileExtension = os.path.splitext(filenameExt) filename, fileExtension = os.path.splitext(filenameExt)
@@ -141,7 +141,7 @@ class ASEHandler(FTPHandler):
df.dateFmt(fileDate), df.dateFmt(fileDate),
fileTime, fileTime,
)) ))
newPath = config["csvfs"] + self.username + "/received/" + unitName.upper( newPath = cfg.csvfs + self.username + "/received/" + unitName.upper(
) + "/" ) + "/"
newFilename = (newPath + filename + "_" + newFilename = (newPath + filename + "_" +
str(ts.timestamp("tms") + fileExtension)) str(ts.timestamp("tms") + fileExtension))
@@ -160,7 +160,7 @@ class ASEHandler(FTPHandler):
logging.error("PID {:>5} >> Error to move {} into {}.".format( logging.error("PID {:>5} >> Error to move {} into {}.".format(
os.getpid(), filenameExt, newFilename)) os.getpid(), filenameExt, newFilename))
# send_mail("Error", # send_mail("Error",
# "OS error move " + filenameExt + " to " + newFilename) # "OS error move " + filenameExt + " to " + newFilename, cfg)
mq_message = "{};{};{};{};{};{};{}".format( mq_message = "{};{};{};{};{};{};{}".format(
@@ -172,7 +172,7 @@ class ASEHandler(FTPHandler):
fileTime, fileTime,
newFilename, newFilename,
) )
write_queue(mq_message) write_queue(mq_message, cfg)
logging.info("PID {:>5} >> queue message: {}.".format( logging.info("PID {:>5} >> queue message: {}.".format(
os.getpid(), mq_message)) os.getpid(), mq_message))
@@ -182,20 +182,21 @@ class ASEHandler(FTPHandler):
def main(): def main():
read_config() cfg = config()
try: try:
authorizer = UnixAuthorizer(rejected_users=["root"], authorizer = UnixAuthorizer(rejected_users=["root"],
require_valid_shell=True) require_valid_shell=True)
handler = ASEHandler handler = ASEHandler
handler.cfg = cfg
handler.authorizer = authorizer handler.authorizer = authorizer
handler.abstracted_fs = UnixFilesystem handler.abstracted_fs = UnixFilesystem
handler.masquerade_address = config["proxyaddr"] handler.masquerade_address = cfg.proxyaddr
_range = list(range(config["firstport"], config["firstport"] + 20)) _range = list(range(cfg.firstport, cfg.firstport + 20))
handler.passive_ports = _range handler.passive_ports = _range
logging.basicConfig( logging.basicConfig(
format="%(asctime)s %(message)s", format="%(asctime)s %(message)s",
filename="/var/log/" + config["logfilename"], filename="/var/log/" + cfg.logfilename,
level=logging.INFO, level=logging.INFO,
) )