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: