diff --git a/dongtai_common/models/user.py b/dongtai_common/models/user.py index 9119b113e..3c0876de2 100644 --- a/dongtai_common/models/user.py +++ b/dongtai_common/models/user.py @@ -70,7 +70,7 @@ def is_system_admin(self): return self.is_superuser == 1 def is_talent_admin(self): - return self.is_superuser == 2 or self.is_superuser == 1 or self.is_superuser == 6 + return self.is_superuser in {2, 1, 6} def get_talent(self): try: diff --git a/dongtai_engine/preheat.py b/dongtai_engine/preheat.py index 60d2a40ee..89a4b9160 100644 --- a/dongtai_engine/preheat.py +++ b/dongtai_engine/preheat.py @@ -56,7 +56,7 @@ def register(cls, function): if "return" in annotation_dict: del annotation_dict["return"] - if annotation_dict == {"user_id": int} or annotation_dict == {"user_id": "int"}: + if annotation_dict == {"user_id": int} or annotation_dict == {"user_id": "int"}: # noqa: PLR1714 pass else: logger.info(f"{function.__name__} annotations not fit in") diff --git a/dongtai_engine/signals/handlers/vul_handler.py b/dongtai_engine/signals/handlers/vul_handler.py index 260214480..4547bfa48 100644 --- a/dongtai_engine/signals/handlers/vul_handler.py +++ b/dongtai_engine/signals/handlers/vul_handler.py @@ -112,7 +112,7 @@ def parse_cookie(req_header: str, taint_value: str) -> str | None: # fixme 解析,然后匹配 _header_list = header.split(":") _header_name = _header_list[0] - if _header_name == "cookie" or _header_name == "Cookie": + if _header_name in ("cookie", "Cookie"): cookie_raw = ":".join(_header_list[1:]) break diff --git a/dongtai_engine/tasks.py b/dongtai_engine/tasks.py index d767bed29..a3aa79166 100644 --- a/dongtai_engine/tasks.py +++ b/dongtai_engine/tasks.py @@ -509,7 +509,7 @@ def vul_recheck(): for index in range(item_length): _header_list = header_raw[index].split(":") _header_name = _header_list[0] - if _header_name == "cookie" or _header_name == "Cookie": + if _header_name in ("cookie", "Cookie"): cookie_index = index cookie_raw = ":".join(_header_list[1:]) break diff --git a/dongtai_web/aggr_vul/aggr_vul_list.py b/dongtai_web/aggr_vul/aggr_vul_list.py index 8c732d140..db01ab300 100644 --- a/dongtai_web/aggr_vul/aggr_vul_list.py +++ b/dongtai_web/aggr_vul/aggr_vul_list.py @@ -402,7 +402,7 @@ def get_vul_list_from_elastic_search( field = "" opt = "" if isinstance(info, dict): - field = list(info.keys())[0] + field = next(iter(info.keys())) opt = "lt" if info[field]["order"] == "desc" else "gt" if isinstance(info, str): if info.startswith("-"): diff --git a/dongtai_web/views/scas.py b/dongtai_web/views/scas.py index c82ed17f5..345cb0f7e 100644 --- a/dongtai_web/views/scas.py +++ b/dongtai_web/views/scas.py @@ -368,7 +368,7 @@ def get_vul_list_from_elastic_searchv2( for info in order_list: field = "" if isinstance(info, dict): - field = list(info.keys())[0] + field = next(iter(info.keys())) if isinstance(info, str): field = info[1:] if info.startswith("-") else info if field == "package_name.keyword":