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 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 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 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 __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()