diff --git a/app/models.py b/app/models.py index 88015191..73a8565b 100644 --- a/app/models.py +++ b/app/models.py @@ -4,6 +4,7 @@ from flask_login import UserMixin, AnonymousUserMixin from itsdangerous import BadSignature, TimedJSONWebSignatureSerializer from time import sleep from werkzeug.security import generate_password_hash, check_password_hash +from werkzeug.utils import secure_filename from . import db, logger, login_manager import os import shutil @@ -318,6 +319,13 @@ class Job(db.Model): """ return '' % self.title + def create_secure_filename(self): + """ + Takes the job.title string nad cratesa a secure filename from this. + """ + filename = secure_filename(self.title) + return filename + def delete(self): """ Delete the job and its inputs and outputs from database and filesystem.