mirror of
https://github.com/fastapi-practices/fastapi_best_architecture.git
synced 2025-08-18 15:00:46 +08:00
e1edcade21
* WIP: add rbac authorization * Perform pre-commit fixes * add rbac route whitelist * add init test data user role associations * Restore database table id naming to fix generic crud base * Add database section value uniqueness settings * Update the test directory to tests * Update route_name file name to health_check * Split user auth and user action interfaces * Fix conflict between merge and current branch * Add pymysql dependencies * Fix RBAC authentication method * Add the select serialisation tool * Fix missing return messages due to global exception handler slicing * Update the user interface with associated relationships * Add items to be completed * Perform pre-commit fixes * Add pre-made routers * Paging data return structure optimisation * Split user auth and user interface tests * Fix user register test data structure error * Fix duplicate named test classes
59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
import sys
|
||
|
||
from fastapi import Depends
|
||
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
|
||
from typing_extensions import Annotated
|
||
|
||
from backend.app.common.log import log
|
||
from backend.app.core.conf import settings
|
||
from backend.app.database.base_class import MappedBase
|
||
|
||
"""
|
||
说明：SqlAlchemy
|
||
"""
|
||
|
||
SQLALCHEMY_DATABASE_URL = (
|
||
f'mysql+asyncmy://{settings.DB_USER}:{settings.DB_PASSWORD}@{settings.DB_HOST}:'
|
||
f'{settings.DB_PORT}/{settings.DB_DATABASE}?charset={settings.DB_CHARSET}'
|
||
)
|
||
|
||
try:
|
||
# 数据库引擎
|
||
async_engine = create_async_engine(SQLALCHEMY_DATABASE_URL, echo=settings.DB_ECHO, future=True, pool_pre_ping=True)
|
||
# log.success('数据库连接成功')
|
||
except Exception as e:
|
||
log.error('❌ 数据库链接失败 {}', e)
|
||
sys.exit()
|
||
else:
|
||
async_db_session = async_sessionmaker(bind=async_engine, autoflush=False, expire_on_commit=False)
|
||
|
||
|
||
async def get_db() -> AsyncSession:
|
||
"""
|
||
session 生成器
|
||
|
||
:return:
|
||
"""
|
||
session = async_db_session()
|
||
try:
|
||
yield session
|
||
except Exception as se:
|
||
await session.rollback()
|
||
raise se
|
||
finally:
|
||
await session.close()
|
||
|
||
|
||
# Session Annotated
|
||
CurrentSession = Annotated[AsyncSession, Depends(get_db)]
|
||
|
||
|
||
async def create_table():
|
||
"""
|
||
创建数据库表
|
||
"""
|
||
async with async_engine.begin() as coon:
|
||
await coon.run_sync(MappedBase.metadata.create_all)
|