ok con classe

FtpCsvReceiver.py

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