From 6bedd5d73c548a0e67971e75f6f4d0ebce9dab4b Mon Sep 17 00:00:00 2001
From: "13621160019@163.com" <13621160019@163.com>
Date: Fri, 30 Apr 2021 10:46:00 +0800
Subject: [PATCH] added gateway process to myems-bacnet service

---
 myems-bacnet/README.md  | 10 ++++-
 myems-bacnet/gateway.py | 88 +++++++++++++++++++++++++++++++++++++++++
 myems-bacnet/main.py    |  5 +++
 3 files changed, 102 insertions(+), 1 deletion(-)
 create mode 100644 myems-bacnet/gateway.py

diff --git a/myems-bacnet/README.md b/myems-bacnet/README.md
index d90acf6c..532ee9a2 100644
--- a/myems-bacnet/README.md
+++ b/myems-bacnet/README.md
@@ -10,7 +10,7 @@ bacpypes
 
 mysql.connector
 
-
+Schedule
 
 ## Installation
 
@@ -38,6 +38,14 @@ Download and install bacpypes library
     $ sudo ufw allow 47808
 ```
 
+Download and install Schedule
+```
+$ cd ~/tools
+$ git clone https://github.com/dbader/schedule.git
+$ cd ~/tools/schedule
+$ sudo python3 setup.py install
+```
+
 Install myems-bacnet service
 ```
     $ cd ~
diff --git a/myems-bacnet/gateway.py b/myems-bacnet/gateway.py
new file mode 100644
index 00000000..c37e9323
--- /dev/null
+++ b/myems-bacnet/gateway.py
@@ -0,0 +1,88 @@
+import json
+import mysql.connector
+import config
+import time
+from datetime import datetime
+import requests
+from decimal import Decimal
+import schedule
+
+########################################################################################################################
+# Gateway Job Procedures
+# Step 1: Verify Gateway Token
+# Step 2: Collect Gateway Information
+# Step 3: Update Gateway Information
+########################################################################################################################
+
+
+def job(logger, ):
+    ################################################################################################################
+    # Step 1: Verify Gateway Token
+    ################################################################################################################
+    cnx_system_db = None
+    cursor_system_db = None
+    try:
+        cnx_system_db = mysql.connector.connect(**config.myems_system_db)
+        cursor_system_db = cnx_system_db.cursor()
+    except Exception as e:
+        logger.error("Error in step 1.1 of Gateway process " + str(e))
+        if cursor_system_db:
+            cursor_system_db.close()
+        if cnx_system_db:
+            cnx_system_db.close()
+        return
+
+    # TODO: choose a more secure method to verify gateway token
+    try:
+        query = (" SELECT name "
+                 " FROM tbl_gateways "
+                 " WHERE id = %s AND token = %s ")
+        cursor_system_db.execute(query, (config.gateway['id'], config.gateway['token']))
+        row = cursor_system_db.fetchone()
+    except Exception as e:
+        logger.error("Error in step 1.2 of gateway process: " + str(e))
+        if cursor_system_db:
+            cursor_system_db.close()
+        if cnx_system_db:
+            cnx_system_db.close()
+        return
+
+    if row is None:
+        logger.error("Error in step 1.3 of gateway process: Not Found ")
+        if cursor_system_db:
+            cursor_system_db.close()
+        if cnx_system_db:
+            cnx_system_db.close()
+        return
+
+    ############################################################################################################
+    # Step 2: Collect Gateway Information
+    ############################################################################################################
+    # todo: get more information, such as CPU/MEMORY/DISK
+    current_datetime_utc = datetime.utcnow()
+
+    ############################################################################################################
+    # Step 3: Update Gateway Information
+    ############################################################################################################
+    update_row = (" UPDATE tbl_gateways "
+                  " SET last_seen_datetime_utc = '" + current_datetime_utc.isoformat() + "' "
+                  " WHERE id = %s ")
+    try:
+        cursor_system_db.execute(update_row, (config.gateway['id'], ))
+        cnx_system_db.commit()
+    except Exception as e:
+        logger.error("Error in step 3.1 of gateway process " + str(e))
+        return
+    finally:
+        if cursor_system_db:
+            cursor_system_db.close()
+        if cnx_system_db:
+            cnx_system_db.close()
+
+
+def process(logger, ):
+    schedule.every(3).minutes.do(job, logger,)
+
+    while True:
+        schedule.run_pending()
+        time.sleep(60)
diff --git a/myems-bacnet/main.py b/myems-bacnet/main.py
index 15dc75cf..7b5fb08d 100644
--- a/myems-bacnet/main.py
+++ b/myems-bacnet/main.py
@@ -1,6 +1,7 @@
 import logging
 from logging.handlers import RotatingFileHandler
 import acquisition
+import gateway
 
 
 def main():
@@ -24,6 +25,10 @@ def main():
     # Create Acquisition Process
     ####################################################################################################################
     acquisition.process(logger,)
+    ####################################################################################################################
+    # Create Gateway Process
+    ####################################################################################################################
+    gateway.process(logger,)
 
 
 if __name__ == "__main__":