config as a module
This commit is contained in:
@@ -10,43 +10,15 @@ import logging
|
||||
|
||||
from smtplib import SMTP_SSL as SMTP, SMTPException, SMTPAuthenticationError
|
||||
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
|
||||
from asebat.config import set_config as config
|
||||
|
||||
from pyftpdlib.handlers import FTPHandler
|
||||
from pyftpdlib.servers import FTPServer
|
||||
from pyftpdlib.authorizers import UnixAuthorizer
|
||||
from pyftpdlib.filesystems import UnixFilesystem
|
||||
|
||||
|
||||
class config():
|
||||
def __init__(self):
|
||||
c = ConfigParser()
|
||||
c.read([
|
||||
"/etc/ase/ftpcsvreceiver.ini",
|
||||
"./ftpcsvreceiver.ini"
|
||||
])
|
||||
|
||||
self.firstport = c.getint("ftpserver", "firstport")
|
||||
self.logfilename = c.get("ftpserver", "logfilename")
|
||||
self.proxyaddr = c.get("ftpserver", "proxyaddr")
|
||||
|
||||
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")
|
||||
|
||||
self.mqhost = c.get("mqserver", "hostname")
|
||||
self.mquser = c.get("mqserver", "user")
|
||||
self.mqpass = c.get("mqserver", "password")
|
||||
|
||||
self.csvfs = c.get("csvfs", "path")
|
||||
|
||||
def send_mail(sev, msg, cfg):
|
||||
msg = MIMEText(cfg.message + "\n" + msg)
|
||||
@@ -74,10 +46,10 @@ def write_queue(msg, cfg):
|
||||
':5672/%2F')
|
||||
connection = pika.BlockingConnection(parameters)
|
||||
channel = connection.channel()
|
||||
channel.queue_declare(queue='task_queue', durable=True)
|
||||
channel.queue_declare(queue=cfg.queue, durable=True)
|
||||
channel.basic_publish(
|
||||
exchange='',
|
||||
routing_key='task_queue',
|
||||
routing_key=cfg.queue,
|
||||
body=msg,
|
||||
properties=pika.BasicProperties(
|
||||
delivery_mode=2, # make message persistent
|
||||
@@ -182,7 +154,7 @@ class ASEHandler(FTPHandler):
|
||||
|
||||
|
||||
def main():
|
||||
cfg = config()
|
||||
cfg = config.config()
|
||||
try:
|
||||
authorizer = UnixAuthorizer(rejected_users=["root"],
|
||||
require_valid_shell=True)
|
||||
|
||||
Reference in New Issue
Block a user