PersonalPublic/fastapi-sqlalchemy-asyncpg
1
0
Fork 0
mirror of https://github.com/grillazz/fastapi-sqlalchemy-asyncpg.git synced 2025-08-26 16:40:40 +03:00
Code Issues Packages Projects Releases Wiki Activity

set postgresql dialect for compile_sql_or_scalar decorator

Browse Source
This commit is contained in:
Jakub Miazek 2024-06-08 14:05:57 +02:00
parent b938b0dee0
commit 042a56c8b5
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
app/models/stuff.py
View File

@ -1,5 +1,6 @@
import uuid
from sqlalchemy.dialects import postgresql
from sqlalchemy import String, select, ForeignKey
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.ext.asyncio import AsyncSession
@ -16,7 +17,7 @@ def compile_sql_or_scalar(func):
async def wrapper(cls, db_session, name, compile_sql=False, *args, **kwargs):
stmt = await func(cls, db_session, name, *args, **kwargs)
if compile_sql:
return stmt.compile(compile_kwargs={"literal_binds": True})
return stmt.compile(dialect=postgresql.dialect(), compile_kwargs={"literal_binds": True})
result = await db_session.execute(stmt)
return result.scalars().first()