mirror of https://github.com/MISP/misp-modules
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
from app import create_app, db
|
|
import argparse
|
|
from flask import render_template
|
|
import os
|
|
from app.utils.init_modules import create_modules_db
|
|
|
|
import signal
|
|
import sys
|
|
import subprocess
|
|
|
|
def signal_handler(sig, frame):
|
|
path = os.path.join(os.getcwd(), "launch.sh")
|
|
req = [path, "-ks"]
|
|
subprocess.call(req)
|
|
sys.exit(0)
|
|
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-i", "--init_db", help="Initialise the db if it not exist", action="store_true")
|
|
parser.add_argument("-r", "--recreate_db", help="Delete and initialise the db", action="store_true")
|
|
parser.add_argument("-d", "--delete_db", help="Delete the db", action="store_true")
|
|
parser.add_argument("-m", "--create_module", help="Create modules in db", action="store_true")
|
|
args = parser.parse_args()
|
|
|
|
os.environ.setdefault('FLASKENV', 'development')
|
|
|
|
app = create_app()
|
|
|
|
@app.errorhandler(404)
|
|
def error_page_not_found(e):
|
|
return render_template('404.html'), 404
|
|
|
|
|
|
if args.init_db:
|
|
with app.app_context():
|
|
db.create_all()
|
|
elif args.recreate_db:
|
|
with app.app_context():
|
|
db.drop_all()
|
|
db.create_all()
|
|
elif args.delete_db:
|
|
with app.app_context():
|
|
db.drop_all()
|
|
elif args.create_module:
|
|
with app.app_context():
|
|
create_modules_db()
|
|
else:
|
|
app.run(host=app.config.get("FLASK_URL"), port=app.config.get("FLASK_PORT"))
|