Preemptively upload task order PDF

2018-10-19 10:57:18 -04:00
parent 3cf0907d0d
commit 5d2b976e5f
8 changed files with 217 additions and 32 deletions
--- a/atst/domain/task_orders.py
+++ b/atst/domain/task_orders.py
@@ -67,3 +67,13 @@ class TaskOrders(object):
                    source=Source.MANUAL,
                    pdf=attachment,
                )
+
+    @classmethod
+    def get_or_create(cls, number, attachment=None, data=None):
+        try:
+            return TaskOrders.get(number)
+        except NotFoundError:
+            data = data or {}
+            return TaskOrders.create(
+                **data, number=number, pdf=attachment, source=Source.MANUAL
+            )