def main(): parser = argparse.ArgumentParser() parser.add_argument("-c", "--clear_db", action="store_true", help="Clear database") parser.add_argument("-d", "--daemon", action="store_true", help="Run as daemon") parser.add_argument("-g", "--initialize_geolocation", action="store_true", help="(Re-)Initialize geolocation module") parser.add_argument( "-u", "--initialize_unsafe", action="store_true", help="Initialize a new set of unsafe URLs, domains and IPs") args = parser.parse_args() if args.clear_db: config = json.load(open("config.json")) db = Database(config["database"]["user"], config["database"]["password"], config["database"]["host"], config["database"]["port"], config["database"]["db"]) db.clear_db() if args.initialize_unsafe: initialize_unsafe_connections_list() if args.initialize_geolocation: GeoLocation.initialize() if args.daemon: Scheduler.run()
def main(): try: s = Scheduler() s.run() except: main()