72 lines
1.6 KiB
Plaintext
72 lines
1.6 KiB
Plaintext
|
#!/usr/bin/env python3
|
||
|
|
import logging
|
||
|
|
import yaml
|
||
|
|
import threading
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
import shelve
|
||
|
|
import getopt
|
||
|
|
|
||
|
|
import src.core as core
|
||
|
|
|
||
|
|
def start_new_thread(func, join=False, args=(), kwargs={}):
|
||
|
|
t = threading.Thread(target=func, args=args, kwargs=kwargs)
|
||
|
|
if not join:
|
||
|
|
t.daemon = True
|
||
|
|
t.start()
|
||
|
|
if join:
|
||
|
|
t.join()
|
||
|
|
|
||
|
|
def readopt(name):
|
||
|
|
global opts
|
||
|
|
for e in opts:
|
||
|
|
if e[0] == name:
|
||
|
|
return e[1]
|
||
|
|
return None
|
||
|
|
|
||
|
|
def usage():
|
||
|
|
print("Usage: %s [-q|-d] [-c config.yaml]" % sys.argv[0])
|
||
|
|
print("Options:")
|
||
|
|
print(" -h Display this text")
|
||
|
|
print(" -q Quiet, set log level to WARNING")
|
||
|
|
print(" -c Location of config file (default: ./config.yaml)")
|
||
|
|
|
||
|
|
def open_db(config):
|
||
|
|
return shelve.open(config["database"])
|
||
|
|
|
||
|
|
def main(configpath, loglevel=logging.INFO):
|
||
|
|
with open(configpath, "r") as f:
|
||
|
|
config = yaml.safe_load(f)
|
||
|
|
|
||
|
|
logging.basicConfig(format="%(levelname)-7s [%(asctime)s] %(message)s", datefmt="%Y-%m-%d %H:%M:%S", level=loglevel)
|
||
|
|
|
||
|
|
db = open_db(config)
|
||
|
|
|
||
|
|
core.init(config, db)
|
||
|
|
|
||
|
|
try:
|
||
|
|
start_new_thread(core.run, join=True)
|
||
|
|
except KeyboardInterrupt:
|
||
|
|
logging.info("Interrupted, exiting")
|
||
|
|
db.close()
|
||
|
|
os._exit(1)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
try:
|
||
|
|
opts, args = getopt.getopt(sys.argv[1:], "hqc:", ["help"])
|
||
|
|
except getopt.GetoptError as e:
|
||
|
|
print(str(e))
|
||
|
|
exit(1)
|
||
|
|
# Process command line args
|
||
|
|
if readopt("-h") or readopt("--help"):
|
||
|
|
usage()
|
||
|
|
exit(0)
|
||
|
|
loglevel = logging.INFO
|
||
|
|
if readopt("-q") is not None:
|
||
|
|
loglevel = logging.WARNING
|
||
|
|
configpath = "./config.yaml"
|
||
|
|
if readopt("-c") is not None:
|
||
|
|
configpath = readopt("-c")
|
||
|
|
# Run the actual program
|
||
|
|
main(configpath, loglevel)
|