From d7ca9e1e8c17934950b1a2edb7e85c03bfd88af0 Mon Sep 17 00:00:00 2001
From: Matthieu Rigal <matthieu.rigal@a2ei.org>
Date: Thu, 21 Nov 2024 14:37:39 +0100
Subject: [PATCH] fix: Support sequences on TimescaleDB like on PostgreSQL

---
 src/sqlacodegen/generators.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/sqlacodegen/generators.py b/src/sqlacodegen/generators.py
index 21eadb63..b41c0445 100644
--- a/src/sqlacodegen/generators.py
+++ b/src/sqlacodegen/generators.py
@@ -72,7 +72,7 @@
 _re_enum_check_constraint = re.compile(r"(?:.*?\.)?(.*?) IN \((.+)\)")
 _re_enum_item = re.compile(r"'(.*?)(?<!\\)'")
 _re_invalid_identifier = re.compile(r"(?u)\W")
-
+POSTGRES_LIKE_NAMES = ("postgresql", "timescaledb",)
 
 @dataclass
 class LiteralImport:
@@ -646,7 +646,7 @@ def fix_column_types(self, table: Table) -> None:
                 pass
 
             # PostgreSQL specific fix: detect sequences from server_default
-            if column.server_default and self.bind.dialect.name == "postgresql":
+            if column.server_default and self.bind.dialect.name in POSTGRES_LIKE_NAMES:
                 if isinstance(column.server_default, DefaultClause) and isinstance(
                     column.server_default.arg, TextClause
                 ):