Fixed crash on unrecognized column types

agronholm · agronholm · commit aff2a5ce4935 · 2019-07-27T13:26:49.000+03:00
Fixes agronholm#72.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -7,6 +7,7 @@ UNRELEASED
 * Dropped support for Python 3.4
 * Added support for Python 3.7 and 3.8
 * Added support for SQLAlchemy 1.3
+* Fixed crash on unknown column types (``NullType``)
 
 
 2.0.1
diff --git a/sqlacodegen/codegen.py b/sqlacodegen/codegen.py
@@ -15,6 +15,7 @@
     Enum, ForeignKeyConstraint, PrimaryKeyConstraint, CheckConstraint, UniqueConstraint, Table,
     Column, Float)
 from sqlalchemy.schema import ForeignKey
+from sqlalchemy.sql.sqltypes import NullType
 from sqlalchemy.types import Boolean, String
 from sqlalchemy.util import OrderedDict
 
@@ -88,7 +89,8 @@ def __init__(self, table):
 
         # Adapt column types to the most reasonable generic types (ie. VARCHAR -> String)
         for column in table.columns:
-            column.type = self._get_adapted_type(column.type, column.table.bind)
+            if not isinstance(column.type, NullType):
+                column.type = self._get_adapted_type(column.type, column.table.bind)
 
     def _get_adapted_type(self, coltype, bind):
         compiled_type = coltype.compile(bind.dialect)
