From: FeeeeK <26704473+FeeeeK@users.noreply.github.com>
Date: Mon, 9 Dec 2024 23:29:55 +0000 (+0300)
Subject: Add missing SmallInteger column spec for asyncpg; Fixes: #12170
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=7c8d988f571f78e8b005d147e36285c8400ed0b4;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

Add missing SmallInteger column spec for asyncpg; Fixes: #12170
---

diff --git a/lib/sqlalchemy/dialects/postgresql/asyncpg.py b/lib/sqlalchemy/dialects/postgresql/asyncpg.py
index d57c94a170..c3bc7b3bdb 100644
--- a/lib/sqlalchemy/dialects/postgresql/asyncpg.py
+++ b/lib/sqlalchemy/dialects/postgresql/asyncpg.py
@@ -281,6 +281,8 @@ class AsyncPgEnum(ENUM):
 class AsyncpgInteger(sqltypes.Integer):
     render_bind_cast = True
 
+class AsyncpgSmallInteger(sqltypes.SmallInteger):
+    render_bind_cast = True
 
 class AsyncpgBigInteger(sqltypes.BigInteger):
     render_bind_cast = True
@@ -1078,6 +1080,7 @@ class PGDialect_asyncpg(PGDialect):
             INTERVAL: AsyncPgInterval,
             sqltypes.Boolean: AsyncpgBoolean,
             sqltypes.Integer: AsyncpgInteger,
+            sqltypes.SmallInteger: AsyncpgSmallInteger,
             sqltypes.BigInteger: AsyncpgBigInteger,
             sqltypes.Numeric: AsyncpgNumeric,
             sqltypes.Float: AsyncpgFloat,