From dbe2e236a99408a4441ba81eb3768902c4421739 Mon Sep 17 00:00:00 2001
From: Meng Li <34143965+Meng6@users.noreply.github.com>
Date: Tue, 3 Dec 2019 10:46:03 -0500
Subject: [PATCH] Fixed issue #16

---
 src/visualization/battery_consumption_rates_barchart.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/visualization/battery_consumption_rates_barchart.py b/src/visualization/battery_consumption_rates_barchart.py
index c2101f69..8a19eaf9 100644
--- a/src/visualization/battery_consumption_rates_barchart.py
+++ b/src/visualization/battery_consumption_rates_barchart.py
@@ -6,21 +6,23 @@ import plotly.graph_objects as go
 def getBatteryConsumptionRatesBarChart(battery_data, pid):
     plot = go.Figure(go.Bar(
                     x=battery_data["battery_daily_avgconsumptionrate"],
-                    y=battery_data["local_date"].apply(lambda x: x.replace("-","/")).tolist(),
+                    y=battery_data["local_date"].apply(lambda x: x.strftime("%Y/%m/%d")).tolist(),
                     orientation='h'))
     plot.update_layout(title="Daily battery consumption rates bar chart for " + pid,
                     xaxis_title="battery drains % per hour",
                     )
     return plot
-    
 
 
-battery_data = pd.read_csv(snakemake.input[0])
+
+battery_data = pd.read_csv(snakemake.input[0], parse_dates=["local_date"])
 pid = snakemake.params["pid"]
 if battery_data.empty:
     empty_html = open(snakemake.output[0], "w")
     empty_html.write("There is no battery data for " + pid)
     empty_html.close()
 else:
+    battery_data.set_index(["local_date"], inplace=True)
+    battery_data = battery_data.resample("1D").asfreq().fillna(0).reset_index()
     plot = getBatteryConsumptionRatesBarChart(battery_data, pid)
     pio.write_html(plot, file=snakemake.output[0], auto_open=False)
\ No newline at end of file