Skip to content

Commit 042a56c

Browse files
committed
set postgresql dialect for compile_sql_or_scalar decorator
1 parent b938b0d commit 042a56c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

app/models/stuff.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import uuid
22

3+
from sqlalchemy.dialects import postgresql
34
from sqlalchemy import String, select, ForeignKey
45
from sqlalchemy.dialects.postgresql import UUID
56
from sqlalchemy.ext.asyncio import AsyncSession
@@ -16,7 +17,7 @@ def compile_sql_or_scalar(func):
1617
async def wrapper(cls, db_session, name, compile_sql=False, *args, **kwargs):
1718
stmt = await func(cls, db_session, name, *args, **kwargs)
1819
if compile_sql:
19-
return stmt.compile(compile_kwargs={"literal_binds": True})
20+
return stmt.compile(dialect=postgresql.dialect(), compile_kwargs={"literal_binds": True})
2021
result = await db_session.execute(stmt)
2122
return result.scalars().first()
2223

0 commit comments

Comments
 (0)