mirror of
https://github.com/insistence/RuoYi-Vue3-FastAPI.git
synced 2026-09-21 12:52:27 +00:00
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from urllib.parse import quote_plus
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncAttrs, async_sessionmaker, create_async_engine
|
|
from sqlalchemy.orm import DeclarativeBase
|
|
|
|
from config.env import DataBaseConfig
|
|
|
|
ASYNC_SQLALCHEMY_DATABASE_URL = (
|
|
f'mysql+asyncmy://{DataBaseConfig.db_username}:{quote_plus(DataBaseConfig.db_password)}@'
|
|
f'{DataBaseConfig.db_host}:{DataBaseConfig.db_port}/{DataBaseConfig.db_database}'
|
|
)
|
|
if DataBaseConfig.db_type == 'postgresql':
|
|
ASYNC_SQLALCHEMY_DATABASE_URL = (
|
|
f'postgresql+asyncpg://{DataBaseConfig.db_username}:{quote_plus(DataBaseConfig.db_password)}@'
|
|
f'{DataBaseConfig.db_host}:{DataBaseConfig.db_port}/{DataBaseConfig.db_database}'
|
|
)
|
|
|
|
async_engine = create_async_engine(
|
|
ASYNC_SQLALCHEMY_DATABASE_URL,
|
|
echo=DataBaseConfig.db_echo,
|
|
max_overflow=DataBaseConfig.db_max_overflow,
|
|
pool_size=DataBaseConfig.db_pool_size,
|
|
pool_recycle=DataBaseConfig.db_pool_recycle,
|
|
pool_timeout=DataBaseConfig.db_pool_timeout,
|
|
)
|
|
AsyncSessionLocal = async_sessionmaker(autocommit=False, autoflush=False, bind=async_engine)
|
|
|
|
|
|
class Base(AsyncAttrs, DeclarativeBase):
|
|
pass
|