mirror of
https://github.com/RhenCloud/Cloud-Index.git
synced 2025-12-06 15:26:10 +08:00
105 lines
4.3 KiB
Python
105 lines
4.3 KiB
Python
"""
|
|
配置管理模块
|
|
集中管理所有环境变量和应用配置
|
|
"""
|
|
|
|
import os
|
|
from typing import Optional
|
|
|
|
import dotenv
|
|
|
|
# 加载环境变量
|
|
dotenv.load_dotenv()
|
|
|
|
|
|
class Config:
|
|
"""应用配置类"""
|
|
|
|
# 存储配置
|
|
STORAGE_TYPE: str = os.getenv("STORAGE_TYPE", "").lower()
|
|
|
|
# R2 配置
|
|
R2_ACCOUNT_ID: Optional[str] = os.getenv("R2_ACCOUNT_ID")
|
|
R2_ACCESS_KEY_ID: Optional[str] = os.getenv("R2_ACCESS_KEY_ID")
|
|
R2_SECRET_ACCESS_KEY: Optional[str] = os.getenv("R2_SECRET_ACCESS_KEY")
|
|
R2_BUCKET_NAME: Optional[str] = os.getenv("R2_BUCKET_NAME")
|
|
R2_PUBLIC_DOMAIN: Optional[str] = os.getenv("R2_PUBLIC_DOMAIN")
|
|
|
|
# GitHub 配置
|
|
GITHUB_TOKEN: Optional[str] = os.getenv("GITHUB_TOKEN")
|
|
GITHUB_REPO: Optional[str] = os.getenv("GITHUB_REPO") # 格式: owner/repo
|
|
GITHUB_BRANCH: str = os.getenv("GITHUB_BRANCH", "main")
|
|
|
|
# OneDrive 配置
|
|
ONEDRIVE_REFRESH_TOKEN: Optional[str] = os.getenv("ONEDRIVE_REFRESH_TOKEN")
|
|
ONEDRIVE_CLIENT_ID: Optional[str] = os.getenv("ONEDRIVE_CLIENT_ID")
|
|
ONEDRIVE_CLIENT_SECRET: Optional[str] = os.getenv("ONEDRIVE_CLIENT_SECRET")
|
|
ONEDRIVE_FOLDER_ID: Optional[str] = os.getenv("ONEDRIVE_FOLDER_ID") # 可选，默认使用 /me/drive/root
|
|
ONEDRIVE_REDIRECT_URI: Optional[str] = os.getenv("ONEDRIVE_REDIRECT_URI") # 可选，刷新令牌时某些应用需要
|
|
|
|
# 应用配置
|
|
HOST: str = os.getenv("HOST", "0.0.0.0")
|
|
PORT: int = int(os.getenv("PORT", "5000"))
|
|
DEBUG: bool = os.getenv("DEBUG", "false").lower() == "true"
|
|
|
|
# 缩略图配置
|
|
THUMB_TTL_SECONDS: int = int(os.getenv("THUMB_TTL_SECONDS", "3600"))
|
|
THUMB_SIZE: tuple[int, int] = (300, 300) # 缩略图尺寸
|
|
|
|
# URL过期时间配置
|
|
PRESIGNED_URL_EXPIRES: int = int(os.getenv("PRESIGNED_URL_EXPIRES", "3600"))
|
|
|
|
@classmethod
|
|
def validate(cls) -> None:
|
|
"""验证必需的配置项是否已设置"""
|
|
if not cls.STORAGE_TYPE:
|
|
raise ValueError("STORAGE_TYPE environment variable is not set. Supported types: r2, github, onedrive")
|
|
|
|
if cls.STORAGE_TYPE == "r2":
|
|
required = ["R2_ACCOUNT_ID", "R2_ACCESS_KEY_ID", "R2_SECRET_ACCESS_KEY", "R2_BUCKET_NAME"]
|
|
missing = [key for key in required if not getattr(cls, key)]
|
|
if missing:
|
|
raise ValueError(f"Missing required R2 configuration: {', '.join(missing)}")
|
|
|
|
elif cls.STORAGE_TYPE == "github":
|
|
required = ["GITHUB_TOKEN", "GITHUB_REPO"]
|
|
missing = [key for key in required if not getattr(cls, key)]
|
|
if missing:
|
|
raise ValueError(f"Missing required GitHub configuration: {', '.join(missing)}")
|
|
|
|
elif cls.STORAGE_TYPE == "onedrive":
|
|
required = ["ONEDRIVE_REFRESH_TOKEN", "ONEDRIVE_CLIENT_ID", "ONEDRIVE_CLIENT_SECRET"]
|
|
missing = [key for key in required if not getattr(cls, key)]
|
|
if missing:
|
|
raise ValueError(f"Missing required OneDrive configuration: {', '.join(missing)}")
|
|
|
|
elif cls.STORAGE_TYPE not in ["r2", "github", "onedrive"]:
|
|
raise ValueError(f"Unsupported storage type: {cls.STORAGE_TYPE}. Supported types: r2, github, onedrive")
|
|
|
|
@classmethod
|
|
def get_storage_config(cls) -> dict:
|
|
"""获取当前存储类型的配置字典"""
|
|
if cls.STORAGE_TYPE == "r2":
|
|
return {
|
|
"account_id": cls.R2_ACCOUNT_ID,
|
|
"access_key_id": cls.R2_ACCESS_KEY_ID,
|
|
"secret_access_key": cls.R2_SECRET_ACCESS_KEY,
|
|
"bucket_name": cls.R2_BUCKET_NAME,
|
|
"public_domain": cls.R2_PUBLIC_DOMAIN,
|
|
}
|
|
elif cls.STORAGE_TYPE == "github":
|
|
return {
|
|
"token": cls.GITHUB_TOKEN,
|
|
"repo": cls.GITHUB_REPO,
|
|
"branch": cls.GITHUB_BRANCH,
|
|
}
|
|
elif cls.STORAGE_TYPE == "onedrive":
|
|
return {
|
|
"client_id": cls.ONEDRIVE_CLIENT_ID,
|
|
"client_secret": cls.ONEDRIVE_CLIENT_SECRET,
|
|
"refresh_token": cls.ONEDRIVE_REFRESH_TOKEN,
|
|
"folder_id": cls.ONEDRIVE_FOLDER_ID,
|
|
"redirect_uri": cls.ONEDRIVE_REDIRECT_URI,
|
|
}
|
|
return {}
|