new templates integrity check to ease updates

client.py

@@ -1,6 +1,6 @@
 import requests
 from ast import literal_eval
-
+from util import *
 
 def send_text_scan(text):
     r = requests.post("http://127.0.0.1:21987/upload",json={"type" : "text", "content" : f"{text}"})
@@ -40,4 +40,6 @@ def send_url(url):
 #send_file("static/qr.jpeg")
 #send_text_scan("test of text scan")
 #send_wifi("This_is_a_wifi_name","wpa","password")
-#send_url("https://www.google.com")
+#send_url("https://www.google.com")
+
+check_templates_update()

util.py

@@ -17,6 +17,10 @@
 from platform import system
 
 
+
+TEMPLATES_FILES = ["favicon.ico","index.html","scan_preview.html","img_preview.html","video_preview.html"]
+
+
 def notify_desktop(title,text):
     # initialize 
     toaster = ToastNotifier()
@@ -58,6 +62,15 @@ def make_qr_url():
 
 
 def check_templates_update():
+
+    # check templates integrity
+    for ele in TEMPLATES_FILES:
+        if not path.exists(f"templates/{ele}"):
+            download_templates()
+            return
+
+
+    # check 10 startup rule
     with open("static/update.Blue","r") as f:
         n = int(f.read())
         if n == 10:
@@ -111,25 +124,11 @@ def is_server_already_running():
 def download_templates():
 
     #get the templates
-    r = get(f"https://raw.githubusercontent.com/copypastaofficial/copypasta/master/templates/index.html",allow_redirects=True)
-    with open("templates/index.html","wb") as f:
-        f.write(r.content)
-
-    r = get(f"https://raw.githubusercontent.com/copypastaofficial/copypasta/master/templates/scan_preview.html",allow_redirects=True)
-    with open("templates/scan_preview.html","wb") as f:
-        f.write(r.content)
-
-    r = get(f"https://raw.githubusercontent.com/copypastaofficial/copypasta/master/templates/img_preview.html",allow_redirects=True)
-    with open("templates/img_preview.html","wb") as f:
-        f.write(r.content)
+    for ele in TEMPLATES_FILES:
 
-    r = get(f"https://raw.githubusercontent.com/copypastaofficial/copypasta/master/templates/video_preview.html",allow_redirects=True)
-    with open("templates/img_preview.html","wb") as f:
-        f.write(r.content)
-
-    r = get(f"https://raw.githubusercontent.com/copypastaofficial/copypasta/master/templates/favicon.ico",allow_redirects=True)
-    with open("static/favicon.ico","wb") as f:
-        f.write(r.content)
+        r = get(f"https://raw.githubusercontent.com/copypastaofficial/copypasta/master/templates/{ele}",allow_redirects=True)
+        with open(f"templates/{ele}","wb") as f:
+            f.write(r.content)
 
     with open("static/update.Blue","w") as f:
         f.write("1")