Add script to ATVMonitor

TechG3n · Dec 7, 2022 · 5a1c927 · 5a1c927
1 parent 469de69
commit 5a1c927
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/wh_receiver/sql/tables.sql b/wh_receiver/sql/tables.sql
@@ -78,7 +78,7 @@ CREATE TABLE IF NOT EXISTS `ATVMonitor` (
   `deviceName` varchar(50) COLLATE utf8mb4_unicode_ci NOT NULL,
   `issue` varchar(100) COLLATE utf8mb4_unicode_ci DEFAULT NULL,
   `action` varchar(100) COLLATE utf8mb4_unicode_ci DEFAULT NULL,
-  PRIMARY KEY (`deviceName`)
+  `script` varchar(100) COLLATE utf8mb4_unicode_ci DEFAULT NULL
 ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;
 
 CREATE TABLE IF NOT EXISTS `version` (

diff --git a/wh_receiver/start_whreceiver.py b/wh_receiver/start_whreceiver.py
@@ -91,13 +91,14 @@ def webhook():
         deviceName = validate_string(request.json["deviceName"])
         issue = validate_string(request.json["issue"])
         action = validate_string(request.json["action"])
+        script = validate_string(request.json["script"])
 
         insert_stmt_monitor = (
-            "INSERT INTO ATVMonitor (timestamp, deviceName, issue, action)"
-            "VALUES ( %s, %s, %s, %s )"
+            "INSERT INTO ATVMonitor (timestamp, deviceName, issue, action, script)"
+            "VALUES ( %s, %s, %s, %s, %s )"
         )
 
-        data_monitor = ( str(timestamp), str(deviceName), str(issue), str(action) )
+        data_monitor = ( str(timestamp), str(deviceName), str(issue), str(action), str(script) )
 
         try:
             connection_object = connection_pool.get_connection()