forked from wiai/doorbot
79 lines
2.1 KiB
Python
79 lines
2.1 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 matplotlib.patches as mpatches
|
|
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 delta.seconds >= 3600:
|
|
for i in range(1, int(delta / increment)):
|
|
intermediate = last[0] + (increment * i)
|
|
data[intermediate.weekday()][intermediate.hour] += 1
|
|
|
|
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))
|
|
|
|
colors = [ im.cmap(im.norm(value)) for value in values ]
|
|
patches = [ mpatches.Patch(color=colors[i], label=str(values[i])) for i in range(len(values))]
|
|
plt.legend(handles=patches, bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0.)
|
|
plt.title("Aggregated opening hours 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")
|
|
plt.savefig(target, format="PNG", transparent=True, bbox_inches="tight")
|
|
|
|
return target
|
|
|
|
|
|
def local():
|
|
with open("log") as src:
|
|
raw = json.load(src)
|
|
plot(raw)
|
|
|
|
|
|
def prod():
|
|
plot(requests.get('https://isfswiaiopen.wiai.de/log').json())
|
|
|
|
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)
|