From 22114dd464517553d7c6f46983f99c57e0b9af2b Mon Sep 17 00:00:00 2001
From: Stephan Porada <sporada@uni-bielefeld.de>
Date: Mon, 18 May 2020 14:47:04 +0200
Subject: [PATCH] Add all necessary models to flask shell context
---
nopaque.py | 17 ++++++++++++-----
1 file changed, 12 insertions(+), 5 deletions(-)
diff --git a/nopaque.py b/nopaque.py
index 9ae49f8c..b027530c 100644
--- a/nopaque.py
+++ b/nopaque.py
@@ -1,7 +1,8 @@
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()
--
GitLab