lorebase-store 0.2.3__tar.gz

lorebase_store-0.2.3/.gitignore

@@ -0,0 +1,79 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+dist/
+*.egg-info/
+.eggs/
+.venv/
+venv/
+.python-eggs/
+pip-wheel-metadata/
+.pytest_cache/
+.mypy_cache/
+.ruff_cache/
+.pyre/
+.pytype/
+.hypothesis/
+.coverage
+.coverage.*
+coverage.xml
+htmlcov/
+.tox/
+.nox/
+*.prof
+
+# Node
+node_modules/
+.pnpm-store/
+.npm/
+.yarn/
+!.yarn/releases
+.pnp.*
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+pnpm-debug.log*
+.vite/
+*.tsbuildinfo
+apps/web/dist/
+apps/web/.vitest/
+
+# Environment and secrets
+.env
+.env.*
+!.env.example
+*.pem
+*.key
+secrets/
+
+# Docker
+docker-compose.override.yaml
+docker-compose.override.yml
+.docker/
+
+# Data and local artifacts
+*.sqlite
+*.db
+/tmp/
+/data/
+tests/perf/results/
+
+# Editors and OS
+.idea/
+.vscode/*
+!.vscode/extensions.json
+!.vscode/settings.example.json
+*.swp
+*.swo
+*~
+.DS_Store
+Thumbs.db
+*.code-workspace
+.history/
+
+# Session worktrees
+.claude/

lorebase_store-0.2.3/PKG-INFO

@@ -0,0 +1,13 @@
+Metadata-Version: 2.5
+Name: lorebase-store
+Version: 0.2.3
+Summary: Lorebase store: persistence, repositories, migrations and materializer
+Requires-Python: >=3.12
+Requires-Dist: alembic>=1.13
+Requires-Dist: asyncpg>=0.29
+Requires-Dist: cryptography>=42
+Requires-Dist: lorebase-core<0.3,>=0.2
+Requires-Dist: lorebase-platform<0.3,>=0.2
+Requires-Dist: opentelemetry-api>=1.25
+Requires-Dist: pgvector>=0.3
+Requires-Dist: sqlalchemy[asyncio]>=2.0.30

lorebase_store-0.2.3/alembic.ini

@@ -0,0 +1,41 @@
+# Alembic configuration for the Lorebase store. The database URL and every other value come
+# from the `store` settings section (LOREBASE_CONFIG, LOREBASE__STORE__*), never from this file.
+# Usage from the repository root: uv run alembic -c packages/store/alembic.ini upgrade head
+
+[alembic]
+script_location = %(here)s/migrations
+path_separator = os
+
+[loggers]
+keys = root,sqlalchemy,alembic
+
+[handlers]
+keys = console
+
+[formatters]
+keys = generic
+
+[logger_root]
+level = WARNING
+handlers = console
+qualname =
+
+[logger_sqlalchemy]
+level = WARNING
+handlers =
+qualname = sqlalchemy.engine
+
+[logger_alembic]
+level = INFO
+handlers =
+qualname = alembic
+
+[handler_console]
+class = StreamHandler
+args = (sys.stderr,)
+level = NOTSET
+formatter = generic
+
+[formatter_generic]
+format = %(levelname)-5.5s [%(name)s] %(message)s
+datefmt = %H:%M:%S

lorebase_store-0.2.3/migrations/env.py

@@ -0,0 +1,106 @@
+"""Alembic environment: async engine from the `database` section, layout from `store`.
+
+`lorebase_store.migrations` may pass settings and a URL override through `config.attributes`;
+anything missing (always, for `alembic upgrade head` on the command line) comes from
+`load_settings()`, so `LOREBASE__DATABASE__DSN` and `LOREBASE_CONFIG` apply.
+"""
+
+from __future__ import annotations
+
+import asyncio
+from logging.config import fileConfig
+from typing import Any
+
+from alembic import context
+from alembic.migration import MigrationContext
+from alembic.script import ScriptDirectory
+from sqlalchemy.engine import Connection
+
+from lorebase_platform.settings import load_settings
+from lorebase_store.db import async_url, create_engine
+from lorebase_store.migrations import (
+    DATABASE_ATTRIBUTE,
+    STORE_ATTRIBUTE,
+    URL_ATTRIBUTE,
+    check_known_revisions,
+    include_object,
+)
+from lorebase_store.models import metadata
+from lorebase_store.settings import (
+    DatabaseSettings,
+    database_settings,
+    store_settings,
+)
+
+config = context.config
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name, disable_existing_loggers=False)
+
+if DATABASE_ATTRIBUTE not in config.attributes or STORE_ATTRIBUTE not in config.attributes:
+    loaded = load_settings()
+    config.attributes.setdefault(DATABASE_ATTRIBUTE, database_settings(loaded))
+    config.attributes.setdefault(STORE_ATTRIBUTE, store_settings(loaded))
+database: DatabaseSettings = config.attributes[DATABASE_ATTRIBUTE]
+url: str = config.attributes.get(URL_ATTRIBUTE) or database.dsn.get_secret_value()
+
+
+def _configure(**options: Any) -> None:
+    context.configure(
+        target_metadata=metadata,
+        include_object=include_object,
+        compare_type=True,
+        transaction_per_migration=True,
+        **options,
+    )
+
+
+def run_offline() -> None:
+    """Emit SQL instead of executing it."""
+    _configure(
+        url=async_url(url).render_as_string(hide_password=False),
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+    )
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def _invoked_command() -> str | None:
+    """Name of the `alembic` command line command being run; None when called from code."""
+    invocation = getattr(config.cmd_opts, "cmd", None)
+    return getattr(invocation[0], "__name__", None) if invocation else None
+
+
+def _run_sync(connection: Connection) -> None:
+    # `stamp --purge` is the manual way to repair the version table, so it is left alone.
+    if _invoked_command() != "stamp":
+        current = MigrationContext.configure(connection).get_current_heads()
+        # The read began a transaction; end it so that migrations manage their own (some run
+        # steps in autocommit blocks).
+        if connection.in_transaction():
+            connection.rollback()
+        check_known_revisions(ScriptDirectory.from_config(config), current)
+    _configure(connection=connection)
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+async def run_online() -> None:
+    """Run migrations on a dedicated, unpooled connection."""
+    engine = create_engine(
+        database,
+        url=url,
+        pooled=False,
+        statement_timeout_ms=database.migration_statement_timeout_ms,
+    )
+    try:
+        async with engine.connect() as connection:
+            await connection.run_sync(_run_sync)
+    finally:
+        await engine.dispose()
+
+
+if context.is_offline_mode():
+    run_offline()
+else:
+    asyncio.run(run_online())

lorebase_store-0.2.3/migrations/script.py.mako

@@ -0,0 +1,27 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+"""
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+from alembic import op
+${imports if imports else ""}
+
+revision: str = ${repr(up_revision)}
+down_revision: str | Sequence[str] | None = ${repr(down_revision)}
+branch_labels: str | Sequence[str] | None = ${repr(branch_labels)}
+depends_on: str | Sequence[str] | None = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    ${downgrades if downgrades else "pass"}