from enum import Enum


class CeleryQueue(str, Enum):
    GEN_CHATAPI_CREATE = "gen_chatapi_create"
    GEN_PROVIDER_POLL = "gen_provider_poll"
    GEN_RESULT_DOWNLOAD = "gen_result_download"
    GEN_RECOVERY = "gen_recovery"
    GEN_PRIVATE_PORTRAIT = "gen_private_portrait"
    DEFAULT = "default"


class CeleryTaskName(str, Enum):
    CHATAPI_CREATE = "generation.chatapi_create_generation_task"
    POLL_GENERATION = "generation.poll_generation_task"
    DOWNLOAD_GENERATION_RESULT = "generation.download_generation_result_task"
    RECOVER_DOWNLOAD = "generation.recover_download_tasks_once"
    RECOVER_GENERATION = "generation.recover_generation_tasks_once"
    DISPATCH_DUE_POLL = "generation.dispatch_due_poll_tasks"
    STARTUP_RECOVERY = "recovery.startup_recovery_once"
    MODULE_ASYNC_RECOVERY = "module_async.recover_module_async_tasks_once"
    SHOT_SPLIT_RECOVERY = "shot_replicate.recover_split_tasks_once"
    PRIVATE_PORTRAIT_POLL_ASSET = "private_portrait.poll_asset_status"
    PRIVATE_PORTRAIT_SYNC_DUE_ASSETS = "private_portrait.sync_due_assets"
    PRIVATE_PORTRAIT_DELETE_ASSET = "private_portrait.delete_asset_remote"
    PRIVATE_PORTRAIT_DELETE_PROJECT = "private_portrait.delete_project_remote"
    PRIVATE_PORTRAIT_RECOVER_REMOTE_DELETES = "private_portrait.recover_remote_deletes"