~migration fix for using sqlite via alembic

ashishnitinpatil · Oct 17, 2017 · c0adae8 · c0adae8
1 parent dbba43a
commit c0adae8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/item_catalog_app/main.py b/item_catalog_app/main.py
@@ -22,7 +22,7 @@ def initialize_apps(app):
     login_manager.init_app(app)
     login_manager.login_view = 'users.login'
     db.init_app(app)
-    Migrate(app, db)
+    Migrate(app, db, render_as_batch=True)
 
 
 def register_blueprints(app):

diff --git a/item_catalog_app/models.py b/item_catalog_app/models.py
@@ -7,9 +7,17 @@
 from flask_sqlalchemy import SQLAlchemy
 from flask_login import LoginManager
 from sqlalchemy.orm import backref
+from sqlalchemy import MetaData
 
 
-db = SQLAlchemy()
+naming_convention = {
+    "ix": 'ix_%(column_0_label)s',
+    "uq": "uq_%(table_name)s_%(column_0_name)s",
+    "ck": "ck_%(table_name)s_%(column_0_name)s",
+    "fk": "fk_%(table_name)s_%(column_0_name)s_%(referred_table_name)s",
+    "pk": "pk_%(table_name)s",
+}
+db = SQLAlchemy(metadata=MetaData(naming_convention=naming_convention))
 
 
 class User(UserMixin, db.Model):