mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-10-03 00:41:19 +00:00
Wu Clan
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)
|