bvandeusenandClaude Opus 5.5
6ec78b530e
routes/search states the derived-facets property rather than the silent whole-corpus bug that motivated it; routes/rulebooks gives the preference restore exception its reasoning without the milestone numbers it cited. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
98 lines
5.1 KiB
Python
98 lines
5.1 KiB
Python
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
|
|
from sqlalchemy.orm import DeclarativeBase
|
|
|
|
from scribe.config import Config
|
|
|
|
# Named rather than inlined so the deadline below is READABLE. SQLAlchemy
|
|
# captures `connect_args` in a closure and merges it at connect time, so an
|
|
# inline dict cannot be recovered from the engine — and a guard that cannot
|
|
# read the value it guards is a guard that passes forever.
|
|
_CONNECT_ARGS: dict = {
|
|
# A DEADLINE ON ESTABLISHING A CONNECTION.
|
|
#
|
|
# asyncpg's `timeout` bounds the CONNECT — the TCP handshake plus session
|
|
# setup — and nothing else. Against a host whose storage has wedged, that
|
|
# handshake does not fail, it waits, and without this the wait is
|
|
# asyncpg's own 60-second default: the entire lifespan budget spent before
|
|
# a single query is even sent. Ten seconds is far longer than a healthy
|
|
# local connect (single-digit milliseconds) and short enough to leave room
|
|
# to fail usefully rather than be killed.
|
|
#
|
|
# WHAT THIS DOES NOT COVER, said plainly so the next reader doesn't assume
|
|
# it does: a query on an already-open connection, which includes
|
|
# `pool_pre_ping`'s liveness check. Bounding those is `command_timeout`,
|
|
# and that is deliberately NOT set here — it would apply to every
|
|
# statement, and this app legitimately runs long ones (the embedding
|
|
# backfills, VACUUM ANALYZE). A blanket statement deadline would trade
|
|
# this failure mode for a worse one. Callers that must not hang — the
|
|
# lifespan hook above all — bound their own await instead; see
|
|
# `_STARTUP_READ_TIMEOUT` in services/db_maintenance_scheduler.py.
|
|
"timeout": 10,
|
|
}
|
|
|
|
engine = create_async_engine(
|
|
Config.DATABASE_URL,
|
|
echo=False,
|
|
pool_pre_ping=True,
|
|
pool_recycle=1800,
|
|
connect_args=_CONNECT_ARGS,
|
|
)
|
|
async_session = async_sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
pass
|
|
|
|
|
|
from scribe.models.base import CreatedAtMixin, TimestampMixin # noqa: E402, F401
|
|
|
|
|
|
from scribe.models.note import Note, TaskPriority, TaskStatus # noqa: E402, F401
|
|
from scribe.models.setting import Setting # noqa: E402, F401
|
|
from scribe.models.user import User # noqa: E402, F401
|
|
from scribe.models.app_log import AppLog # noqa: E402, F401
|
|
from scribe.models.password_reset import PasswordResetToken # noqa: E402, F401
|
|
from scribe.models.invitation import InvitationToken # noqa: E402, F401
|
|
from scribe.models.embedding import ( # noqa: E402, F401
|
|
MilestoneEmbedding, NoteEmbedding, RuleEmbedding, SystemEmbedding,
|
|
)
|
|
from scribe.models.retrieval_log import RetrievalLog # noqa: E402, F401
|
|
from scribe.models.retrieval_tuning import RetrievalTuningEvent # noqa: E402, F401
|
|
from scribe.models.note_usage import NoteUsageEvent # noqa: E402, F401
|
|
from scribe.models.rule_usage import RuleUsageEvent # noqa: E402, F401
|
|
from scribe.models.system_usage import SystemUsageEvent # noqa: E402, F401
|
|
from scribe.models.retrieval_judgment import RetrievalJudgment # noqa: E402, F401
|
|
from scribe.models.moment_mapping import MomentMapping # noqa: E402, F401
|
|
from scribe.models.project import Project # noqa: E402, F401
|
|
from scribe.models.milestone import Milestone # noqa: E402, F401
|
|
from scribe.models.task_log import TaskLog # noqa: E402, F401
|
|
from scribe.models.note_draft import NoteDraft # noqa: E402, F401
|
|
from scribe.models.note_version import NoteVersion # noqa: E402, F401
|
|
from scribe.models.rule_version import RuleVersion # noqa: E402, F401
|
|
from scribe.models.note_supersession import NoteSupersession # noqa: E402, F401
|
|
from scribe.models.group import Group, GroupMembership # noqa: E402, F401
|
|
from scribe.models.share import NoteShare, ProjectShare # noqa: E402, F401
|
|
from scribe.models.notification import Notification # noqa: E402, F401
|
|
from scribe.models.api_key import ApiKey # noqa: E402, F401
|
|
from scribe.models.user_profile import UserProfile # noqa: E402, F401
|
|
# Imported before rulebook: rule_systems foreign-keys canonical_systems.
|
|
from scribe.models.canonical_system import CanonicalSystem # noqa: E402, F401
|
|
from scribe.models.rulebook import ( # noqa: E402, F401
|
|
Rulebook, RulebookTopic, Rule, RuleRelation, rule_moments, rule_systems,
|
|
)
|
|
# After notes and rules: it foreign-keys both.
|
|
from scribe.models.lesson_rule_link import LessonNoRule, LessonRuleLink # noqa: E402, F401
|
|
# After rules: it foreign-keys them.
|
|
from scribe.models.rule_moment_judgment import RuleMomentJudgment # noqa: E402, F401
|
|
from scribe.models.repo_binding import RepoBinding # noqa: E402, F401
|
|
from scribe.models.forge_connection import ForgeConnection # noqa: E402, F401
|
|
from scribe.models.code_shape import CodeShape, CodeShapeConsumer, CodeShapeEvent, CodeShapeUse # noqa: E402, F401
|
|
from scribe.models.system import System, RecordSystem # noqa: E402, F401
|
|
from scribe.models.design_system import DesignSystem, DesignToken # noqa: E402, F401
|
|
# After notes, projects and rulebook topics: family canon foreign-keys all
|
|
# three.
|
|
from scribe.models.family import ( # noqa: E402, F401
|
|
FamilyAdoption, FamilyDecision, FamilyIdea, FamilyIdeaPlatform,
|
|
FamilyIdeaReference, Platform, ProjectPlatform,
|
|
)
|