mirror of https://github.com/snachodog/mybuddy.git
fix lint
This commit is contained in:
parent
8c34075696
commit
37b3205cf8
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue