feat: add hook strategy type in createion.

HXSecurity · Aug 3, 2023 · 5cf91a1 · 5cf91a1
1 parent 90c91b8
commit 5cf91a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dongtai_web/views/engine_hook_rule_add.py b/dongtai_web/views/engine_hook_rule_add.py
@@ -218,7 +218,7 @@ def post(self, request):
             return R.failure(data=e.detail, msg=_("Incomplete parameter, please check again"))
         # for compatibility only
         # the "type" not in ser.validated_data should be remove.
-        if ser.validated_data["type"] == 4 or (
+        if ("type" in ser.validated_data and ser.validated_data["type"] == 4) or (
             "type" not in ser.validated_data and ser.validated_data["rule_target"] == ""
         ):
             hook_type = IastStrategyModel.objects.filter(