Add Alembic to project for database migrations
This commit is contained in:
parent
7755bfb46d
commit
7ce43a2bfe
15 changed files with 256 additions and 49 deletions
|
|
@ -1,3 +1 @@
|
|||
from .metadata import metadata_obj
|
||||
from .model import Model
|
||||
from .sqlalchemy import SQLAlchemy
|
||||
|
|
|
|||
|
|
@ -1,17 +0,0 @@
|
|||
from sqlalchemy import MetaData
|
||||
|
||||
__all__ = [
|
||||
'metadata_obj',
|
||||
]
|
||||
|
||||
# Define naming convention for constraints
|
||||
_naming_convention = {
|
||||
"ix": 'ix_%(column_0_label)s',
|
||||
"uq": "uq_%(table_name)s_%(column_0_name)s",
|
||||
"ck": "ck_%(table_name)s_%(constraint_name)s",
|
||||
"fk": "fk_%(table_name)s_%(column_0_name)s_%(referred_table_name)s",
|
||||
"pk": "pk_%(table_name)s"
|
||||
}
|
||||
|
||||
# Create global metadata object for database schemas
|
||||
metadata_obj = MetaData(naming_convention=_naming_convention)
|
||||
|
|
@ -1,10 +0,0 @@
|
|||
from sqlalchemy.orm import declarative_base
|
||||
|
||||
from .metadata import metadata_obj
|
||||
|
||||
__all__ = [
|
||||
'Model',
|
||||
]
|
||||
|
||||
# Generate declarative base class for database models
|
||||
Model = declarative_base(name='Model', metadata=metadata_obj)
|
||||
|
|
@ -1,12 +1,11 @@
|
|||
from typing import Optional, cast
|
||||
|
||||
from flask import Flask
|
||||
from sqlalchemy import MetaData, create_engine
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.engine import Engine
|
||||
from sqlalchemy.orm import Session, scoped_session, sessionmaker
|
||||
|
||||
from tofu_api.common.config import Config
|
||||
from .metadata import metadata_obj
|
||||
|
||||
__all__ = [
|
||||
'SQLAlchemy',
|
||||
|
|
@ -70,22 +69,3 @@ class SQLAlchemy:
|
|||
# For all further purposes, the scoped session should be treated like a regular Session object.
|
||||
# Use cast() so we can use Session as the type annotation.
|
||||
return cast(Session, self._scoped_session)
|
||||
|
||||
@property
|
||||
def metadata(self) -> MetaData:
|
||||
"""
|
||||
Database metadata object.
|
||||
"""
|
||||
return metadata_obj
|
||||
|
||||
def create_all_tables(self) -> None:
|
||||
"""
|
||||
Create tables in the database for all models defined in the metadata.
|
||||
"""
|
||||
self.metadata.create_all(self.engine)
|
||||
|
||||
def drop_all_tables(self) -> None:
|
||||
"""
|
||||
Delete tables in the database for all models defined in the metadata.
|
||||
"""
|
||||
self.metadata.drop_all(self.engine)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue