diff --git a/atst/models/request.py b/atst/models/request.py index d7d1f96c..9926aa12 100644 --- a/atst/models/request.py +++ b/atst/models/request.py @@ -216,3 +216,14 @@ class Request(Base, mixins.TimestampsMixin): @property def displayname(self): return self.latest_revision.name or self.id + + @property + def contracting_officer_full_name(self): + if self.latest_revision.fname_co: + return "{} {}".format( + self.latest_revision.fname_co, self.latest_revision.lname_co + ) + + @property + def contracting_officer_email(self): + return self.latest_revision.email_co diff --git a/atst/routes/workspaces.py b/atst/routes/workspaces.py index 9879d90c..b1fed5f5 100644 --- a/atst/routes/workspaces.py +++ b/atst/routes/workspaces.py @@ -105,16 +105,20 @@ def workspace_reports(workspace_id): prev_month = current_month - timedelta(days=28) two_months_ago = prev_month - timedelta(days=28) - # lets just say it expires on Christmas... ho ho ho - expiration_date = date(2018, 12, 25) - remaining_difference = expiration_date - today - remaining_days = remaining_difference.days + expiration_date = workspace.request.task_order.expiration_date + if expiration_date: + remaining_difference = expiration_date - today + remaining_days = remaining_difference.days + else: + remaining_days = 0 return render_template( "workspaces/reports/index.html", cumulative_budget=Reports.cumulative_budget(workspace), workspace_totals=Reports.workspace_totals(workspace), monthly_totals=Reports.monthly_totals(workspace), + jedi_request=workspace.request, + task_order=workspace.request.task_order, current_month=current_month, prev_month=prev_month, two_months_ago=two_months_ago, diff --git a/script/seed_sample.py b/script/seed_sample.py index ee113afd..237b10a3 100644 --- a/script/seed_sample.py +++ b/script/seed_sample.py @@ -21,24 +21,25 @@ WORKSPACE_USERS = [ "last_name": "Knight", "email": "knight@mil.gov", "workspace_role": "developer", - "dod_id": "0000000001" + "dod_id": "0000000001", }, { "first_name": "Mario", "last_name": "Hudson", "email": "hudson@mil.gov", "workspace_role": "ccpo", - "dod_id": "0000000002" + "dod_id": "0000000002", }, { "first_name": "Louise", "last_name": "Greer", "email": "greer@mil.gov", "workspace_role": "admin", - "dod_id": "0000000003" + "dod_id": "0000000003", }, ] + def seed_db(): users = [] for dev_user in DEV_USERS.values(): @@ -65,8 +66,13 @@ def seed_db(): request = requests[0] request.task_order = TaskOrderFactory.build() + request = Requests.update( + request.id, {"financial_verification": RequestFactory.mock_financial_data()} + ) - workspace = Workspaces.create(request, name="{}'s workspace".format(user.first_name)) + workspace = Workspaces.create( + request, name="{}'s workspace".format(user.first_name) + ) for workspace_user in WORKSPACE_USERS: Workspaces.create_member(user, workspace, workspace_user) @@ -75,7 +81,7 @@ def seed_db(): workspace=workspace, name="First Project", description="This is our first project.", - environment_names=["dev", "staging", "prod"] + environment_names=["dev", "staging", "prod"], ) diff --git a/templates/workspaces/reports/index.html b/templates/workspaces/reports/index.html index 6da7f860..82bc92cf 100644 --- a/templates/workspaces/reports/index.html +++ b/templates/workspaces/reports/index.html @@ -56,7 +56,7 @@