Added workaround for missing python_type in postgres UUID column

Fixes #184.

Author: agronholm

CHANGES.rst

@@ -5,6 +5,7 @@ Version history
 
 - Migrated all packaging/testing configuration to ``pyproject.toml``
 . Fixed ``KeyError`` when rendering an index without any columns
+- Worked around PostgreSQL UUID columns getting ``Any`` as the type annotation
 
 **3.0.0b3**
 

src/sqlacodegen/generators.py

@@ -1231,7 +1231,9 @@ def collect_imports_for_column(self, column: Column[Any]) -> None:
         try:
             python_type = column.type.python_type
         except NotImplementedError:
-            self.add_literal_import("typing", "Any")
+            if column.type.__class__.__name__ != "UUID":
+                # Workaround for https://github.com/sqlalchemy/sqlalchemy/issues/7943
+                self.add_literal_import("typing", "Any")
         else:
             self.add_import(python_type)
 
@@ -1266,7 +1268,11 @@ def render_column_attribute(self, column_attr: ColumnAttribute) -> str:
         try:
             python_type = column.type.python_type
         except NotImplementedError:
-            python_type_name = "Any"
+            if column.type.__class__.__name__ == "UUID":
+                # Workaround for https://github.com/sqlalchemy/sqlalchemy/issues/7943
+                python_type_name = "str"
+            else:
+                python_type_name = "Any"
         else:
             python_type_name = python_type.__name__
 

tests/test_generators.py

@@ -6,6 +6,7 @@
 from _pytest.fixtures import FixtureRequest
 from sqlalchemy import PrimaryKeyConstraint
 from sqlalchemy.dialects import mysql, postgresql
+from sqlalchemy.dialects.postgresql import UUID
 from sqlalchemy.engine import Engine, create_engine
 from sqlalchemy.schema import (
     CheckConstraint,
@@ -2393,3 +2394,34 @@ class SimpleItems:
                 container: Optional[SimpleContainers] = field(default=None, metadata={'sa': relationship('SimpleContainers', back_populates='simple_items')})
             """,
         )
+
+    def test_uuid_type_annotation(self, generator: CodeGenerator) -> None:
+        Table(
+            "simple",
+            generator.metadata,
+            Column("id", UUID, primary_key=True),
+        )
+
+        validate_code(
+            generator.generate(),
+            """\
+            from __future__ import annotations
+
+            from dataclasses import dataclass, field
+
+            from sqlalchemy import Column
+            from sqlalchemy.dialects.postgresql import UUID
+            from sqlalchemy.orm import registry
+
+            mapper_registry = registry()
+
+
+            @mapper_registry.mapped
+            @dataclass
+            class Simple:
+                __tablename__ = 'simple'
+                __sa_dataclass_metadata_key__ = 'sa'
+
+                id: str = field(init=False, metadata={'sa': Column(UUID, primary_key=True)})
+            """,
+        )