From f7dc3929b9d066f047461fb0fa01ce7ceddf97e8 Mon Sep 17 00:00:00 2001
From: Cole Deck <cole@deck.sh>
Date: Tue, 2 May 2023 15:41:05 -0500
Subject: [PATCH] update client, build docker successfully

---
 client.py | 37 +++++++++++++++++++++++++------------
 1 file changed, 25 insertions(+), 12 deletions(-)

diff --git a/client.py b/client.py
index 5d9ac9b..18fd5e9 100644
--- a/client.py
+++ b/client.py
@@ -2,11 +2,15 @@
 
 import logging
 import asyncio
+import numpy as np
+import cv2 as cv
 
 from amqtt.client import MQTTClient
 from amqtt.mqtt.constants import QOS_0, QOS_1, QOS_2
 import time
 import json
+import yaml
+import base64
 
 logger = logging.getLogger(__name__)
 
@@ -19,15 +23,27 @@ def get_uptime():
     print(uptime_seconds)
     return uptime_seconds
 
+def get_frame():
+        cap = cv.VideoCapture(0)
+        if not cap.isOpened():
+                print("Cannot open camera")
+                exit()
+        ret, frame = cap.read()
+        gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)
+        frame = gray
+        if not ret:
+                print("Can't receive frame (stream end?). Exiting ...")
+        print(len(base64.b64encode(frame)))
+        height, width = frame.shape[:2]
+        print(height, width)
+        return base64.b64encode(frame).decode('utf-8')
 async def test_coro():
     C = MQTTClient()
     print("START" + str(get_uptime()))
-    await C.connect('mqtt://127.0.0.1:1883/')
+    await C.connect('mqtt://100.79.112.65:1883')
     tasks = [
-
-        #asyncio.ensure_future(C.publish('sensors/pico1/distance', encoder({"data": 23.0, "sensorID": 0, "timestamp": time.time_ns()}))),
-        #asyncio.ensure_future(C.publish('a/b', b'TEST MESSAGE WITH QOS_1', qos=QOS_1)),
-        asyncio.ensure_future(C.publish('sensors/pico1/uptime', encoder({"data": get_uptime(), "sensorID": 0, "timestamp": time.time_ns()}), qos=QOS_2)),
+        asyncio.ensure_future(C.publish('sensors/pi1/camera', encoder({"data": str(get_frame()), "sensorID": 0, "timestamp": time.time_ns()}), qos=QOS_2)),
+        asyncio.ensure_future(C.publish('sensors/pi1/uptime', encoder({"data": get_uptime(), "sensorID": 0, "timestamp": time.time_ns()}), qos=QOS_2)),
     ]
     await asyncio.wait(tasks)
     logger.info("messages published")
@@ -37,16 +53,13 @@ async def test_coro():
 
 if __name__ == '__main__':
     formatter = "[%(asctime)s] %(name)s {%(filename)s:%(lineno)d} %(levelname)s - %(message)s"
-    logging.basicConfig(level=logging.INFO, format=formatter)
+    logging.basicConfig(level=logging.DEBUG, format=formatter)
     with open('client.yaml', 'r') as fileread:
         clientconfig = yaml.safe_load(fileread)
-    subdict(clientconfig, None)
-    print(sublist)
-    x = 0
-    
-    while x < 5:
+    print(clientconfig)
+
+    while True:
         asyncio.get_event_loop().run_until_complete(test_coro())
-        x = x + 1
     #asyncio.get_event_loop().run_until_complete(test_coro2())