From 54b4eb73ad72bc547d2f4ddf8b8f4ac236fca3b4 Mon Sep 17 00:00:00 2001
From: Ryan Chen <ryanjchen2@gmail.com>
Date: Sat, 19 Aug 2023 15:56:53 -0700
Subject: [PATCH] Added weakness by day and week

---
 main.py | 55 ++++++++++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 48 insertions(+), 7 deletions(-)

diff --git a/main.py b/main.py
index 93d5e41..5a34fde 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,7 @@
 import sqlite3
+from pprint import pprint
 import matplotlib.pyplot as plt
-
+import numpy as np
 from datetime import datetime
 from collections import defaultdict
 
@@ -45,14 +46,54 @@ def weaknessInAnHour(results):
     ax.bar(restaurants.keys(), restaurants.values())
     plt.show()
 
-    
+def weaknessPerMonth(results):
+    weaknesses = [Weakness(row) for row in results]
+    months = defaultdict(lambda: 0)
+    for weakness in weaknesses:
+        months[weakness.time.month] += weakness.total
+
+    fig, ax = plt.subplots()
+    sorted_keys = list(months.keys())
+    sorted_keys.sort()
+    sorted_amounts = [months.get(key) for key in sorted_keys]
+    ax.bar(sorted_keys, sorted_amounts)
+    plt.show()
+
+def weaknessPerDayOverYear(results):
+    weaknesses = [Weakness(row) for row in results]
+    months = defaultdict(lambda: 0)
+
+    week_array = [[0] * 7 for x in range(53)]
+
+    for weakness in weaknesses:
+        week = weakness.time.isocalendar().week - 1
+        day = weakness.time.isocalendar().weekday - 1
+        week_array[week][day] += weakness.total
+
+    week_array = np.rot90(np.array(week_array)).round()
+    fig, ax = plt.subplots()
+    im = ax.imshow(week_array)
+
+    ax.set_yticks(np.arange(7), labels=["Sunday", "Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday"])
+    ax.set_xticks(np.arange(53), labels=[ f"Week {x+1}" for x in range(53)])
+
+    plt.setp(ax.get_xticklabels(), rotation=45, ha="right",
+         rotation_mode="anchor")
+    for i in range(53):
+        for j in range(7):
+            text = ax.text(i, j, week_array[j, i],
+                           ha="center", va="center", color="w", fontsize=5.0)
+
+
+    ax.set_title("akshay weakness in 2022")
+    fig.tight_layout()
+    plt.show()
+
+
+
 
 with sqlite3.connect("doordash.db") as connection:
     c = connection.cursor()
     results = c.execute("select STORE_NAME, DELIVERY_TIME, sum(cast(SUBTOTAL as decimal)) from doordash group by DELIVERY_TIME, STORE_NAME;")
 
-    weaknessInAnHour(results)
-
- 
-
-
+    weaknessPerDayOverYear(results)