From 37b3205cf8de419a75f5560a7dbaa2396d22d71a Mon Sep 17 00:00:00 2001
From: alzyee <fexfsd@gmail.com>
Date: Sun, 30 Jan 2022 20:20:50 -0500
Subject: [PATCH] fix lint

---
 reports/graphs/feeding_amounts.py | 32 ++++++++++++++++---------------
 1 file changed, 17 insertions(+), 15 deletions(-)

diff --git a/reports/graphs/feeding_amounts.py b/reports/graphs/feeding_amounts.py
index ebc76802..0de7720b 100644
--- a/reports/graphs/feeding_amounts.py
+++ b/reports/graphs/feeding_amounts.py
@@ -7,26 +7,27 @@ 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.
     """
-    feeding_types=[
+    feeding_types = [
             'breast milk',
             'formula',
             'fortified breast milk',
             'solid food',
         ]
-    feeding_types_desc=[
+    feeding_types_desc = [
             'Breast milk',
             'Formula',
             'Fortified breast milk',
             'Solid food',
         ]
-    total_idx=len(feeding_types)+1 #+1 for aggregate total
-    totals_list=list()
+    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:
@@ -34,18 +35,18 @@ def feeding_amounts(instances):
         date = end.date()
         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
+                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()]
-    
-    #sum each feeding type for graph
-    amounts_array=[]
+
+    # 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=[]
+        amounts_array.append([round(a, 2) for a in totals_list[i].values()])
+
+    traces = []
     for i in range(total_idx-1):
         traces.append(go.Bar(
             name=feeding_types_desc[i],
@@ -64,7 +65,6 @@ def feeding_amounts(instances):
         text=amounts_array[total_idx-1]
     ))
 
-
     layout_args = utils.default_graph_layout_options()
     layout_args['title'] = _('<b>Total Feeding Amounts</b>')
     layout_args['xaxis']['title'] = _('Date')
@@ -76,5 +76,7 @@ def feeding_amounts(instances):
         'layout': go.Layout(**layout_args)
     })
     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)
     return utils.split_graph_output(output)