diff --git a/website/app/session_class.py b/website/app/session_class.py index ba988f5..2e32cab 100644 --- a/website/app/session_class.py +++ b/website/app/session_class.py @@ -3,7 +3,7 @@ import json from queue import Queue from threading import Thread from uuid import uuid4 -from .utils.utils import query_post_query, query_get_module, get_object +from .utils.utils import query_post_query, query_get_module, get_object, get_limit_queries from . import home_core as HomeModel import uuid from . import db @@ -179,7 +179,7 @@ class Session_class: histories = History.query.all() - while len(histories) > 200: + while len(histories) > get_limit_queries(): history = History.query.order_by(History.id).all() session = Session_db.query.filter_by(id=history[0].session_id) if not History_Tree.query.filter_by(session_uuid=session.uuid): diff --git a/website/app/utils/utils.py b/website/app/utils/utils.py index 12a649d..8476894 100644 --- a/website/app/utils/utils.py +++ b/website/app/utils/utils.py @@ -71,4 +71,7 @@ def gen_admin_password(): print(f"##########################\n## Admin password ##\n## {password} ##\n##########################") config["ADMIN"]["ADMIN_PASSWORD"] = password with open(CONF_PATH, "w") as conffile: - config.write(conffile) \ No newline at end of file + config.write(conffile) + +def get_limit_queries(): + return Config.QUERIES_LIMIT \ No newline at end of file diff --git a/website/conf/config.py b/website/conf/config.py index bb49a8a..3ab9666 100644 --- a/website/conf/config.py +++ b/website/conf/config.py @@ -4,6 +4,8 @@ class Config: FLASK_URL = '127.0.0.1' FLASK_PORT = 7008 MISP_MODULE = '127.0.0.1:6666' + + QUERIES_LIMIT = 200 class DevelopmentConfig(Config): DEBUG = True