mirror of https://github.com/MISP/misp-modules
45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
import json
|
|
import os
|
|
from ..db_class.db import db
|
|
from ..db_class.db import Module, Config, Module_Config
|
|
from .utils import query_get_module
|
|
|
|
|
|
def create_modules_db():
|
|
modules = query_get_module()
|
|
if not "message" in modules:
|
|
for module in modules:
|
|
m = Module.query.filter_by(name=module["name"]).first()
|
|
input_attr = ""
|
|
if "input" in module["mispattributes"]:
|
|
input_attr = json.dumps(module["mispattributes"]["input"])
|
|
if not m:
|
|
m = Module(
|
|
name=module["name"],
|
|
description=module["meta"]["description"],
|
|
is_active=True,
|
|
request_on_query=False,
|
|
input_attr=input_attr
|
|
)
|
|
db.session.add(m)
|
|
db.session.commit()
|
|
|
|
|
|
if "config" in module["meta"]:
|
|
for conf in module["meta"]["config"]:
|
|
c = Config.query.filter_by(name=conf).first()
|
|
if not c:
|
|
c = Config(
|
|
name = conf
|
|
)
|
|
db.session.add(c)
|
|
db.session.commit()
|
|
|
|
mc = Module_Config(
|
|
module_id=m.id,
|
|
config_id=c.id
|
|
)
|
|
db.session.add(mc)
|
|
db.session.commit()
|
|
else:
|
|
print("[-] Error in misp-modules. It might not running.") |