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