mirror of https://github.com/MISP/misp-modules
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
|
import os
|
||
|
|
import argparse
|
||
|
|
import subprocess
|
||
|
|
import time
|
||
|
|
from app_creation import main
|
||
|
|
|
||
|
|
import signal
|
||
|
|
import sys
|
||
|
|
def signal_handler(sig, frame):
|
||
|
|
kill_script()
|
||
|
|
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", "--reload_db", help="Delete and initialise the db", action="store_true")
|
||
|
|
parser.add_argument("-l", "--launch", help="Launch the app", action="store_true")
|
||
|
|
parser.add_argument("-ks", "--killscript", help="Kill screen running background", action="store_true")
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
def kill_script():
|
||
|
|
r = ["screen", "-ls", "|", "egrep", "[0-9]+.misp_mod", "|", "cut", "-d.", "-f1"]
|
||
|
|
process = subprocess.Popen(r, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||
|
|
out, err = process.communicate()
|
||
|
|
if out:
|
||
|
|
subprocess.call(["screen", "-X", "-S", "misp_mod", "quit"])
|
||
|
|
|
||
|
|
if args.init_db:
|
||
|
|
main(init_db=True)
|
||
|
|
elif args.reload_db:
|
||
|
|
main(recreate_db=True)
|
||
|
|
elif args.launch:
|
||
|
|
os.environ.setdefault('FLASKENV', 'development')
|
||
|
|
kill_script()
|
||
|
|
subprocess.call(["screen", "-dmS", "misp_mod"])
|
||
|
|
r = ["screen", "-S", "misp_mod", "-X", "screen", "-t", "misp_modules_server", "bash", "-c", "../env/bin/misp-modules", "-l", "127.0.0.1;", "read x"]
|
||
|
|
subprocess.call(r)
|
||
|
|
time.sleep(2)
|
||
|
|
main(create_module=True)
|
||
|
|
main()
|
||
|
|
elif args.killscript:
|
||
|
|
kill_script()
|