def __init__(self, config): self.config = config self.dbengine = get_engine(config) self.beanstalk_host = os.environ.get("BEANSTALK_HOST", config.defaults.get('beanstalk_host', 'localhost')) self.beanstalk_port = int(os.environ.get("BEANSTALK_PORT", config.defaults.get('beanstalk_port', 11300))) self.beanstalk = beanstalkc.Connection( host=self.beanstalk_host, port=self.beanstalk_port ) self.msg_num = 0 self.sum_time = time.time()
def __init__(self, config=None, **kwargs): self.config = config try: if 'TZ' not in os.environ: os.environ["TZ"] = config.defaults.tz time.tzset() except: pass self.db = scoped_session(sessionmaker(bind=get_engine(self.config), autocommit=False, autoflush=True)) settings = dict( cookie_secret=os.environ.get('cookie_secret', "12oETzKXQAGaYdkL5gEmGeJJFuYh7EQnp2XdTP1o/Vo="), login_url="/login", template_path=os.path.join(os.path.dirname(__file__), "views"), static_path=os.path.join(os.path.dirname(__file__), "static"), xsrf_cookies=True, db_engine = get_engine(self.config), api_secret=config.defaults.secret, debug=config.defaults.debug, db=self.db, xheaders=True, ) self.cache = CacheManager(**parse_cache_config_options({ 'cache.type': 'file', 'cache.data_dir': '/tmp/cache/data', 'cache.lock_dir': '/tmp/cache/lock' })) self.tp_lookup = TemplateLookup(directories=[settings['template_path']], default_filters=['decode.utf8'], input_encoding='utf-8', output_encoding='utf-8', encoding_errors='replace', module_directory="/tmp/toughlogger") self.init_permit() cyclone.web.Application.__init__(self, permit.all_handlers, **settings)
def run(): parser = argparse.ArgumentParser() parser.add_argument('-serv', '--serv', action='store_true', default=False, dest='serv', help='run logger') parser.add_argument('-syslogd', '--syslogd', action='store_true', default=False, dest='syslogd', help='run syslogd') parser.add_argument('-t', '--tcp', action='store_true', default=False, dest='tcp',help='run syslogd via tcp') parser.add_argument('-u', '--udp', action='store_true', default=False, dest='udp', help='run syslogd via udp') parser.add_argument('-ttask', '--table_task', action='store_true', default=False, dest='table_task',help='run table task') parser.add_argument('-wtask', '--write_task', action='store_true', default=False, dest='write_task',help='run write task') parser.add_argument('-initdb', '--initdb', action='store_true', default=False, dest='initdb', help='run initdb') parser.add_argument('-port', '--port', type=int, default=0, dest='port', help='admin port') parser.add_argument('-debug', '--debug', action='store_true', default=False, dest='debug', help='debug option') parser.add_argument('-x', '--xdebug', action='store_true', default=False, dest='xdebug', help='xdebug option') parser.add_argument('-c', '--conf', type=str, default="/etc/toughlogger.conf", dest='conf', help='config file') args = parser.parse_args(sys.argv[1:]) config = iconfig.Config(args.conf) if args.debug or args.xdebug: config.defaults.debug = True if args.port > 0: config.server.port = int(args.port) if args.serv: from toughlogger.console import logger_app logger_app.run(config) if args.syslogd: from toughlogger.syslog import udp_server, tcp_server if args.tcp: tcp_server.run(config) if args.udp: udp_server.run(config) if args.table_task: from toughlogger.syslog import table_task table_task.run(config) if args.write_task: from toughlogger.syslog import write_task write_task.run(config) if args.initdb: init_db.update(get_engine(config))
def __init__(self, config): self.dbengine = get_engine(config) self.sql_tpl = 'mysql' in self.dbengine.driver and mysql_create_sql_tpl or create_sql_tpl