Skip to content
Snippets Groups Projects
Commit 22114dd4 authored by Stephan Porada's avatar Stephan Porada :speech_balloon:
Browse files

Add all necessary models to flask shell context

parent 8f4cd9a3
No related branches found
No related tags found
No related merge requests found
import eventlet
eventlet.monkey_patch() # noqa
from app import create_app, db, socketio
from app.models import Corpus, Job, Role, User, NotificationData
from app.models import (Corpus, CorpusFile, Job, JobInput, JobResult,
NotificationData, NotificationEmailData, Role, User)
from flask_migrate import Migrate, upgrade
import os
......@@ -12,11 +13,17 @@ migrate = Migrate(app, db)
@app.shell_context_processor
def make_shell_context():
return {'db': db,
'Corpus': Corpus,
return {'Corpus': Corpus,
'CorpusFile': CorpusFile,
'db': db,
'Job': Job,
'User': User,
'NotificationData': NotificationData}
'JobInput': JobInput,
'JobResult': JobResult,
'NotificationData': NotificationData,
'NotificationEmailData': NotificationEmailData,
'Role': Role,
'User': User
}
@app.cli.command()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment