Bad commit sadge
This commit is contained in:
18
main.py
18
main.py
@@ -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])
|
||||||
|
|||||||
Reference in New Issue
Block a user