t510599/cloudprog-hw4
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: runtime errors

Browse Source
This commit is contained in:
Tony Yang 2023-05-01 15:10:04 +08:00
parent d5c7d62507
commit c37017d291
Signed by: t510599
GPG Key ID: D88388851C28715D
1 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
led_control.py
View File

@ -1,7 +1,9 @@
import sys
import json import json
import uuid import uuid
import asyncio import asyncio
import datetime import datetime
import traceback
import cv2 import cv2
import requests import requests
@ -76,7 +78,7 @@ def on_message_received(topic, payload, dup, qos, retain, **kwargs):
def parse_led_payload(payload): def parse_led_payload(payload):
data = json.loads(payload) data = json.loads(payload.decode())
if data.get("faces"): if data.get("faces"):
led_control(True) led_control(True)
else: else:
@ -91,7 +93,7 @@ def led_control(light):
def parse_s3_url_payload(payload): def parse_s3_url_payload(payload):
data = json.dumps(payload) data = json.loads(payload.decode())
put_presigned_s3_url(data["url"], take_picture()) put_presigned_s3_url(data["url"], take_picture())
@ -136,6 +138,7 @@ async def main():
) )
subscribe_result = subscribe_future.result() subscribe_result = subscribe_future.result()
print(subscribe_result)
print(f"Subscribed to {LED_TOPIC}") print(f"Subscribed to {LED_TOPIC}")
subscribe_future, packet_id = mqtt_connection.subscribe( subscribe_future, packet_id = mqtt_connection.subscribe(
@ -145,7 +148,8 @@ async def main():
) )
subscribe_result = subscribe_future.result() subscribe_result = subscribe_future.result()
print(f"Subscribed to {LED_TOPIC}") print(subscribe_result)
print(f"Subscribed to {S3_URL_TOPIC}")
while True: while True:
command = input("> ") command = input("> ")
@ -154,17 +158,28 @@ async def main():
filename = datetime.datetime.strftime(datetime.datetime.now(), "%Y-%m-%d-%H-%M-%S") + "." + FORMAT filename = datetime.datetime.strftime(datetime.datetime.now(), "%Y-%m-%d-%H-%M-%S") + "." + FORMAT
publish_future, packet_id = mqtt_connection.publish( publish_future, packet_id = mqtt_connection.publish(
topic=S3_UPLOAD_TOPIC, topic=S3_UPLOAD_TOPIC,
qos=1, payload=json.dumps({ "filename": filename }),
payload=json.dumps({ "filename": filename }) qos=mqtt.QoS.AT_LEAST_ONCE
) )
publish_result = publish_future.result() publish_result = publish_future.result()
print(publish_result)
print(f"Upload request {filename} sent!") print(f"Upload request {filename} sent!")
elif command == "q":
break
def handler(loop, context):
e = context['exception']
print(*traceback.format_exception(None, e, e.__traceback__), file=sys.stderr, flush=True)
if not DEBUG:
GPIO.cleanup()
loop.stop()
# run # run
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
loop.set_exception_handler(lambda loop, context: [print(context['message']), GPIO.cleanup()]) loop.set_exception_handler(handler)
loop.create_task(main()) loop.create_task(main())
loop.run_forever() loop.run_forever()