breakout feeding types

This commit is contained in:
alzyee 2022-01-30 18:18:17 -05:00 committed by Christopher Charbonneau Wells
parent acde3156c6
commit 8c34075696
1 changed files with 47 additions and 13 deletions

View File

@ -7,30 +7,63 @@ import plotly.graph_objs as go
from reports import utils
def feeding_amounts(instances):
"""
Create a graph showing daily feeding amounts over time.
:param instances: a QuerySet of Feeding instances.
:returns: a tuple of the the graph's html and javascript.
"""
totals = {}
feeding_types=[
'breast milk',
'formula',
'fortified breast milk',
'solid food',
]
feeding_types_desc=[
'Breast milk',
'Formula',
'Fortified breast milk',
'Solid food',
]
total_idx=len(feeding_types)+1 #+1 for aggregate total
totals_list=list()
for i in range(total_idx):
totals_list.append({})
for instance in instances:
end = timezone.localtime(instance.end)
date = end.date()
if date not in totals.keys():
totals[date] = 0
totals[date] += instance.amount or 0
if date not in totals_list[total_idx-1].keys():
for item in totals_list:
item[date]=0
feeding_idx=feeding_types.index(instance.type)
totals_list[feeding_idx][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()]
amounts = [round(amount, 2) for amount in totals.values()]
trace = go.Bar(
name=_('Total feeding amount'),
x=list(totals.keys()),
y=amounts,
#sum each feeding type for graph
amounts_array=[]
for i in range(total_idx):
amounts_array.append([round(amount, 2) for amount in totals_list[i].values()])
traces=[]
for i in range(total_idx-1):
traces.append(go.Bar(
name=feeding_types_desc[i],
x=list(totals_list[total_idx-1].keys()),
y=amounts_array[i],
text=amounts_array[i]
))
traces.append(go.Bar(
name=_('Total'),
x=list(totals_list[total_idx-1].keys()),
y=zeros,
hoverinfo='text',
textposition='outside',
text=amounts
)
text=amounts_array[total_idx-1]
))
layout_args = utils.default_graph_layout_options()
layout_args['title'] = _('<b>Total Feeding Amounts</b>')
@ -39,8 +72,9 @@ def feeding_amounts(instances):
layout_args['yaxis']['title'] = _('Feeding amount')
fig = go.Figure({
'data': [trace],
'data': traces,
'layout': go.Layout(**layout_args)
})
fig.update_layout(barmode='stack')
output = plotly.plot(fig, output_type='div', include_plotlyjs=False)
return utils.split_graph_output(output)