def __init__(self): settings.config_init() # Find correct log level level = { 0: logging.DEBUG, 1: logging.INFO, 2: logging.WARNING, 3: logging.ERROR, 4: logging.CRITICAL }[settings.LOG_LEVEL] # Set up the global log log_format = '[%(asctime)s] %(message)s' log_datefmt = '%d.%m.%Y %I:%M:%S' if settings.DEBUG: logging.basicConfig(stream=sys.stdout, level=level, format=log_format, datefmt=log_datefmt) else: logging.basicConfig(filename=settings.LOG_FILE, filemode='ab', level=level, format=log_format, datefmt=log_datefmt) self.log = logging.getLogger(__name__) # Start up database & AMQP consumer db_init(settings.DATABASE_CONFIG) self.consumer = DownloadConsumer(settings.AMQP_URL) # All done self.log.info(u"Init OK & daemon running.")
def create_admin(_username, _password): settings.config_init() db_init(settings.DATABASE_CONFIG) pw_hash = pbkdf2_sha256.encrypt(_password) s = db_session() user = User(username=_username, nickname=_username, email="*****@*****.**", password=pw_hash, level=USERLEVELS['admin']) s.add(user) try: s.commit() print("User '{}' created.".format(_username)) except IntegrityError: print("User {} already exists.".format(_username)) finally: s.close()
finally: s.close() if 'create_admin' in sys.argv: username = raw_input("Username: "******"Event name: ") s = db_session() event = Event(name=name, visible=True) s.add(event) s.commit() s.close() exit(0) if 'list_events' in sys.argv: settings.config_init()
s.close() if 'create_admin' in sys.argv: username = raw_input("Username: "******"Event name: ") s = db_session() event = Event(name=name, visible=True) s.add(event) s.commit() s.close() exit(0) if 'list_events' in sys.argv: settings.config_init()
# parse command line arguments parser = argparse.ArgumentParser() parser.add_argument('-f', '--statuspath', help='path to dpkg status file', default='/var/lib/dpkg/status') parser.add_argument('-p', '--wsgiport', help='port of wsgi server', default='8080') parser.add_argument('-d', '--debug', action="count", help='turn on debugging output') args = parser.parse_args() # pull in arguments to vars force type wsgiport = int(args.wsgiport) status_file = str(args.statuspath) # initialize database conn = db.db_init(status_file) def REST_api(environ, start_response): # get arguments parameters = parse_qs(environ.get('QUERY_STRING', '')) # query functions wdict = funcs.query(conn, parameters) # format response from wdict status = wdict['status'] headers = wdict['headers'] status_string = wdict['status_string'] start_response(status, headers)