working PDF uploads for task orders

2018-08-24 14:40:41 -04:00
parent 54d1e7235b
commit 900bb7b7ad
3 changed files with 12 additions and 3 deletions
--- a/atst/domain/task_orders.py
+++ b/atst/domain/task_orders.py
@@ -3,6 +3,7 @@ from flask import current_app as app

 from atst.database import db
 from atst.models.task_order import TaskOrder, Source
+from atst.models.attachment import Attachment
 from .exceptions import NotFoundError


@@ -53,6 +54,9 @@ class TaskOrders(object):

        except NotFoundError:
            if task_order_data:
+                pdf_file = task_order_data.pop("pdf")
+                # should catch the error here
+                attachment = Attachment.attach(pdf_file)
                return TaskOrders.create(
-                    **task_order_data, number=number, source=Source.MANUAL
+                    **task_order_data, number=number, source=Source.MANUAL, pdf=attachment
                )