Skip to content

Commit f1560bb

Browse files
committed
BUG: add support for no classes opt arg
1 parent 41ffbbc commit f1560bb

File tree

3 files changed

+9
-5
lines changed

3 files changed

+9
-5
lines changed

sqlacodegen/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
version = __version__ = '1.1.6.0'
1+
version = __version__ = '1.1.6.1'

sqlacodegen/codegen.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -527,7 +527,7 @@ class CodeGenerator(object):
527527

528528
def __init__(self, metadata, noindexes=False, noconstraints=False,
529529
nojoined=False, noinflect=False, nobackrefs=False,
530-
flask=False, ignore_cols=None):
530+
flask=False, ignore_cols=None, noclasses=False):
531531
super(CodeGenerator, self).__init__()
532532

533533
if noinflect:
@@ -600,9 +600,9 @@ def __init__(self, metadata, noindexes=False, noconstraints=False,
600600
continue
601601

602602
# Only form model classes for tables that have a primary key and are not association tables
603-
if not table.primary_key or table.name in association_tables:
603+
if not table.primary_key or table.name in association_tables or noclasses:
604604
model = ModelTable(table)
605-
else:
605+
elif not noclasses:
606606
model = ModelClass(table, links[table.name], inflect_engine, not nojoined)
607607
classes[model.name] = model
608608

sqlacodegen/main.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,5 +56,9 @@ def main():
5656
outfile = codecs.open(args.outfile, 'w', encoding='utf-8') if args.outfile else sys.stdout
5757
generator = CodeGenerator(metadata, args.noindexes, args.noconstraints,
5858
args.nojoined, args.noinflect, args.nobackrefs,
59-
args.flask, ignore_cols)
59+
args.flask, ignore_cols, args.noclasses)
6060
generator.render(outfile)
61+
62+
63+
if __name__ == '__main__':
64+
main()

0 commit comments

Comments
 (0)