Update generators.py

Allow for spaces between enum options
agronholm · Sep 28, 2023 · fe5ed72 · fe5ed72
1 parent c68d2a8
commit fe5ed72
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/sqlacodegen/generators.py b/src/sqlacodegen/generators.py
@@ -68,9 +68,9 @@
     from importlib.metadata import version
 
 _sqla_version = tuple(int(x) for x in version("sqlalchemy").split(".")[:2])
-_re_boolean_check_constraint = re.compile(r"(?:.*?\.)?(.*?) IN \(0, 1\)")
+_re_boolean_check_constraint = re.compile(r"(?:.*?\.)?(.*?)\s+IN\s*\(0, 1\)")   
 _re_column_name = re.compile(r'(?:(["`]?).*\1\.)?(["`]?)(.*)\2')
-_re_enum_check_constraint = re.compile(r"(?:.*?\.)?(.*?) IN \((.+)\)")
+_re_enum_check_constraint = re.compile(r"(?:.*?\.)?(.*?)\s+IN\s*\((.+)\)")      
 _re_enum_item = re.compile(r"'(.*?)(?<!\\)'")
 _re_invalid_identifier = re.compile(r"(?u)\W")