From 7c8d988f571f78e8b005d147e36285c8400ed0b4 Mon Sep 17 00:00:00 2001
From: FeeeeK <26704473+FeeeeK@users.noreply.github.com>
Date: Tue, 10 Dec 2024 02:29:55 +0300
Subject: [PATCH] Add missing SmallInteger column spec for asyncpg; Fixes:
 #12170

---
 lib/sqlalchemy/dialects/postgresql/asyncpg.py | 3 +++
 1 file changed, 3 insertions(+)

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,
-- 
2.47.3