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

Use date range for mock data instead of current year only

Browse Source
This commit is contained in:
Patrick Smith 2019-01-17 14:44:23 -05:00
parent 84bfdc819c
commit 3cd2644a88
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
atst/domain/csp/reports.py
View File

@ -296,10 +296,13 @@ class MockReportingProvider(ReportingInterface):
else:
budget_months = {}
this_year = pendulum.now().year
end = pendulum.now()
start = end.subtract(months=12)
period = pendulum.period(start, end)
all_months = OrderedDict()
for m in range(1, 13):
month_str = "{month:02d}/{year}".format(month=m, year=this_year)
for t in period.range("months"):
month_str = "{month:02d}/{year}".format(month=t.month, year=t.year)
all_months[month_str] = budget_months.get(month_str, None)
return {"months": all_months}