diff --git a/app/models.py b/app/models.py index 8801519139c43f5bc7109b5c4f09250d81331bd2..73a8565ba2f7ea68828cddaa835ad2575a182fda 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 '<Job %r>' % 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.