manage uses sqlalchemy to get the list of tables instead of sql

Robpol86 · Jul 26, 2016 · 9391dc4 · 9391dc4
1 parent 5bdb9fc
commit 9391dc4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/manage.py b/manage.py
@@ -254,10 +254,10 @@ def create_all():
     app = create_app(parse_options())
     log = logging.getLogger(__name__)
     with app.app_context():
-        tables_before = {t[0] for t in db.session.execute('SHOW TABLES')}
+        tables_before = db.engine.table_name()
         db.create_all()
-        tables_after = {t[0] for t in db.session.execute('SHOW TABLES')}
-    created_tables = tables_after - tables_before
+        tables_after = db.engine.table_name()
+    created_tables = set(tables_after) - set(tables_before)
     for table in created_tables:
         log.info('Created table: {}'.format(table))