Bad commit sadge

This commit is contained in:
Ryan Chen
2023-08-21 21:07:42 -07:00
parent bfe5274896
commit e6e0b43c4c

18
main.py
View File

@@ -1,4 +1,5 @@
import sqlite3 import sqlite3
import sys
from pprint import pprint from pprint import pprint
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import numpy as np import numpy as np
@@ -72,7 +73,7 @@ def weaknessPerDayOverYear(results):
day = weakness.time.isocalendar().weekday - 1 day = weakness.time.isocalendar().weekday - 1
week_array[week][day] += weakness.total 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() fig, ax = plt.subplots()
im = ax.imshow(week_array, cmap="Reds") im = ax.imshow(week_array, cmap="Reds")
@@ -106,11 +107,24 @@ def weaknessPerDayOverYear(results):
fig.tight_layout() fig.tight_layout()
plt.show() 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: with sqlite3.connect("doordash.db") as connection:
c = connection.cursor() c = connection.cursor()
results = c.execute("select STORE_NAME, DELIVERY_TIME, sum(cast(SUBTOTAL as decimal)) from doordash group by DELIVERY_TIME, STORE_NAME;") results = c.execute("select STORE_NAME, DELIVERY_TIME, sum(cast(SUBTOTAL as decimal)) from doordash group by DELIVERY_TIME, STORE_NAME;")
weaknessPerDayOverYear(results) weaknessPerDayOverYear(results)
"""
if __name__ == "__main__":
if len(sys.argv) < 2:
raise Exception("input csv missing")
importCSV(sys.argv[1])