SpreadSheets commited on
Commit
7e666bd
·
1 Parent(s): 75ff6be

feat: add automatic migration handling in app context

Browse files
Files changed (1) hide show
  1. app/__init__.py +8 -1
app/__init__.py CHANGED
@@ -1,8 +1,10 @@
1
  from flask_dance.contrib.google import make_google_blueprint
2
  from flask_dance.consumer.storage.sqla import SQLAlchemyStorage
 
3
  from flask_sqlalchemy import SQLAlchemy
 
4
  from flask_login import LoginManager
5
- from flask_migrate import Migrate
6
  from flask import Flask
7
 
8
  db = SQLAlchemy()
@@ -42,4 +44,9 @@ def create_app():
42
  )
43
  app.register_blueprint(google_bp, url_prefix="/login")
44
 
 
 
 
 
 
45
  return app
 
1
  from flask_dance.contrib.google import make_google_blueprint
2
  from flask_dance.consumer.storage.sqla import SQLAlchemyStorage
3
+ from flask_migrate import Migrate, upgrade
4
  from flask_sqlalchemy import SQLAlchemy
5
+ from alembic.command import revision
6
  from flask_login import LoginManager
7
+ from alembic.config import Config
8
  from flask import Flask
9
 
10
  db = SQLAlchemy()
 
44
  )
45
  app.register_blueprint(google_bp, url_prefix="/login")
46
 
47
+ with app.app_context():
48
+ config = Config("migrations/alembic.ini")
49
+ revision(config, autogenerate=True, message="Auto migration")
50
+ upgrade()
51
+
52
  return app