From f1560bb90ef782de76797bc448028225ad3788c0 Mon Sep 17 00:00:00 2001 From: ksindi Date: Mon, 12 Sep 2016 05:46:40 -0400 Subject: [PATCH] BUG: add support for no classes opt arg --- sqlacodegen/__init__.py | 2 +- sqlacodegen/codegen.py | 6 +++--- sqlacodegen/main.py | 6 +++++- 3 files changed, 9 insertions(+), 5 deletions(-) diff --git a/sqlacodegen/__init__.py b/sqlacodegen/__init__.py index 3ef3af2..e3c5f37 100644 --- a/sqlacodegen/__init__.py +++ b/sqlacodegen/__init__.py @@ -1 +1 @@ -version = __version__ = '1.1.6.0' +version = __version__ = '1.1.6.1' diff --git a/sqlacodegen/codegen.py b/sqlacodegen/codegen.py index cf36410..a10a3dd 100644 --- a/sqlacodegen/codegen.py +++ b/sqlacodegen/codegen.py @@ -527,7 +527,7 @@ class CodeGenerator(object): def __init__(self, metadata, noindexes=False, noconstraints=False, nojoined=False, noinflect=False, nobackrefs=False, - flask=False, ignore_cols=None): + flask=False, ignore_cols=None, noclasses=False): super(CodeGenerator, self).__init__() if noinflect: @@ -600,9 +600,9 @@ def __init__(self, metadata, noindexes=False, noconstraints=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: + if not table.primary_key or table.name in association_tables or noclasses: model = ModelTable(table) - else: + elif not noclasses: model = ModelClass(table, links[table.name], inflect_engine, not nojoined) classes[model.name] = model diff --git a/sqlacodegen/main.py b/sqlacodegen/main.py index 19ea70e..8b6a0f7 100644 --- a/sqlacodegen/main.py +++ b/sqlacodegen/main.py @@ -56,5 +56,9 @@ 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.flask, ignore_cols, args.noclasses) generator.render(outfile) + + +if __name__ == '__main__': + main() \ No newline at end of file