From d41d4deaac01cc9cad94bd03dec1c36e93d000cf Mon Sep 17 00:00:00 2001 From: Asuka Minato Date: Mon, 1 Sep 2025 16:40:26 +0900 Subject: [PATCH] example enum to StrEnum (#24877) Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> --- api/controllers/service_api/wraps.py | 10 +++++----- api/core/app/apps/base_app_queue_manager.py | 8 ++++---- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/api/controllers/service_api/wraps.py b/api/controllers/service_api/wraps.py index d2473c15af..cc4b5f65bd 100644 --- a/api/controllers/service_api/wraps.py +++ b/api/controllers/service_api/wraps.py @@ -1,7 +1,7 @@ import time from collections.abc import Callable from datetime import timedelta -from enum import Enum +from enum import StrEnum, auto from functools import wraps from typing import Optional @@ -23,14 +23,14 @@ from models.model import ApiToken, App, EndUser from services.feature_service import FeatureService -class WhereisUserArg(Enum): +class WhereisUserArg(StrEnum): """ Enum for whereis_user_arg. """ - QUERY = "query" - JSON = "json" - FORM = "form" + QUERY = auto() + JSON = auto() + FORM = auto() class FetchUserArg(BaseModel): diff --git a/api/core/app/apps/base_app_queue_manager.py b/api/core/app/apps/base_app_queue_manager.py index 9da0bae56a..07b557d3d5 100644 --- a/api/core/app/apps/base_app_queue_manager.py +++ b/api/core/app/apps/base_app_queue_manager.py @@ -1,7 +1,7 @@ import queue import time from abc import abstractmethod -from enum import Enum +from enum import IntEnum, auto from typing import Any, Optional from sqlalchemy.orm import DeclarativeMeta @@ -19,9 +19,9 @@ from core.app.entities.queue_entities import ( from extensions.ext_redis import redis_client -class PublishFrom(Enum): - APPLICATION_MANAGER = 1 - TASK_PIPELINE = 2 +class PublishFrom(IntEnum): + APPLICATION_MANAGER = auto() + TASK_PIPELINE = auto() class AppQueueManager: