mirror of
https://github.com/michivonah/themepark-alerts.git
synced 2025-12-22 20:36:27 +01:00
101 lines
4.3 KiB
Python
101 lines
4.3 KiB
Python
# Themepark Wait Time Alerts
|
|
# Michi von Ah - October 2023 (Last Updated on October 2024)
|
|
# Thanks to https://www.wartezeiten.app/ for their API
|
|
|
|
import requests
|
|
import os
|
|
from dotenv import load_dotenv
|
|
import time
|
|
|
|
load_dotenv()
|
|
|
|
# Global defintions
|
|
subscribedAttractions = os.getenv('SUBS').split(",")
|
|
currentTimes = {}
|
|
refreshTime = int(os.getenv('CHECK_INTERVAL')) if os.getenv('CHECK_INTERVAL') else 30
|
|
notificationType = os.getenv('NOTIFICATION_TYPE') if os.getenv('NOTIFICATION_TYPE') else "discord"
|
|
themepark = os.getenv('THEMEPARK') if os.getenv('THEMEPARK') else "europapark"
|
|
|
|
# Send messagess via specified notificationType
|
|
def sendMessage(message, notificationType):
|
|
match notificationType.lower():
|
|
case "ntfy":
|
|
endpoint = os.getenv('NTFY_URL')
|
|
accessToken = os.getenv('NTFY_ACCESS_TOKEN')
|
|
if accessToken:
|
|
header = {
|
|
"Authorization":f"Bearer {accessToken}"
|
|
}
|
|
try:
|
|
response = requests.post(url=endpoint, headers=header, data=message)
|
|
except Exception as error:
|
|
raise Exception(f"Got an error while sending the notification: {error}")
|
|
else:
|
|
try:
|
|
response = requests.post(url=endpoint, data=message)
|
|
except Exception as error:
|
|
raise Exception(f"Got an error while sending the notification: {error}")
|
|
return response
|
|
case "discord" | _:
|
|
endpoint = os.getenv('DISCORD_WEBHOOK')
|
|
data = {
|
|
"content": message,
|
|
}
|
|
try:
|
|
response = requests.post(url=endpoint, json=data)
|
|
return response
|
|
except Exception as error:
|
|
raise Exception(f"Got an error while sending the notification: {error}")
|
|
|
|
# Check for the current waiting times
|
|
def checkTimes(subscribedAttractions, themepark):
|
|
try:
|
|
endpoint = "https://api.wartezeiten.app/v1/waitingtimes"
|
|
|
|
header = {
|
|
"language":"de",
|
|
"park":themepark
|
|
}
|
|
|
|
req = requests.get(url=endpoint, headers=header)
|
|
except:
|
|
raise Exception(f"API Request to endpoint {endpoint} failed.")
|
|
try:
|
|
result = req.json()
|
|
except:
|
|
raise Exception("Format of API response is invalid. (JSON expected)")
|
|
try:
|
|
attractions = result
|
|
for attraction in attractions:
|
|
if isinstance(attraction, dict) and "code" in attraction:
|
|
if attraction["code"] in subscribedAttractions:
|
|
if attraction["status"] == "opened":
|
|
refreshTime = 30
|
|
if not attraction["code"] in currentTimes: currentTimes[attraction["code"]] = attraction["waitingtime"];
|
|
if currentTimes[attraction["code"]] > attraction["waitingtime"]:
|
|
sendMessage(f"Waiting time of {attraction['name']} sank to {attraction['waitingtime']} Minutes!", notificationType)
|
|
elif currentTimes[attraction["code"]] < attraction["waitingtime"]:
|
|
sendMessage(f"Waiting time for {attraction['name']} increased to {attraction['waitingtime']} Minutes!", notificationType)
|
|
currentTimes[attraction["code"]] = attraction["waitingtime"]
|
|
else:
|
|
refreshTime = 180
|
|
else:
|
|
print(f"Info: Attraction was skipt because it has an invalid data structure. Affacted attraction: {attraction}")
|
|
except Exception as error:
|
|
raise Exception(f"Got an error while checking for differences since the last API call. Error: {error}")
|
|
|
|
# Main Loop
|
|
# Checks every 30 seconds for changes in the waiting times of the subscribed attractions
|
|
# If some attractions are closed the check will only be executed every 180 seconds
|
|
if __name__ == '__main__':
|
|
print("Themepark Wait Time Alerts")
|
|
print("By Michi von Ah")
|
|
print("Big thanks to the wartezeiten.app API!")
|
|
while True:
|
|
try:
|
|
checkTimes(subscribedAttractions, themepark)
|
|
print(f"Checked for updates at {time.strftime('%H:%M:%S', time.localtime())}")
|
|
except Exception as error:
|
|
raise Exception(error)
|
|
time.sleep(refreshTime)
|
|
|