__init__.py
python
| 1 | """ |
| 2 | Database module for MuseHub. |
| 3 | |
| 4 | Provides async SQLAlchemy support with PostgreSQL and SQLite. |
| 5 | """ |
| 6 | from __future__ import annotations |
| 7 | |
| 8 | from musehub.db.database import ( |
| 9 | get_db, |
| 10 | init_db, |
| 11 | close_db, |
| 12 | AsyncSessionLocal, |
| 13 | ) |
| 14 | from musehub.db.models import User, AccessToken |
| 15 | from musehub.db import muse_cli_models as muse_cli_models # noqa: F401 — register CLI commit tables with Base |
| 16 | from musehub.db import musehub_models as musehub_models # noqa: F401 — register with Base |
| 17 | from musehub.db import musehub_label_models as musehub_label_models # noqa: F401 — register with Base |
| 18 | from musehub.db import musehub_collaborator_models as musehub_collaborator_models # noqa: F401 — register with Base |
| 19 | from musehub.db import musehub_stash_models as musehub_stash_models # noqa: F401 — register with Base |
| 20 | |
| 21 | __all__ = [ |
| 22 | "get_db", |
| 23 | "init_db", |
| 24 | "close_db", |
| 25 | "AsyncSessionLocal", |
| 26 | "User", |
| 27 | "AccessToken", |
| 28 | ] |