This commit is contained in:
alzyee 2022-01-30 20:20:50 -05:00 committed by Christopher Charbonneau Wells
parent 8c34075696
commit 37b3205cf8
1 changed files with 17 additions and 15 deletions

View File

@ -7,26 +7,27 @@ import plotly.graph_objs as go
from reports import utils from reports import utils
def feeding_amounts(instances): def feeding_amounts(instances):
""" """
Create a graph showing daily feeding amounts over time. Create a graph showing daily feeding amounts over time.
:param instances: a QuerySet of Feeding instances. :param instances: a QuerySet of Feeding instances.
:returns: a tuple of the the graph's html and javascript. :returns: a tuple of the the graph's html and javascript.
""" """
feeding_types=[ feeding_types = [
'breast milk', 'breast milk',
'formula', 'formula',
'fortified breast milk', 'fortified breast milk',
'solid food', 'solid food',
] ]
feeding_types_desc=[ feeding_types_desc = [
'Breast milk', 'Breast milk',
'Formula', 'Formula',
'Fortified breast milk', 'Fortified breast milk',
'Solid food', 'Solid food',
] ]
total_idx=len(feeding_types)+1 #+1 for aggregate total total_idx = len(feeding_types) + 1 # +1 for aggregate total
totals_list=list() totals_list = list()
for i in range(total_idx): for i in range(total_idx):
totals_list.append({}) totals_list.append({})
for instance in instances: for instance in instances:
@ -34,18 +35,18 @@ def feeding_amounts(instances):
date = end.date() date = end.date()
if date not in totals_list[total_idx-1].keys(): if date not in totals_list[total_idx-1].keys():
for item in totals_list: for item in totals_list:
item[date]=0 item[date] = 0
feeding_idx=feeding_types.index(instance.type) feeding_idx = feeding_types.index(instance.type)
totals_list[feeding_idx][date]+= instance.amount or 0 totals_list[feeding_idx][date] += instance.amount or 0
totals_list[total_idx-1][date]+= instance.amount or 0 totals_list[total_idx-1][date] += instance.amount or 0
zeros = [0 for a in totals_list[total_idx-1].values()] zeros = [0 for a in totals_list[total_idx-1].values()]
#sum each feeding type for graph # sum each feeding type for graph
amounts_array=[] amounts_array = []
for i in range(total_idx): for i in range(total_idx):
amounts_array.append([round(amount, 2) for amount in totals_list[i].values()]) amounts_array.append([round(a, 2) for a in totals_list[i].values()])
traces=[] traces = []
for i in range(total_idx-1): for i in range(total_idx-1):
traces.append(go.Bar( traces.append(go.Bar(
name=feeding_types_desc[i], name=feeding_types_desc[i],
@ -64,7 +65,6 @@ def feeding_amounts(instances):
text=amounts_array[total_idx-1] text=amounts_array[total_idx-1]
)) ))
layout_args = utils.default_graph_layout_options() layout_args = utils.default_graph_layout_options()
layout_args['title'] = _('<b>Total Feeding Amounts</b>') layout_args['title'] = _('<b>Total Feeding Amounts</b>')
layout_args['xaxis']['title'] = _('Date') layout_args['xaxis']['title'] = _('Date')
@ -76,5 +76,7 @@ def feeding_amounts(instances):
'layout': go.Layout(**layout_args) 'layout': go.Layout(**layout_args)
}) })
fig.update_layout(barmode='stack') fig.update_layout(barmode='stack')
fig.update_yaxes(automargin=True)
fig.update_xaxes(automargin=True)
output = plotly.plot(fig, output_type='div', include_plotlyjs=False) output = plotly.plot(fig, output_type='div', include_plotlyjs=False)
return utils.split_graph_output(output) return utils.split_graph_output(output)