95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
import argparse
|
|
import datetime
|
|
import json
|
|
import time
|
|
from tempfile import NamedTemporaryFile
|
|
import logging
|
|
|
|
import requests
|
|
import schedule
|
|
|
|
logging.basicConfig(format='%(asctime)s %(levelname)s %(name)s:%(message)s', level=logging.DEBUG, datefmt="%Y.%m.%d %H:%M:%S")
|
|
log = logging.getLogger(__name__)
|
|
|
|
def parse_time(string):
|
|
return datetime.datetime.strptime(string, "%Y-%m-%d %H:%M:%S")
|
|
|
|
def get_status():
|
|
status = requests.get("https://isfswiaiopen.wiai.de?json").json()
|
|
status["timestamp"] = parse_time(status['timestamp'])
|
|
return status
|
|
|
|
def get_status_text(config, src=get_status):
|
|
return config["texts"][str(get_status()["doorstate"])]
|
|
|
|
def post(chats, text, token):
|
|
url = "https://api.telegram.org/bot{token}/sendMessage".format(token=token)
|
|
for chat in chats:
|
|
response = requests.post(url, data={'chat_id': chats[chat], "text": text})
|
|
log.info("post message: %s", response.status_code)
|
|
|
|
def main(args={"config": "settings.json"}):
|
|
log.info("run once")
|
|
config = json.load(open(args['config']))
|
|
text = get_status_text(config)
|
|
post(config['groups'], text, config['token'])
|
|
post_plot(config)
|
|
|
|
def loop(args={"config": "settings.json"}):
|
|
log.info("prepare loop")
|
|
config = json.load(open(args['config']))
|
|
setup(config)
|
|
while True:
|
|
try:
|
|
do_loop(config)
|
|
except Exception as e:
|
|
log.exception(e)
|
|
time.sleep(config['sleep'])
|
|
|
|
def do_loop(config):
|
|
last_state = None
|
|
while True:
|
|
log.info("enter loop")
|
|
changed = False
|
|
new_state = get_status()
|
|
if last_state is None:
|
|
changed = True
|
|
elif not last_state["doorstate"] == new_state["doorstate"]:
|
|
changed = True
|
|
if changed:
|
|
last_state = new_state
|
|
text = get_status_text(config, lambda: last_state)
|
|
post(config["groups"], text, config["token"])
|
|
log.info("run pending tasks")
|
|
schedule.run_pending()
|
|
log.info("sleep")
|
|
time.sleep(config['sleep'])
|
|
|
|
def post_plot(config):
|
|
from plot import get_plot
|
|
with NamedTemporaryFile() as target:
|
|
image_url = 'https://api.telegram.org/bot{token}/sendPhoto'.format(token=config['token'])
|
|
photo, last = get_plot(target)
|
|
files = {'photo': photo}
|
|
if last + datetime.timedelta(days=1) < datetime.datetime.today():
|
|
log.info("skipping image, no new updates...")
|
|
return
|
|
for chat in config['groups']:
|
|
files['photo'].seek(0)
|
|
values = {"chat_id":config['groups'][chat]}
|
|
r = requests.post(image_url, files=files, data=values)
|
|
log.info("post photo: %s", r.status_code)
|
|
|
|
def setup(config):
|
|
schedule.every(config['photo_interval']).seconds.do(lambda: post_plot(config))
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="DoorStateBot")
|
|
parser.add_argument("--config", "-c", default="settings.json", help="Configuration file")
|
|
parser.add_argument("--loop", "-l", action="store_true", help="Loop")
|
|
args = parser.parse_args()
|
|
if args.loop:
|
|
loop(vars(args))
|
|
else:
|
|
main(vars(args))
|