Changes Done:

1. Changed version to 1.1.8
2. Added support for --notables (if we only want classes for all models and not tables)
3. Fixed a bug for ENUM renamed to db.Enum
4. Added check to not add the model imports when --flask is supplied
5. Added requirements.txt for the developers to contribute

Author: Rohit Shende

requirements.txt

@@ -0,0 +1,21 @@
+apipkg==1.5
+atomicwrites==1.4.0
+attrs==19.3.0
+colorama==0.4.3
+execnet==1.7.1
+importlib-metadata==1.6.0
+inflect==4.1.0
+more-itertools==8.2.0
+packaging==20.3
+pep8==1.7.1
+pluggy==0.13.1
+py==1.8.1
+PyMySQL==0.9.3
+pyparsing==2.4.7
+pytest==5.4.2
+pytest-cache==1.0
+pytest-pep8==1.0.6
+six==1.14.0
+SQLAlchemy==1.3.17
+wcwidth==0.1.9
+zipp==3.1.0

setup.py

@@ -42,7 +42,9 @@ def run_tests(self):
         'Programming Language :: Python :: 3.2',
         'Programming Language :: Python :: 3.3',
         'Programming Language :: Python :: 3.4',
-        'Programming Language :: Python :: 3.5'
+        'Programming Language :: Python :: 3.5',
+        'Programming Language :: Python :: 3.6',
+        'Programming Language :: Python :: 3.7'
     ],
     keywords=['sqlalchemy', 'sqlacodegen', 'flask'],
     license='MIT',

sqlacodegen/__init__.py

@@ -1 +1 @@
-version = __version__ = '1.1.7'
+version = __version__ = '1.1.8'

sqlacodegen/codegen.py

@@ -126,7 +126,12 @@ def _render_column_type(coltype):
             else:
                 args.append(repr(value))
 
-    text = _flask_prepend + coltype.__class__.__name__
+    # In case of ENUM from sqlalchemy.dialects, the flask used db.Enum
+    if coltype.__class__.__name__ == "ENUM":
+        text = _flask_prepend + "Enum"
+    else:
+        text = _flask_prepend + coltype.__class__.__name__
+
     if args:
         text += '({0})'.format(', '.join(args))
 
@@ -529,7 +534,7 @@ class CodeGenerator(object):
 
     def __init__(self, metadata, noindexes=False, noconstraints=False,
                  nojoined=False, noinflect=False, nobackrefs=False,
-                 flask=False, ignore_cols=None, noclasses=False, nocomments=False):
+                 flask=False, ignore_cols=None, noclasses=False, nocomments=False, notables=False):
         super(CodeGenerator, self).__init__()
 
         if noinflect:
@@ -603,15 +608,22 @@ def __init__(self, metadata, noindexes=False, noconstraints=False,
                                     table.c[colname].type = Enum(*options, native_enum=False)
                                 continue
 
-            # Only form model classes for tables that have a primary key and are not association tables
-            if not table.primary_key or table.name in association_tables or noclasses:
+            # Only generate classes when notables is set to True
+            if notables:
+                model = ModelClass(table, links[table.name], inflect_engine, not nojoined)
+                classes[model.name] = model
+            elif not table.primary_key or table.name in association_tables or noclasses:
+                # Only form model classes for tables that have a primary key and are not association tables
                 model = ModelTable(table)
             elif not noclasses:
                 model = ModelClass(table, links[table.name], inflect_engine, not nojoined)
                 classes[model.name] = model
 
             self.models.append(model)
-            model.add_imports(self.collector)
+
+            # collect imports for models only if flask is not specified
+            if not self.flask:
+                model.add_imports(self.collector)
 
         # Nest inherited classes in their superclasses to ensure proper ordering
         for model in classes.values():

sqlacodegen/main.py

@@ -33,6 +33,7 @@ def main():
     parser.add_argument('--nojoined', action='store_true', help="don't autodetect joined table inheritance")
     parser.add_argument('--noinflect', action='store_true', help="don't try to convert tables names to singular form")
     parser.add_argument('--noclasses', action='store_true', help="don't generate classes, only tables")
+    parser.add_argument('--notables', action='store_true', help="don't generate tables, only classes")
     parser.add_argument('--outfile', help='file to write output to (default: stdout)')
     parser.add_argument('--nobackrefs', action='store_true', help="don't include backrefs")
     parser.add_argument('--flask', action='store_true', help="use Flask-SQLAlchemy columns")
@@ -57,7 +58,7 @@ def main():
     outfile = codecs.open(args.outfile, 'w', encoding='utf-8') if args.outfile else sys.stdout
     generator = CodeGenerator(metadata, args.noindexes, args.noconstraints,
                               args.nojoined, args.noinflect, args.nobackrefs,
-                              args.flask, ignore_cols, args.noclasses, args.nocomments)
+                              args.flask, ignore_cols, args.noclasses, args.nocomments, args.notables)
     generator.render(outfile)