-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
54 changed files
with
438 additions
and
38 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -61,6 +61,7 @@ known-local-folder = [ | |
"auth", | ||
"config", | ||
"core", | ||
"filmin", | ||
"infra", | ||
"shared", | ||
"utils", | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
File renamed without changes.
Empty file.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,101 @@ | ||
import asyncio | ||
|
||
from logging.config import fileConfig | ||
|
||
from alembic import context | ||
from sqlalchemy import engine_from_config, pool | ||
from sqlalchemy.engine import Connection | ||
from sqlalchemy.ext.asyncio import AsyncEngine | ||
|
||
from config import settings | ||
import infra.database.sqlalchemy.models.core | ||
import infra.database.sqlalchemy.models.filmin # noqa | ||
|
||
from infra.database.sqlalchemy.sqlalchemy import metadata | ||
|
||
|
||
# this is the Alembic Config object, which provides | ||
# access to the values within the .ini file in use. | ||
config = context.config | ||
|
||
config.set_main_option('sqlalchemy.url', settings.DATABASE_URL) | ||
|
||
# Interpret the config file for Python logging. | ||
# This line sets up loggers basically. | ||
fileConfig(config.config_file_name) | ||
|
||
|
||
# add your model's MetaData object here | ||
# for 'autogenerate' support | ||
# from myapp import mymodel | ||
# target_metadata = mymodel.Base.metadata | ||
# target_metadata = None | ||
|
||
|
||
target_metadata = metadata | ||
|
||
|
||
# other values from the config, defined by the needs of env.py, | ||
# can be acquired: | ||
# my_important_option = config.get_main_option("my_important_option") | ||
# ... etc. | ||
|
||
|
||
def run_migrations_offline() -> None: | ||
"""Run migrations in 'offline' mode. | ||
This configures the context with just a URL | ||
and not an Engine, though an Engine is acceptable | ||
here as well. By skipping the Engine creation | ||
we don't even need a DBAPI to be available. | ||
Calls to context.execute() here emit the given string to the | ||
script output. | ||
""" | ||
url = config.get_main_option('sqlalchemy.url') | ||
|
||
context.configure( | ||
url=url, | ||
target_metadata=target_metadata, | ||
literal_binds=True, | ||
dialect_opts={'paramstyle': 'named'}, | ||
) | ||
|
||
with context.begin_transaction(): | ||
context.run_migrations() | ||
|
||
|
||
def do_run_migrations(connection: Connection) -> None: | ||
context.configure(connection=connection, target_metadata=target_metadata) | ||
|
||
with context.begin_transaction(): | ||
context.run_migrations() | ||
|
||
|
||
async def run_migrations_online() -> None: | ||
"""Run migrations in 'online' mode. | ||
In this scenario we need to create an Engine | ||
and associate a connection with the context. | ||
""" | ||
connectable = AsyncEngine( | ||
engine_from_config( | ||
config.get_section(config.config_ini_section), | ||
prefix='sqlalchemy.', | ||
poolclass=pool.NullPool, | ||
future=True, | ||
) | ||
) | ||
|
||
async with connectable.connect() as connection: | ||
await connection.run_sync(do_run_migrations) | ||
|
||
await connectable.dispose() | ||
|
||
|
||
if context.is_offline_mode(): | ||
run_migrations_offline() | ||
else: | ||
asyncio.run(run_migrations_online()) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,24 @@ | ||
"""${message} | ||
|
||
Revision ID: ${up_revision} | ||
Revises: ${down_revision | comma,n} | ||
Create Date: ${create_date} | ||
|
||
""" | ||
from alembic import op | ||
import sqlalchemy as sa | ||
${imports if imports else ""} | ||
|
||
# revision identifiers, used by Alembic. | ||
revision = ${repr(up_revision)} | ||
down_revision = ${repr(down_revision)} | ||
branch_labels = ${repr(branch_labels)} | ||
depends_on = ${repr(depends_on)} | ||
|
||
|
||
def upgrade() -> None: | ||
${upgrades if upgrades else "pass"} | ||
|
||
|
||
def downgrade() -> None: | ||
${downgrades if downgrades else "pass"} |
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
Empty file.
File renamed without changes.
File renamed without changes.
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,80 @@ | ||
import abc | ||
|
||
from asyncio import current_task | ||
from contextlib import asynccontextmanager | ||
import contextvars | ||
import logging | ||
from typing import TYPE_CHECKING, Any, AsyncContextManager, AsyncIterator, Dict, Optional, cast | ||
|
||
from sqlalchemy.ext.asyncio import AsyncSession, async_scoped_session, create_async_engine | ||
from sqlalchemy.orm import sessionmaker | ||
|
||
from config import settings | ||
|
||
|
||
if TYPE_CHECKING: | ||
from sqlalchemy.ext.asyncio.engine import AsyncEngine | ||
|
||
|
||
logger = logging.getLogger(__name__) | ||
|
||
|
||
db_session_context: contextvars.ContextVar = contextvars.ContextVar('db_ctx', default={'session': None, 'level': 0}) | ||
|
||
|
||
class AbstractDatabase(abc.ABC): | ||
@abc.abstractmethod | ||
def session(self) -> AsyncContextManager[AsyncSession]: | ||
... | ||
|
||
|
||
def _get_current_task_id() -> int: | ||
return id(current_task()) | ||
|
||
|
||
# @singleton | ||
class Database(AbstractDatabase): | ||
def __init__(self) -> None: | ||
self.engine: AsyncEngine = create_async_engine(settings.DATABASE_URL, echo=True, future=True) | ||
session_local: sessionmaker = sessionmaker( | ||
self.engine, class_=AsyncSession, autocommit=False, autoflush=False, expire_on_commit=False | ||
) | ||
# SessionLocal = sessionmaker(bind=engine, autocommit=False, autoflush=False, class_=AsyncSession) | ||
self._session_factory = async_scoped_session(session_local, scopefunc=_get_current_task_id) | ||
|
||
@asynccontextmanager | ||
async def session(self) -> AsyncIterator[AsyncSession]: | ||
db_session: Optional[Dict[str, Any]] = None | ||
db_session = db_session_context.get() or {'session': None, 'level': 0} | ||
if db_session['level'] == 0: | ||
session: AsyncSession = cast(AsyncSession, self._session_factory()) | ||
db_session['session'] = session | ||
# await session.begin() | ||
logger.debug('session begin', extra={'level': db_session['level']}) | ||
|
||
else: | ||
session = db_session['session'] | ||
db_session['level'] = (db_session['level'] or 0) + 1 | ||
db_session_context.set(db_session) | ||
|
||
try: | ||
yield session | ||
except Exception: | ||
logger.exception('Session rollback because of exception') | ||
await session.rollback() | ||
logger.debug('session rollback') | ||
raise | ||
else: | ||
# db_session = db_session_context.get() or {'session': None, 'level': 0} | ||
if db_session['level'] == 0: | ||
await session.commit() | ||
logger.debug('session commit', extra={'level': db_session['level']}) | ||
finally: | ||
# db_session = db_session_context.get() or {'session': None, 'level': 0} | ||
if db_session['level'] == 0: | ||
await session.close() | ||
logger.debug('session close', extra={'level': db_session['level']}) | ||
db_session_context.set(None) | ||
else: | ||
db_session['level'] = (db_session['level'] or 0) - 1 | ||
db_session_context.set(db_session) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,12 @@ | ||
from sqlalchemy.schema import MetaData | ||
|
||
|
||
metadata = MetaData( | ||
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', | ||
} | ||
) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.