Added support for Geoalchemy2

Fixes #29.
agronholm · May 7, 2018 · c3f51d8 · c3f51d8
1 parent 5e1bd42
commit c3f51d8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -14,6 +14,8 @@ Version history
 
 * Added an option to only generate tables
 
+* Added support for Geoalchemy2
+
 
 1.1.6
 -----

diff --git a/sqlacodegen/codegen.py b/sqlacodegen/codegen.py
@@ -17,6 +17,12 @@
 from sqlalchemy.types import Boolean, String
 from sqlalchemy.util import OrderedDict
 
+# Conditionally import Geoalchemy2 to enable reflection support
+try:
+    import geoalchemy2  # noqa: F401
+except ImportError:
+    pass
+
 _re_boolean_check_constraint = re.compile(r"(?:(?:.*?)\.)?(.*?) IN \(0, 1\)")
 _re_column_name = re.compile(r'(?:(["`]?)(?:.*)\1\.)?(["`]?)(.*)\2')
 _re_enum_check_constraint = re.compile(r"(?:(?:.*?)\.)?(.*?) IN \((.+)\)")