Adding GUI update feature (server-side)

server/frontend/app/classes/update.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+from app.utils import read_config
+import subprocess as sp
+import requests
+import json
+import os
+import re
+
+
+class Update(object):
+
+    def __init__(self):
+        self.project_url = "https://api.github.com/repos/KasperskyLab/TinyCheck/tags"
+        self.app_path = "/usr/share/tinycheck"
+        return None
+
+    def check_version(self):
+        """
+            Check if a new version of TinyCheck is available 
+            by quering the Github api and comparing the last
+            tag with the VERSION file content.
+        """
+        if read_config(("frontend", "update")):
+            res = requests.get(self.project_url)
+            res = json.loads(res.content.decode("utf8"))
+
+            with open(os.path.join(self.app_path, "VERSION")) as f:
+                if f.read() != res[0]["name"]:
+                    return {"status": True,
+                            "message": "A new version is available"}
+                else:
+                    return {"status": True,
+                            "message": "This is the latest version"}
+        else:
+            return {"status": False,
+                    "message": "You don't have rights to do this operation."}
+
+    def update_instance(self):
+        """
+            Update the instance by executing 
+            the update script.
+        """
+        if read_config(("frontend", "update")):
+            try:
+                os.chdir(self.app_path)
+                sp.Popen(["bash", os.path.join(self.app_path, "update.sh")])
+                return {"status": True,
+                        "message": "Update successfully launched"}
+            except:
+                return {"status": False,
+                        "message": "Issue during the update"}
+        else:
+            return {"status": False,
+                    "message": "You don't have rights to do this operation."}