diff --git a/dongtai_engine/plugins/strategy_headers.py b/dongtai_engine/plugins/strategy_headers.py index 122169a8c..965842379 100644 --- a/dongtai_engine/plugins/strategy_headers.py +++ b/dongtai_engine/plugins/strategy_headers.py @@ -98,7 +98,8 @@ def check_response_header(method_pool): method_pool, position='HTTP Response Header') except Exception as e: - logger.error("check_response_header failed, reason: " + str(e)) + logger.error("check_response_header failed, reason: " + str(e), + exc_info=e) def save_vul(vul_type, method_pool, position=None, data=None): @@ -185,8 +186,8 @@ def save_vul(vul_type, method_pool, position=None, data=None): client_ip=method_pool.clent_ip, param_name=None, method_pool_id=method_pool.id, - project_version_id=vul.agent.project_version_id, - project_id=vul.agent.bind_project_id, + project_version_id=method_pool.agent.project_version_id, + project_id=method_pool.agent.bind_project_id, language=method_pool.agent.language, server_id=method_pool.agent.server_id, ) diff --git a/dongtai_protocol/report/handler/narmal_vul_handler.py b/dongtai_protocol/report/handler/narmal_vul_handler.py index d1b87754a..c4e65b29f 100644 --- a/dongtai_protocol/report/handler/narmal_vul_handler.py +++ b/dongtai_protocol/report/handler/narmal_vul_handler.py @@ -279,7 +279,7 @@ def save(self): strategy_id=strategy_id, uri=self.http_uri, http_method=self.http_method, - project_version_id=iast_vul.agent.project_version_id, + project_version_id=self.agent.project_version_id, pk__lt=iast_vul.id, ).delete() header_vul = None diff --git a/dongtai_web/projecttemplate/update_department_data.py b/dongtai_web/projecttemplate/update_department_data.py index 060df2e68..ef0ef45ed 100644 --- a/dongtai_web/projecttemplate/update_department_data.py +++ b/dongtai_web/projecttemplate/update_department_data.py @@ -10,5 +10,5 @@ def update_department_data(): else: department_dict[department.id] = f"{department_dict[department.parent_id]}-{department.id}" department.department_path = department_dict[department.id] - department.token = Token.generate_key() + department.token = Token().generate_key() department.save()