From 042a56c8b5997d69406849505a39ea4e11c023ea Mon Sep 17 00:00:00 2001 From: Jakub Miazek Date: Sat, 8 Jun 2024 14:05:57 +0200 Subject: [PATCH] set postgresql dialect for compile_sql_or_scalar decorator --- app/models/stuff.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/app/models/stuff.py b/app/models/stuff.py index 006640b..f842810 100644 --- a/app/models/stuff.py +++ b/app/models/stuff.py @@ -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()