Add __tablename__s and primary keys to models

SamuelMarks · May 27, 2015 · 430ac68 · 430ac68
1 parent 6efab08
commit 430ac68
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flask_swagger_codegen/templates/models.py.j2 b/flask_swagger_codegen/templates/models.py.j2
@@ -12,6 +12,8 @@ db = SQLAlchemy()
 
 
 class {{ name }}(db.Model):
+    __tablename__ = '{{ name | lower }}s'
+    id = db.Column(db.Integer(), primary_key=True)
     {%- for n, field in schema.fields.iteritems() %}
     {{n}} = db.Column(db.{{ field | replace('fields.', '') }})
     {%- endfor %}