zcbot/core/storage/__init__.py

37 lines
826 B
Python

"""§7 B 阶段:Storage 落 PG。
入口:
from core.storage import get_engine, session_scope, ensure_local_sentinel
from core.storage.models import User, Task, Message
ZCBOT_DB_URL 环境变量必填(本地连测试 / staging PG;SaaS 连生产 PG)。
未设置时 get_engine() 抛 RuntimeError 并指引设置。
"""
from .engine import (
ensure_local_sentinel,
get_engine,
session_scope,
)
from .models import SENTINEL_USER_ID
from .utils import (
NoSubtaskError,
check_no_subtask,
ensure_local_task_row,
get_task,
update_task,
upsert_task,
)
__all__ = [
"NoSubtaskError",
"SENTINEL_USER_ID",
"check_no_subtask",
"ensure_local_sentinel",
"ensure_local_task_row",
"get_engine",
"get_task",
"session_scope",
"update_task",
"upsert_task",
]