From 280df98c2b961c77f86715fb818bf99df652e967 Mon Sep 17 00:00:00 2001
From: ahmed531998 <ahmed531998@aucegypt.edu>
Date: Mon, 24 Jul 2023 10:53:10 +0200
Subject: [PATCH] debug

---
 main.py | 101 +++++++++++++++++++++++++++++++++-----------------------
 1 file changed, 60 insertions(+), 41 deletions(-)

diff --git a/main.py b/main.py
index 21e93b7..88e9427 100644
--- a/main.py
+++ b/main.py
@@ -32,68 +32,87 @@ cors = CORS(app, resources={r"/api/predict": {"origins": url},
                             r"/health": {"origins": "*"}
                            })
 users = {}
+alive = "alive"
 
 def vre_fetch():
     while True:
-        time.sleep(1000)
-        print('getting new material')
-        #users[token]['args']['vre'].get_vre_update()
-        #users[token]['args']['vre'].index_periodic_update()
-        #users[token]['args']['rg'].update_index(vre.get_index())
-        #users[token]['args']['rg'].update_db(vre.get_db())
-        vre.get_vre_update()
-        vre.index_periodic_update()
-        rg.update_index(vre.get_index())
-        rg.update_db(vre.get_db())
+        try:
+            time.sleep(1000)
+            print('getting new material')
+            #users[token]['args']['vre'].get_vre_update()
+            #users[token]['args']['vre'].index_periodic_update()
+            #users[token]['args']['rg'].update_index(vre.get_index())
+            #users[token]['args']['rg'].update_db(vre.get_db())
+            vre.get_vre_update()
+            vre.index_periodic_update()
+            rg.update_index(vre.get_index())
+            rg.update_db(vre.get_db())
+        except Exception as e:
+            alive = "dead_vre_fetch"
+            
 
 
 def user_interest_decay(token):
     while True:
-        if token in users:
-            print("decaying interests after 3 minutes for " + users[token]['username'])
-            time.sleep(180)
-            users[token]['user'].decay_interests()
-        else:
-             break
+        try:
+            if token in users:
+                print("decaying interests after 3 minutes for " + users[token]['username'])
+                time.sleep(180)
+                users[token]['user'].decay_interests()
+            else:
+                break
+        except Exception as e:
+            alive = "dead_interest_decay"
 
 
 def clear_inactive():
     while True:
-        time.sleep(1)
-        for username in users:
-            if users[username]['activity'] > 3600:
-                del users[username]
-            users[username]['activity'] += 1
+        try:
+            time.sleep(1)
+            for username in users:
+                if users[username]['activity'] > 3600:
+                    del users[username]
+                users[username]['activity'] += 1
+        except Exception as e:
+            alive = "dead_clear_inactive"
 
 @app.route("/health", methods=['GET'])
 def health():
-    return "Success", 200
+    if alive=="alive":
+        return "Success", 200
+    else:
+        return alive, 500
 
 @app.route("/api/dm", methods=['POST'])
 def init_dm():
-    token = request.get_json().get("token")
-    status = request.get_json().get("stat")
-    if status == "start":
-        message = {"stat": "waiting"}
-    elif status == "set":
-        headers = {"gcube-token": token, "Accept": "application/json"}
-        if token not in users:
-            url = 'https://api.d4science.org/rest/2/people/profile'
-            response = requests.get(url, headers=headers)
-            if response.status_code == 200:
-                username = response.json()['result']['username']
-                name = response.json()['result']['fullname']
+    try:
+        token = request.get_json().get("token")
+        status = request.get_json().get("stat")
+        if status == "start":
+            message = {"stat": "waiting"}
+        elif status == "set":
+            headers = {"gcube-token": token, "Accept": "application/json"}
+            if token not in users:
+                url = 'https://api.d4science.org/rest/2/people/profile'
+                response = requests.get(url, headers=headers)
+                if response.status_code == 200:
+                    username = response.json()['result']['username']
+                    name = response.json()['result']['fullname']
                 
-                users[token] = {'username': username, 'name': name, 'dm': DM(), 'activity': 0, 'user': User(username, token)}
+                    users[token] = {'username': username, 'name': name, 'dm': DM(), 'activity': 0, 'user': User(username, token)}
 
-                threading.Thread(target=user_interest_decay, args=(token,), name='decayinterest_'+users[token]['username']).start()
+                    threading.Thread(target=user_interest_decay, args=(token,), name='decayinterest_'+users[token]['username']).start()
 
-                message = {"stat": "done"}
+                    message = {"stat": "done"}
+                else:
+                    message = {"stat": "rejected"}
             else:
-                message = {"stat": "rejected"}
-        else:
-            message = {"stat": "done"}
-    return message
+                message = {"stat": "done"}
+        return message
+    except Exception as e:
+        message = {"stat": "init_dm_error"}
+        return message
+
 
 @app.route("/api/predict", methods=['POST'])
 def predict():