From 11f43ccbd23c507a8a0fbcd3e96ead447f4d8ee0 Mon Sep 17 00:00:00 2001
From: Jason Cox <jason@jasonacox.com>
Date: Fri, 18 Oct 2024 21:58:17 -0700
Subject: [PATCH] Add multi-threaded example

---
 examples/README.md    |   6 ++-
 examples/threading.py | 107 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 112 insertions(+), 1 deletion(-)
 create mode 100644 examples/threading.py

diff --git a/examples/README.md b/examples/README.md
index 0a06a0e4..c6d266ad 100644
--- a/examples/README.md
+++ b/examples/README.md
@@ -45,4 +45,8 @@ turn_on('Dining Room')
 
 ## Tuya Cloud API Examples
 
-[cloud.py](cloud.py) -  Example that uses the Tinytuya `Cloud` class and functions to access the Tuya Cloud API to pull device information and control the device via the cloud.
\ No newline at end of file
+[cloud.py](cloud.py) -  Example that uses the Tinytuya `Cloud` class and functions to access the Tuya Cloud API to pull device information and control the device via the cloud.
+
+## Multi-Threaded Example
+
+[threading.py](threading.py) - Example that uses python threading to connect to multiple devices and listen for updates.
diff --git a/examples/threading.py b/examples/threading.py
new file mode 100644
index 00000000..a919d3ee
--- /dev/null
+++ b/examples/threading.py
@@ -0,0 +1,107 @@
+# TinyTuya Example
+# -*- coding: utf-8 -*-
+"""
+ TinyTuya - Multi-threaded Example 
+ 
+ This demonstrates how to use threading to listen for status updates from multiple
+ Tuya devices.
+
+ Setup:
+    Set the config for each device and the script will create a thread for each device
+    to listen for status updates.  The main thread will continue to run and can be used
+    to send commands to the devices.
+
+ Author: Jason A. Cox
+ For more information see https://github.com/jasonacox/tinytuya
+
+"""
+
+import threading
+import time
+import tinytuya
+
+# Define the devices to control
+config = {
+    "TuyaDevices": [
+        {
+            "Address": "192.168.1.10",
+            "Device ID": "00112233445566778899",
+            "Local Key": "1234567890123abc",
+            "Version": "3.3",
+        },
+        {
+            "Address": "192.168.1.11",
+            "Device ID": "10112233445566778899",
+            "Local Key": "1234567890123abc",
+            "Version": "3.3",
+        },
+        {
+            "Address": "192.168.1.12",
+            "Device ID": "20112233445566778899",
+            "Local Key": "1234567890123abc",
+            "Version": "3.3",
+        },
+        {
+            "Address": "192.168.1.13",
+            "Device ID": "30112233445566778899",
+            "Local Key": "1234567890123abc",
+            "Version": "3.3",
+        }
+    ]
+}
+
+# Create array, devices, that is an array of tinytuya.Device objects
+devices = []
+for i in config["TuyaDevices"]:
+    d = tinytuya.Device(i["Device ID"], i["Address"], i["Local Key"], version=i["Version"])
+    devices.append(d)  # Add the device to the devices array
+
+# Function to listen for status updates from each device
+def getDeviceStatuses():
+    global devices
+    global statuses
+
+    def listen_for_status_updates(device, index):
+        # Enable persistent connection to the device
+        device.set_socketPersistent(True)
+        
+        # Call status() once to establish connection and get initial status
+        initial_status = device.status()
+        print(f"INITIAL status from {device.id}: {initial_status}")
+        statuses[index] = {"id": device.id, "status": initial_status["dps"]}
+        
+        # Infinite loop to listen for status updates
+        while True:
+            # Listen for updates from the device (TinyTuya handles receiving updates)
+            updated_status = device.receive()
+            
+            if updated_status:
+                print(f"UPDATE status from {device.id}: {updated_status}")
+                # We may only get one DPS, so just update that one item
+                if "dps" in updated_status:
+                    for key in updated_status["dps"]:
+                        statuses[index]["status"][key] = updated_status["dps"][key]
+                        print(f" - Updated status for {device.id} DPS {key} to {updated_status['dps'][key]}")
+                # This would be a good place to push a socket.io update to a web client
+            
+            # Add a small delay to avoid overwhelming the system
+            time.sleep(0.1)  # 100ms
+
+    threads = []
+    
+    # Create and start a thread for each device
+    for index, device in enumerate(devices):
+        print(f"Starting thread for device {device.id}")
+        thread = threading.Thread(target=listen_for_status_updates, args=(device, index))
+        thread.daemon = True  # Daemon threads exit with the main program
+        threads.append(thread)
+        thread.start()
+
+# Example usage
+statuses = [None] * len(devices)  # Initialize statuses list to hold results for each device
+
+getDeviceStatuses()
+
+# Optionally, keep the main program running indefinitely
+while True:
+    time.sleep(1)  # Keep the main thread alive