evan/ewconfig
1
Fork 0
Code

better error handling

Browse Source
This commit is contained in:
Evan Pratten 2024-02-20 19:16:11 -05:00
parent 599c66f436
commit f703768161
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
scripts/pwnagotchi-watch
View File

@ -40,11 +40,24 @@ def main() -> int:
# Continuously fetch the pwnagotchi screen
while True:
# If the window is closed, exit
for event in pygame.event.get():
if event.type == pygame.QUIT:
return 0
# Attempt to fetch the pwnagotchi screen
try:
# Fetch the pwnagotchi screen
logger.debug(f"Fetching pwnagotchi screen from {args.host}:{args.port}")
response = requests.get(f"http://{args.host}:{args.port}/ui", auth=(args.username, args.password))
response.raise_for_status()
try:
response = requests.get(f"http://{args.host}:{args.port}/ui", auth=(args.username, args.password), timeout=1)
response.raise_for_status()
except Exception as e:
logger.error(f"Failed to connect to pwnagotchi at {args.host}:{args.port}")
logger.error(f"Error: {e}")
time.sleep(args.refresh_rate)
continue
screen_data = response.content
# Draw the screen to the window
@ -64,10 +77,6 @@ def main() -> int:
time.sleep(args.refresh_rate)
# If the window is closed, exit
for event in pygame.event.get():
if event.type == pygame.QUIT:
return 0
return 0