pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Include all of year's months in dropdown

Browse Source
This commit is contained in:
richard-dds 2018-10-02 13:05:52 -04:00
parent ad49d3f71f
commit 8498b04af8
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
atst/domain/reports.py
View File

@ -1,4 +1,6 @@
from itertools import groupby from itertools import groupby
from collections import OrderedDict
import pendulum
MONTHLY_SPEND_AARDVARK = { MONTHLY_SPEND_AARDVARK = {
@ -234,8 +236,14 @@ class Reports:
@classmethod @classmethod
def cumulative_budget(cls, workspace): def cumulative_budget(cls, workspace):
if workspace.name in REPORT_FIXTURE_MAP: if workspace.name in REPORT_FIXTURE_MAP:
months = REPORT_FIXTURE_MAP[workspace.name]["cumulative"] budget_months = REPORT_FIXTURE_MAP[workspace.name]["cumulative"]
else: else:
months = {} budget_months = {}
return {"months": months} this_year = pendulum.now().year
all_months = OrderedDict()
for m in range(1, 13):
month_str = "{month:02d}/{year}".format(month=m, year=this_year)
all_months[month_str] = budget_months.get(month_str, None)
return {"months": all_months}