fix: Cleaning up the code base
This commit is contained in:
@@ -8,3 +8,4 @@ __version__ = "1.0.0"
|
||||
|
||||
from .engine import engine, async_engine
|
||||
from .session import SessionLocal, AsyncSessionLocal, get_db, get_async_db
|
||||
from .models import Base
|
||||
|
||||
@@ -1,13 +1,13 @@
|
||||
from sqlalchemy.ext.asyncio import create_async_engine, AsyncEngine
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy import create_engine, Engine
|
||||
from app.core.config import settings
|
||||
|
||||
# =========================
|
||||
# SYNCHRONOUS ENGINE
|
||||
# =========================
|
||||
# Pour opérations sync classiques
|
||||
engine = create_engine(
|
||||
settings.DATABASE_URL.replace("+aiomysql", ""), # remove async part if present
|
||||
engine: Engine = create_engine(
|
||||
settings.database_url.replace('<driver>', "pymysql"),
|
||||
echo=settings.DEBUG,
|
||||
future=True
|
||||
)
|
||||
@@ -17,7 +17,7 @@ engine = create_engine(
|
||||
# =========================
|
||||
# Pour opérations async avec async SQLAlchemy
|
||||
async_engine: AsyncEngine = create_async_engine(
|
||||
settings.DATABASE_URL if "+aiomysql" in settings.DATABASE_URL else settings.DATABASE_URL.replace("mysql+pymysql", "mysql+aiomysql"),
|
||||
settings.database_url.replace('<driver>', "aiomysql"),
|
||||
echo=settings.DEBUG,
|
||||
future=True
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user