31 lines
858 B
Python
31 lines
858 B
Python
"""路径与运行配置。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import re
|
|
from pathlib import Path
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
BASE_DIR = Path(__file__).resolve().parent.parent
|
|
load_dotenv(BASE_DIR / ".env")
|
|
|
|
OUTPUT_DIR = Path(os.getenv("AI_IMAGE_OUTPUT_DIR") or BASE_DIR / "output")
|
|
WEB_DIR = BASE_DIR / "web"
|
|
DATA_DIR = Path(os.getenv("AI_IMAGE_DATA_DIR") or BASE_DIR / "data")
|
|
DB_PATH = Path(os.getenv("AI_IMAGE_DB") or DATA_DIR / "ai_image.db")
|
|
SECRET_PATH = DATA_DIR / "secret.key"
|
|
SKILL_PATH = BASE_DIR / "SKILL.md"
|
|
|
|
HOST = os.getenv("AI_IMAGE_HOST", "127.0.0.1")
|
|
PORT = int(os.getenv("AI_IMAGE_PORT", "8765"))
|
|
|
|
IMAGE_SUFFIXES = {".png", ".jpg", ".jpeg", ".webp"}
|
|
PROFILE_RE = re.compile(r"^[\w-]{1,64}$")
|
|
|
|
|
|
def ensure_dirs() -> None:
|
|
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
|
|
DATA_DIR.mkdir(parents=True, exist_ok=True)
|