From a8b40813821a7f53f873548501b16df9125f561e Mon Sep 17 00:00:00 2001
From: Jim Bennett <jim.bennett@microsoft.com>
Date: Thu, 12 Mar 2020 19:14:15 +0000
Subject: [PATCH] Working code

---
 .gitignore       | 105 ++++++++++++++++++++++++++++++++++++++++++++
 app.py           | 112 +++++++++++++++++++++++++++++++++++++++++++++++
 requirements.txt |   3 ++
 3 files changed, 220 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 app.py
 create mode 100644 requirements.txt

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..55ede8c
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,105 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.DS_Store
\ No newline at end of file
diff --git a/app.py b/app.py
new file mode 100644
index 0000000..7268575
--- /dev/null
+++ b/app.py
@@ -0,0 +1,112 @@
+import unicornhat as uh
+import os, asyncio, threading, random, time
+from dotenv import load_dotenv
+from azure.iot.device.aio import IoTHubDeviceClient, ProvisioningDeviceClient
+from azure.iot.device import MethodResponse
+
+load_dotenv()
+
+id_scope = os.getenv('ID_SCOPE')
+device_id = os.getenv('DEVICE_ID')
+primary_key = os.getenv('PRIMARY_KEY')
+
+uh.set_layout(uh.PHAT)
+uh.brightness(1)
+
+def set_colour(colour):
+    r = '0x' + colour[0:2]
+    g = '0x' + colour[2:4]
+    b = '0x' + colour[4:6]
+
+    r_value = int(r, 0)
+    g_value = int(g, 0)
+    b_value = int(b, 0)
+
+    print('Updating color: r =', r_value, ', g =', g_value, ', b =', b_value)
+
+    uh.set_all(r_value, g_value, b_value)
+    uh.show()
+
+async def main():
+    # provision the device
+    async def register_device():
+        provisioning_device_client = ProvisioningDeviceClient.create_from_symmetric_key(
+            provisioning_host='global.azure-devices-provisioning.net',
+            registration_id=device_id,
+            id_scope=id_scope,
+            symmetric_key=primary_key,
+        )
+
+        return await provisioning_device_client.register()
+
+    results = await asyncio.gather(register_device())
+    registration_result = results[0]
+
+    # build the connection string
+    conn_str='HostName=' + registration_result.registration_state.assigned_hub + \
+                ';DeviceId=' + device_id + \
+                ';SharedAccessKey=' + primary_key
+
+    # The client object is used to interact with your Azure IoT Central.
+    device_client = IoTHubDeviceClient.create_from_connection_string(conn_str)
+
+    # connect the client.
+    print('Connecting')
+    await device_client.connect()
+    print('Connected')
+
+    # Get the current colour
+    twin = await device_client.get_twin()
+    print('Got twin: ', twin)
+
+    try:
+        desired_colour = twin['reported']['colour']
+        set_colour(desired_colour)
+    except Error:
+        print("Couldn't load twin,")
+
+    # listen for commands
+    async def command_listener(device_client):
+        while True:
+            method_request = await device_client.receive_method_request()
+
+            print('Call made to', method_request.name)
+
+            colour = method_request.payload
+            
+            print('payload', colour)
+
+            if isinstance(colour, dict):
+                colour = colour['colour']
+
+            print('payload', colour)
+
+            set_colour(colour)
+
+            payload = {'result': True, 'data': colour}
+            method_response = MethodResponse.create_from_method_request(
+                method_request, 200, payload
+            )
+            await device_client.send_method_response(method_response)
+
+            # Write the colour back as a property
+            await device_client.patch_twin_reported_properties({'colour':colour})
+
+    # async loop that controls the lights
+    async def main_loop():
+        global mode
+        while True:
+            await asyncio.sleep(1)
+
+    listeners = asyncio.gather(command_listener(device_client))
+
+    await main_loop()
+
+    # Cancel listening
+    listeners.cancel()
+
+    # Finally, disconnect
+    await device_client.disconnect()
+
+if __name__ == '__main__':
+    asyncio.run(main())
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..b76e9a2
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,3 @@
+unicornhat
+azure-iot-device
+python-dotenv
\ No newline at end of file