Added try/except to prevent exiting of sunset thread

This commit is contained in:
Ashish D'Souza 2023-07-30 02:43:47 -04:00
parent fa157287e1
commit 9d490f8ce4
1 changed files with 29 additions and 21 deletions

View File

@ -3,7 +3,7 @@ import sys
import json import json
from time import sleep from time import sleep
from threading import Thread from threading import Thread
from datetime import datetime, timedelta from datetime import datetime, time, timedelta
import requests import requests
from flask import Blueprint, request, jsonify from flask import Blueprint, request, jsonify
@ -14,13 +14,18 @@ blueprint = Blueprint('detection', __name__)
def get_sunset() -> datetime: def get_sunset() -> datetime:
sunset_date = datetime.now().date()
try:
IPGEOLOCATION_API_KEY = os.environ['IPGEOLOCATION_API_KEY'] IPGEOLOCATION_API_KEY = os.environ['IPGEOLOCATION_API_KEY']
ipgeolocation_api_response = requests.get(f'https://api.ipgeolocation.io/astronomy?apiKey={IPGEOLOCATION_API_KEY}&location=Winter+Haven,+FL') ipgeolocation_api_response = requests.get(f'https://api.ipgeolocation.io/astronomy?apiKey={IPGEOLOCATION_API_KEY}&location=Winter+Haven,+FL')
ipgeolocation_api_response.raise_for_status() ipgeolocation_api_response.raise_for_status()
astronomical_json = json.loads(ipgeolocation_api_response.content) astronomical_json = json.loads(ipgeolocation_api_response.content)
sunset_time = datetime.strptime(astronomical_json['sunset'], '%H:%M').time() sunset_time = datetime.strptime(astronomical_json['sunset'], '%H:%M').time()
sunset_date = datetime.now().date() except Exception as e:
print(e, file=sys.stderr)
sunset_time = time(20, 00, 00)
finally:
if sunset_time < datetime.now().time(): if sunset_time < datetime.now().time():
# Sunset has already passed today # Sunset has already passed today
sunset_date += timedelta(days=1) sunset_date += timedelta(days=1)
@ -29,6 +34,7 @@ def get_sunset() -> datetime:
def reset_all_camera_detection_at_sunset(): def reset_all_camera_detection_at_sunset():
while True: while True:
try:
sunset = get_sunset() sunset = get_sunset()
print(f'Waiting until {sunset} to reset detection for all cameras...', file=sys.stderr) print(f'Waiting until {sunset} to reset detection for all cameras...', file=sys.stderr)
seconds_until_sunset = (sunset - datetime.now()).total_seconds() seconds_until_sunset = (sunset - datetime.now()).total_seconds()
@ -40,6 +46,8 @@ def reset_all_camera_detection_at_sunset():
all_camera_names = json.loads(frigate_api_response.content)['cameras'].keys() all_camera_names = json.loads(frigate_api_response.content)['cameras'].keys()
for camera_name in all_camera_names: for camera_name in all_camera_names:
set_camera_detection(camera_name, True) set_camera_detection(camera_name, True)
except Exception as e:
print(e, file=sys.stderr)
def set_camera_detection(camera: str, value: bool, delay: int = 0) -> None: def set_camera_detection(camera: str, value: bool, delay: int = 0) -> None: