def init_model(engine=None):
if not engine:
config = get_config()
dburi = config.get('rodburi', None)
if not dburi:
dburi = config.get('rwdburi')
engine = _create_engine(dburi)
return sessionmaker(bind=engine, expire_on_commit=False)
def init_model(engine=None):
if not engine:
config = get_config()
dburi = config.get('rodburi', None)
if not dburi:
dburi = config.get('rwdburi')
engine = _create_engine(dburi)
return sessionmaker(bind=engine, expire_on_commit=False)
def __init__(self, socket_path=None):
self.config = common.get_config()
self.lockfile = None
self.requests = {}
self.socket_path = socket_path
if self.socket_path:
self.socket_path = os.path.abspath(self.socket_path)
self._server = None
def __init__(self, socket_path=None):
self.config = common.get_config()
self.lockfile = None
self.requests = {}
self.socket_path = socket_path
if self.socket_path:
self.socket_path = os.path.abspath(self.socket_path)
self._server = None
def run(self):
if not self.args:
self.log.error('no argument provided')
return 1
config = get_config()
for arg in self.args:
if arg not in config:
continue
print('%s: %s' % (arg, config[arg]))
def __init__(self, socket_path=None):
self.__requests = {}
self.__pending = Queue()
self.__buffer = None
self.__config = common.get_config()
self.__sender = None
self.__receiver = None
self.__socket = None
self.__socket_path = socket_path
self.__request_counter = 0
def __init__(self, socket_path=None):
self.__requests = {}
self.__pending = Queue()
self.__buffer = None
self.__config = common.get_config()
self.__sender = None
self.__receiver = None
self.__socket = None
self.__socket_path = socket_path
self.__request_counter = 0
def run(self):
if not self.args:
self.log.error('no argument provided')
return 1
config = get_config()
for arg in self.args:
if arg not in config:
continue
print ('%s: %s' % (arg, config[arg]))
def __init__(self):
self.config = get_config()
self.rpcserver = RPCServer()
self.core = Core(self)
self.rpcserver.add_receiver(JSONReceiver())
self.rpcserver.register_object(self.core)
self.rpcserver.register_object(Health())
self.rpcserver.register_object(Qpsmtpd())
self.running = False
def __init__(self):
self.config = get_config()
self.rpcserver = RPCServer()
self.core = Core(self)
self.db_session = init_model()
self.rpcserver.add_receiver(JSONReceiver())
self.rpcserver.register_object(self.core)
self.rpcserver.register_object(Health())
self.rpcserver.register_object(Qpsmtpd())
self.running = False
def _create_engine(dburi, debug=False, **engine_args):
config = get_config()
engine_args['echo'] = debug
if dburi.startswith('mysql'):
engine_args['max_overflow'] = config.get('max_overflow')
engine_args['pool_recycle'] = 1800
engine_args['pool_size'] = config.get('pool_size')
if not config.get('python_procs'):
procs.use_procedures('mysql')
if dburi.startswith('mysql+mysqlconnector'):
dburi = str(dburi)
else:
procs.use_procedures('py')
return create_engine(dburi, **engine_args)
def _create_engine(dburi, debug=False, **engine_args):
config = get_config()
engine_args['echo'] = debug
if dburi.startswith('mysql'):
engine_args['max_overflow'] = config.get('max_overflow')
engine_args['pool_recycle'] = 1800
engine_args['pool_size'] = config.get('pool_size')
if not config.get('python_procs'):
procs.use_procedures('mysql')
if dburi.startswith('mysql+mysqlconnector'):
dburi = str(dburi)
else:
procs.use_procedures('py')
return create_engine(dburi, **engine_args)
def send_notifications(self):
postmaster = 'postmaster@' + get_config('defaulthost')
user_msg = open(get_config_dir('user_quota.msg')).read()
# Build up the params for the messages
params = {
'usage': self.usage,
'date': formatdate(),
'to': self.user,
'mailbox': self.user
}
# Build the warning for the user
for key, value in params.iteritems():
user_msg = user_msg.replace(':' + key, value)
# Send the user warning
smtp = smtplib.SMTP('localhost')
smtp.sendmail(postmaster, self.user, user_msg)
# If we don't have any additional addresses then stop
if not self.addrs:
reactor.stop()
return
# Get the admin warning message
admin_msg = open(get_config_dir('admin_quota.msg')).read()
for addr in self.addrs:
params['to'] = addr
msg = admin_msg
# Build the admin message
for key, value in params.iteritems():
msg = msg.replace(':' + key, value)
try:
smtp.sendmail(postmaster, addr, msg)
except:
self.log.warning('Failed sending to %s', addr)
def send_notifications(self):
postmaster = 'postmaster@' + get_config('defaulthost')
user_msg = open(get_config_dir('user_quota.msg')).read()
# Build up the params for the messages
params = {
'usage': self.usage,
'date': formatdate(),
'to': self.user,
'mailbox': self.user
}
# Build the warning for the user
for key, value in params.iteritems():
user_msg = user_msg.replace(':' + key, value)
# Send the user warning
smtp = smtplib.SMTP('localhost')
smtp.sendmail(postmaster, self.user, user_msg)
# If we don't have any additional addresses then stop
if not self.addrs:
reactor.stop()
return
# Get the admin warning message
admin_msg = open(get_config_dir('admin_quota.msg')).read()
for addr in self.addrs:
params['to'] = addr
msg = admin_msg
# Build the admin message
for key, value in params.iteritems():
msg = msg.replace(':' + key, value)
try:
smtp.sendmail(postmaster, addr, msg)
except:
self.log.warning('Failed sending to %s', addr)
def rw_connect():
dburi = get_config().get('rwdburi')
init_rw_model(_create_engine(dburi))
def connect():
config = get_config()
dburi = config.get('rodburi', None)
if not dburi:
dburi = config.get('rwdburi')
init_model(_create_engine(dburi))
def __init__(self, socket_path=None, threaded=True):
self.config = common.get_config()
self.jobs = []
self.methods = {}
self.receivers = []
self.running = False
def __init__(self, socket_path=None, threaded=True):
self.config = common.get_config()
self.jobs = []
self.methods = {}
self.receivers = []
self.running = False
def __before__(self, method):
config = get_config()
self.username = config.get('test_username')
self.password = config.get('test_password')
def rw_connect():
dburi = get_config().get('rwdburi')
init_rw_model(_create_engine(dburi))
def connect():
config = get_config()
dburi = config.get('rodburi', None)
if not dburi:
dburi = config.get('rwdburi')
init_model(_create_engine(dburi))
def __before__(self, method):
config = get_config()
self.username = config.get('test_username')
self.password = config.get('test_password')
def __init__(self):
self.config = get_config()
def __init__(self):
self.config = get_config()
