misp-modules/website/app/utils/init_modules.py

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.")