Work around wrong class names

SamuelMarks · May 27, 2015 · 6efab08 · 6efab08
1 parent 2d4ee3b
commit 6efab08
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/flask_swagger_codegen/templates/models.py.j2 b/flask_swagger_codegen/templates/models.py.j2
@@ -4,7 +4,6 @@
 
 # models are for SQL-alchemy
 
-from marshmallow import fields
 from flask_sqlalchemy import SQLAlchemy
 db = SQLAlchemy()
 
@@ -14,7 +13,7 @@ db = SQLAlchemy()
 
 class {{ name }}(db.Model):
     {%- for n, field in schema.fields.iteritems() %}
-    {{n}} = db.Column(db.{{ field }})
+    {{n}} = db.Column(db.{{ field | replace('fields.', '') }})
     {%- endfor %}
 
 {%- endfor %}