Fixed broker_connect.py issue

Author: jmashon

broker_connect.py

@@ -9,6 +9,7 @@ def __init__(self):
         self.client = None
 
     def connect(self):
+        print("connect() was called from broker_connect.py")
         """Establish persistent connection with message broker."""
         self.client = mqtt.Client()
         self.client.username_pw_set(

broker_functions.py

@@ -143,9 +143,34 @@ def shutdown(self):
         self.broker_connect.publish(shutdown_message)
         return print("Triggered device shutdown.")
 
-    # calibrate_camera() --> sequence (broker message)
+    def calibrate_camera(self):
+        calibrate_message = {
+            **RPC_REQUEST,
+            "args": {
+                "label": "3fcd2d32-302b-46c9-9448-b4f30c3a9bb1",
+                "priority": 600
+            },
+            "body": {
+                "kind": "execute_script",
+                "args": {
+                    "label": "camera-calibration"
+                },
+                "body": {
+                    "kind": "pair",
+                    "args": {
+                        "label": "CAMERA_CALIBRATION_easy_calibration",
+                        "value": "\"TRUE\""
+                    }
+                }
+            }
+        }
+
+        self.broker_connect.publish(calibrate_message)
+        # return ...
+
     # photo_grid() --> sequence (broker message)
 
+
     def control_servo(self, pin, angle):
         if angle < 0 or angle > 180:
             return print("ERROR: Servo angle constrained to 0-180 degrees.")
@@ -348,9 +373,43 @@ def axis_length(self, axis='all'):
     # mark_as() --> sequence (broker message)
 
     # verify_tool() --> check
+
     # mount_tool() --> check
     # dismount_tool() --> check
 
+    def mount_tool(self, x, y, z):
+        mount_tool_message = {
+            **RPC_REQUEST,
+            "args": {
+                "label": "6a6b10a6-46a8-4b97-8b66-c55021c02bbd",
+                "priority": 600
+            },
+            "body": [
+                {
+                    "kind": "execute",
+                    "args": {
+                        "sequence_id": 24350
+                    },
+                    "body": [
+                        {
+                            "kind": "parameter_application",
+                            "args": {
+                                "label": "Tool",
+                                "data_value": {
+                                    "kind": "coordinate",
+                                    "args": {
+                                        "x": 0,
+                                        "y": 40,
+                                        "z": 0
+                                    }
+                                }
+                            }
+                        }
+                    ]
+                }
+            ]
+        }
+
     # water() --> check
     # dispense() --> check
 

main.py

@@ -17,13 +17,15 @@ def get_token(self, email, password, server="https://my.farm.bot"):
         self.api.api_connect.token = token_data
 
         self.broker.token = token_data
-        self.broker.broker_connect = token_data
+        self.broker.broker_connect.token = token_data
 
         return token_data
 
     def get_info(self, endpoint, id=None):
         return self.api.get_info(endpoint, id)
 
+    def set_info(self, endpoint, field, value, id=None):
+        return self.api.set_info(endpoint, field, value, id)
+
     def connect(self):
         self.broker.broker_connect.connect()
-        return print("Connected to message broker.")