75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
import argparse
|
|
from datetime import date, datetime, timedelta
|
|
import json
|
|
|
|
import requests
|
|
import matplotlib
|
|
matplotlib.use('Agg')
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
|
|
|
|
DAYS_ABBR = ["Mo", "Di", "Mi", "Do", "Fr", "Sa", "So"]
|
|
|
|
|
|
def parse_time(string):
|
|
return datetime.strptime(string, "%Y-%m-%d %H:%M:%S")
|
|
|
|
|
|
def plot(raw, target="wiai.png"):
|
|
data = np.zeros([7, 24])
|
|
last = (parse_time(raw[0]['timestamp']), 0)
|
|
first = last
|
|
increment = timedelta(hours=1)
|
|
for log in raw:
|
|
date = parse_time(log['timestamp'])
|
|
state = log["doorstate"]
|
|
|
|
delta = date - last[0]
|
|
if last[1] and delta >= increment:
|
|
for i in range(1, int(delta / increment)):
|
|
intermediate = last[0] + (increment * i)
|
|
data[intermediate.weekday()][intermediate.hour] += last[1]
|
|
if (not last[0].hour == date.hour) or (date - last[0]) > increment:
|
|
data[date.weekday()][date.hour] += state
|
|
last = (date, state)
|
|
|
|
values = np.unique(data.ravel())
|
|
fig, ax = plt.subplots()
|
|
im = ax.imshow(data)
|
|
ax.set_yticks(np.arange(7))
|
|
ax.set_yticklabels(DAYS_ABBR)
|
|
ax.set_xticks(np.arange(24))
|
|
plt.title("Aggregated opening count FS WIAI")
|
|
plt.figtext(0.5, 0.25, "created: " + str(date.today()), ha="center")
|
|
plt.figtext(0.5, 0.2, str(first[0]) + " → " + str(last[0]), ha="center")
|
|
cax = plt.axes((0.95, 0.15, 0.05, 0.5))
|
|
plt.colorbar(im, cax=cax)
|
|
plt.savefig(target, format="PNG", transparent=True, bbox_inches="tight")
|
|
return target, last[0]
|
|
|
|
|
|
def local(target):
|
|
with open("log") as src:
|
|
raw = json.load(src)
|
|
plot(raw,target=target)
|
|
|
|
|
|
def prod(target):
|
|
return plot(requests.get('https://isfswiaiopen.wiai.de/log').json(), target=target)
|
|
|
|
def get_plot(target):
|
|
return plot(requests.get('https://isfswiaiopen.wiai.de/log').json(), target=target)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description="plot openness of FS WIAI")
|
|
parser.add_argument('-l', '--local', action='store_true')
|
|
parser.add_argument('-o', '--output', default="wiai.png")
|
|
|
|
args = parser.parse_args()
|
|
if args.local:
|
|
local(args.output)
|
|
else:
|
|
prod(args.output)
|