constants.py

import os
from enum import IntEnum
from pathlib import Path


def get_base(dir_name: str) -> Path:
    path = Path(base) if (base := os.getenv(f"{dir_name.upper()}_PATH")) else Path(__file__).parent / dir_name
    path.mkdir(parents=True, exist_ok=True)
    return path


DEFAULT_CONFIG_PATH = get_base("config") / "config.json"

DEFAULT_DATABASE_PATH = get_base("data") / "data.db"

DEFAULT_THUMB_PATH = get_base("thumb")

FFMPEG_COMMAND = "ffmpeg"

MIGRATE_COMMAND = "aerich"


class MediaType(IntEnum):
    VIDEO = 2
    AUDIO = 12
    VIDEO_COLLECTION = 21


class MediaStatus(IntEnum):
    NORMAL = 1  # 正常稿件
    INVISIBLE = 2  # 不可见稿件
    DELETED = 3  # 已失效视频

    @property
    def text(self) -> str:
        return {MediaStatus.NORMAL: "normal", MediaStatus.INVISIBLE: "invisible", MediaStatus.DELETED: "deleted"}[self]


class NfoMode(IntEnum):
    MOVIE = 1
    TVSHOW = 2
    EPISODE = 3
    UPPER = 4


TORTOISE_ORM = {
    "connections": {"default": f"sqlite://{DEFAULT_DATABASE_PATH}"},
    "apps": {"models": {"models": ["models", "aerich.models"], "default_connection": "default"}},
    "use_tz": True,
}