From 28c04cad8bf33dc37d6a9a7b620f54429c105ff6 Mon Sep 17 00:00:00 2001
From: mathieu <math.gaudron@gmail.com>
Date: Sun, 27 Feb 2022 17:18:09 +0100
Subject: [PATCH] Default behavior : display stats for the last 30 days only

---
 generate_html.py | 27 +++++++++++++++++++++------
 generate_html.sh |  2 +-
 2 files changed, 22 insertions(+), 7 deletions(-)

diff --git a/generate_html.py b/generate_html.py
index fab3f86..cb83830 100644
--- a/generate_html.py
+++ b/generate_html.py
@@ -5,6 +5,7 @@ import logging
 import os
 import sqlite3
 from datetime import datetime
+from datetime import timedelta
 from pathlib import Path
 
 import pandas as pd
@@ -36,16 +37,23 @@ def parse_args():
         default=HERE / "stats.html",
     )
     parser.add_argument("--debug", "-d", action="store_true", help="Run in debug mode")
+    parser.add_argument("-n", "--n-days-histo", dest="n_days_histo",
+                    type=int,
+                    default=30,
+                    help="Number of day to show")
     args = parser.parse_args()
-    return args.sqlite, args.html, args.debug
+    return args.sqlite, args.html, args.debug, args.n_days_histo
 
 
-def generate_html_report(sqlite_fn: Path, html_fn: Path):
+def generate_html_report(sqlite_fn: Path, html_fn: Path, n_days_histo):
     # Create your connection.
     log.info(f"Read sqlite {sqlite_fn}")
     cnx = sqlite3.connect(sqlite_fn)
 
-    df = pd.read_sql_query(f"SELECT * FROM {TABLE_NAME}", cnx)
+    first_date = datetime.today() - timedelta(days = n_days_histo)
+    log.debug(f"First date : {first_date}")
+
+    df = pd.read_sql_query(f"SELECT * FROM {TABLE_NAME} WHERE date > '{first_date}'", cnx)
     df = df.set_index("date")
     log.info(f"Find {len(df)} items")
 
@@ -58,6 +66,12 @@ def generate_html_report(sqlite_fn: Path, html_fn: Path):
         # dfi["granu"] = i
         dfs = dfs.append(dfi)
 
+    #breakpoint()
+
+    # Remove empty conlumn
+    dfs = dfs.dropna(axis=1, how='all')
+    #breakpoint()
+
     log.info(f"Get {len(dfs)} items after sub-sampling")
     fig = px.area(dfs)
     fig.for_each_trace(lambda trace: trace.update(fillcolor=trace.line.color))
@@ -68,19 +82,20 @@ def generate_html_report(sqlite_fn: Path, html_fn: Path):
         title="Utilisation de RAM pour les docker-compose de l'eunuque (pour voir une stat, mettre la sourie sur "
         f"le haut des courbes, vers les points), généré le {now.strftime('%d/%m/%Y %H:%M:%S')}",
         xaxis_title="Temps",
-        yaxis_title="RAM",
+        yaxis_title="RAM (MiB)",
+        legend={"traceorder":"reversed"}
     )
     fig.write_html(html_fn, include_plotlyjs="cdn")
     log.info(f"Report generate in {html_fn}")
 
 
 def main():
-    sqlite_fn, html_fn, debug = parse_args()
+    sqlite_fn, html_fn, debug, n_days_histo = parse_args()
     if debug:
         logging.basicConfig(level=logging.DEBUG)
     else:
         logging.basicConfig(level=logging.INFO)
-    generate_html_report(sqlite_fn, html_fn)
+    generate_html_report(sqlite_fn, html_fn, n_days_histo)
 
 
 if __name__ == "__main__":
diff --git a/generate_html.sh b/generate_html.sh
index 9579aa4..04e10cb 100755
--- a/generate_html.sh
+++ b/generate_html.sh
@@ -7,7 +7,7 @@ SCRIPT_PATH="$( cd "$(dirname "$0")" >/dev/null 2>&1 ; pwd -P )"
 pushd "${SCRIPT_PATH}" > /dev/null
 
 
-venv/bin/python generate_html.py stats.sqlite /home/caracals/services/org-caracals-files/files/stats.html
+venv/bin/python generate_html.py stats.sqlite /home/caracals/services/org.caracals.files/files/stats.html
 
 
 # Back to  original path