anti
172a002d41
- Add `get_repository()` factory function to select DB implementation at runtime via DECNET_DB_TYPE env var - Extract BaseRepository abstract interface from SQLiteRepository - Update dependencies to use factory-based repository injection - Add DECNET_DB_TYPE env var support (defaults to sqlite) - Refactor models and repository base class for cross-dialect compatibility
30 lines
981 B
Python
30 lines
981 B
Python
"""
|
|
Repository factory — selects a :class:`BaseRepository` implementation based on
|
|
``DECNET_DB_TYPE`` (``sqlite`` or ``mysql``).
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from typing import Any
|
|
|
|
from decnet.web.db.repository import BaseRepository
|
|
|
|
|
|
def get_repository(**kwargs: Any) -> BaseRepository:
|
|
"""Instantiate the repository implementation selected by ``DECNET_DB_TYPE``.
|
|
|
|
Keyword arguments are forwarded to the concrete implementation:
|
|
|
|
* SQLite accepts ``db_path``.
|
|
* MySQL accepts ``url`` and engine tuning knobs (``pool_size``, …).
|
|
"""
|
|
db_type = os.environ.get("DECNET_DB_TYPE", "sqlite").lower()
|
|
|
|
if db_type == "sqlite":
|
|
from decnet.web.db.sqlite.repository import SQLiteRepository
|
|
return SQLiteRepository(**kwargs)
|
|
if db_type == "mysql":
|
|
from decnet.web.db.mysql.repository import MySQLRepository
|
|
return MySQLRepository(**kwargs)
|
|
raise ValueError(f"Unsupported database type: {db_type}")
|