Workaround SQLAlchemy 2.0.42 DOMAIN type adaptation regression (#435)

* Bugfix

* Added CHANGES.rst entry

* Comment fix

Author: sheinbergon

CHANGES.rst

@@ -1,6 +1,9 @@
 Version history
 ===============
 
+**UNRELEASED**
+- Fix Postgres ``DOMAIN`` adaptation regression introduced in SQLAlchemy 2.0.42 (PR by @sheinbergon)
+
 **3.1.1**
 
 - Fallback ``NotImplemented`` errors encountered when accessing ``python_type`` for

pyproject.toml

@@ -29,7 +29,7 @@ classifiers = [
 ]
 requires-python = ">=3.9"
 dependencies = [
-    "SQLAlchemy >= 2.0.29,<2.0.42",
+    "SQLAlchemy >= 2.0.29",
     "inflect >= 4.0.0",
     "importlib_metadata; python_version < '3.10'",
     "stdlib-list; python_version < '3.10'"

src/sqlacodegen/generators.py

@@ -718,6 +718,20 @@ def get_adapted_type(self, coltype: Any) -> Any:
                     if coltype.schema:
                         kw["schema"] = coltype.schema
 
+                # Hack to fix Postgres DOMAIN type adaptation, broken as of SQLAlchemy 2.0.42
+                # For additional information - https://github.com/agronholm/sqlacodegen/issues/416#issuecomment-3417480599
+                if supercls is DOMAIN:
+                    if coltype.default:
+                        kw["default"] = coltype.default
+                    if coltype.constraint_name is not None:
+                        kw["constraint_name"] = coltype.constraint_name
+                    if coltype.not_null:
+                        kw["not_null"] = coltype.not_null
+                    if coltype.check is not None:
+                        kw["check"] = coltype.check
+                    if coltype.create_type:
+                        kw["create_type"] = coltype.create_type
+
                 try:
                     new_coltype = coltype.adapt(supercls)
                 except TypeError: