TaskOrders gets EDA client from current Flask app

atst/domain/task_orders.py

@@ -1,4 +1,5 @@
 from sqlalchemy.orm.exc import NoResultFound
+from flask import current_app as app
 
 from atst.database import db
 from atst.models.task_order import TaskOrder
@@ -8,22 +9,22 @@ from .exceptions import NotFoundError
 class TaskOrders(object):
 
     @classmethod
-    def get(cls, order_number, client=None):
+    def get(cls, order_number):
         try:
             task_order = (
                 db.session.query(TaskOrder).filter_by(number=order_number).one()
             )
         except NoResultFound:
-            if client:
-                task_order = TaskOrders._get_from_eda(order_number, client)
+            if TaskOrders._client():
+                task_order = TaskOrders._get_from_eda(order_number)
             else:
                 raise NotFoundError("task_order")
 
         return task_order
 
     @classmethod
-    def _get_from_eda(cls, order_number, client):
-        to_data = client.get_contract(order_number, status="y")
+    def _get_from_eda(cls, order_number):
+        to_data = TaskOrders._client().get_contract(order_number, status="y")
         if to_data:
             return TaskOrders.create(to_data["contract_no"])
         else:
@@ -37,3 +38,7 @@ class TaskOrders(object):
         db.session.commit()
 
         return task_order
+
+    @classmethod
+    def _client(cls):
+        return app.eda_client

atst/forms/financial.py

@@ -119,15 +119,9 @@ class BaseFinancialForm(ValidatedForm):
 
 
 class FinancialForm(BaseFinancialForm):
-    def __init__(self, *args, **kwargs):
-        self.eda_client = kwargs.get("eda_client")
-        if self.eda_client:
-            del(kwargs["eda_client"])
-        super().__init__(*args, **kwargs)
-
     def validate_task_order_number(form, field):
         try:
-            TaskOrders.get(field.data, client=form.eda_client)
+            TaskOrders.get(field.data)
         except NotFoundError:
             raise ValidationError("Task Order number not found")
 

atst/routes/requests/financial_verification.py

@@ -1,6 +1,5 @@
 from flask import render_template, redirect, url_for
 from flask import request as http_request
-from flask import current_app as app
 
 from . import requests_bp
 from atst.domain.requests import Requests
@@ -11,7 +10,7 @@ def financial_form(data):
     if http_request.args.get("extended"):
         return ExtendedFinancialForm(data=data)
     else:
-        return FinancialForm(data=data, eda_client=app.eda_client)
+        return FinancialForm(data=data)
 
 
 @requests_bp.route("/requests/verify/<string:request_id>", methods=["GET"])