From e6e0b43c4c177bf5267ab7d452aca4d021b1f17a Mon Sep 17 00:00:00 2001 From: Ryan Chen Date: Mon, 21 Aug 2023 21:07:42 -0700 Subject: [PATCH] Bad commit sadge --- main.py | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 6158279..31ab5c8 100644 --- a/main.py +++ b/main.py @@ -1,4 +1,5 @@ import sqlite3 +import sys from pprint import pprint import matplotlib.pyplot as plt import numpy as np @@ -72,7 +73,7 @@ def weaknessPerDayOverYear(results): day = weakness.time.isocalendar().weekday - 1 week_array[week][day] += weakness.total - week_array = np.flipud(np.rot90(np.array(week_array)).round().astype(int)) + week_array = np.rot90(np.array(week_array)).round().astype(int) fig, ax = plt.subplots() im = ax.imshow(week_array, cmap="Reds") @@ -106,11 +107,24 @@ def weaknessPerDayOverYear(results): fig.tight_layout() plt.show() +def importCSV(filename: str): + filepart = filename.split(".")[0] + conn = sqlite3.connect(f"{filepart}.db") + c = conn.cursor() + c.execute(".mode csv") + c.execute(f".import {filename} doordash") + conn.commit() - +""" 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;") weaknessPerDayOverYear(results) + """ + +if __name__ == "__main__": + if len(sys.argv) < 2: + raise Exception("input csv missing") + importCSV(sys.argv[1])