From 2c89a044d49035371a1b92787628d83bde73c549 Mon Sep 17 00:00:00 2001
From: John Lancaster <32917998+jsl12@users.noreply.github.com>
Date: Fri, 4 Oct 2024 13:14:54 +0000
Subject: [PATCH] started traffic notification app

---
 apps/traffic/traffic.py   | 64 +++++++++++++++++++++++++++++++++++++++
 apps/traffic/traffic.yaml |  3 ++
 2 files changed, 67 insertions(+)
 create mode 100644 apps/traffic/traffic.py
 create mode 100644 apps/traffic/traffic.yaml

diff --git a/apps/traffic/traffic.py b/apps/traffic/traffic.py
new file mode 100644
index 0000000..e22bb78
--- /dev/null
+++ b/apps/traffic/traffic.py
@@ -0,0 +1,64 @@
+from datetime import timedelta
+
+from appdaemon import ADAPI
+from appdaemon.adbase import ADBase
+from appdaemon.entity import Entity
+from appdaemon.models.notification import AndroidData
+
+
+class TrafficAlert(ADAPI):
+    notified: bool = False
+
+    def initialize(self):
+        self.log(f'Traffic time: {self.traffic_time}')
+        self.traffic_entity.listen_state(
+            self.handle_state_change,
+            attribute='duration',
+            # constrain_state=lambda d: d > 30.0
+        )
+        self.handle_state_change(new=self.traffic_time.total_seconds() / 60)
+
+    @property
+    def traffic_entity(self) -> Entity:
+        return self.get_entity('sensor.work_to_home')
+
+    @property
+    def traffic_time(self) -> timedelta:
+        return timedelta(minutes=float(self.traffic_entity.get_state('duration')))
+
+    def notify_android(self, device: str, **data):
+        model = AndroidData.model_validate(data)
+        res = self.call_service(
+            f'notify/mobile_app_{device}', **model.model_dump())
+        return res
+
+    def handle_state_change(self,
+                            entity: str = None,
+                            attribute: str = None,
+                            old: str = None,
+                            new: str = None,
+                            **kwargs: dict):
+        self.log(f'Travel time changed: {new}')
+
+        if not self.notified:
+            actions = [
+                {
+                    'action': 'URI',
+                    'title': 'See travel times',
+                    'uri': "https://grafana.john-stream.com/d/f4ff212e-c786-40eb-b615-205121f482e3/travel-time-details?orgId=1&from=1727927004390&to=1728013404390"
+                }
+            ]
+
+            if new > 40.0: 
+                self.notify_android('pixel_5', **{
+                    'title': 'Heavy Traffic',
+                    'message': 'Get moving!',
+                    'data': {'tag': 'traffic', 'color': 'red', 'actions': actions},
+                })
+            elif new > 30.0:
+                self.notify_android('pixel_5', **{
+                    'title': 'Increaing Traffic',
+                    'message': 'Something needs to happen',
+                    'data': {'tag': 'traffic', 'color': 'yellow', 'actions': actions},
+                })
+            self.notified = True
diff --git a/apps/traffic/traffic.yaml b/apps/traffic/traffic.yaml
new file mode 100644
index 0000000..d2b3e07
--- /dev/null
+++ b/apps/traffic/traffic.yaml
@@ -0,0 +1,3 @@
+traffic:
+  module: traffic
+  class: TrafficAlert
\ No newline at end of file