From 35ef339f8f17ca6d8e7343c3b3ccbad58feb5fcc Mon Sep 17 00:00:00 2001 From: richard-dds Date: Thu, 8 Nov 2018 14:44:00 -0500 Subject: [PATCH] Check for attr rather than classname --- atst/filters.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/atst/filters.py b/atst/filters.py index 426b9651..c9aa682f 100644 --- a/atst/filters.py +++ b/atst/filters.py @@ -1,8 +1,6 @@ import re import datetime from flask import current_app as app -from werkzeug.datastructures import FileStorage -from atst.models.attachment import Attachment def iconSvg(name): @@ -48,10 +46,7 @@ def mixedContentToJson(value): if ( isinstance(value, dict) and "task_order" in value - and ( - isinstance(value["task_order"]["pdf"], FileStorage) - or isinstance(value["task_order"]["pdf"], Attachment) - ) + and hasattr(value["task_order"]["pdf"], "filename") ): value["task_order"]["pdf"] = value["task_order"]["pdf"].filename return app.jinja_env.filters["tojson"](value)