Merge pull request #672 from NebraLtd/backup_updates

Add wingbits to backup

hw_diag/utilities/backup/__init__.py

@@ -6,14 +6,16 @@
 from hw_diag.utilities.backup.myst import MystBackupRestore
 from hw_diag.utilities.backup.thingsix import ThingsIXBackupRestore
 from hw_diag.utilities.backup.nebra import NebraBackupRestore
+from hw_diag.utilities.backup.wingbits import WingbitsBackupRestore
 from hw_diag.utilities.db import get_value, set_value
 from hw_diag.utilities.crypto import empty_hash
 
 
 PLUGINS = [
     MystBackupRestore,
     ThingsIXBackupRestore,
-    NebraBackupRestore
+    NebraBackupRestore,
+    WingbitsBackupRestore
 ]
 
 

hw_diag/utilities/backup/wingbits.py

@@ -0,0 +1,18 @@
+import os
+import shutil
+
+from hw_diag.utilities.backup.base import BaseBackupRestore
+
+
+CONFIG_FILE = '/var/nebra/wingbits.json'
+
+
+class WingbitsBackupRestore(BaseBackupRestore):
+    name = 'WINGBITS'
+
+    def backup(self):
+        os.mkdir(self.tmpdir)
+        shutil.copyfile(CONFIG_FILE, '%s/wingbits.json' % self.tmpdir)
+
+    def restore(self):
+        os.system('cp %s/wingbits.json %s' % (self.tmpdir, CONFIG_FILE))  # nosec

hw_diag/views/backup_restore.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import time
 
 from flask import Blueprint
 from flask import render_template
@@ -29,11 +30,12 @@
 def get_backup_page():
     diagnostics = read_diagnostics_file()
     claim_deeplink = claim_miner_deeplink()
-    
+
     return render_template(
         'backup_restore.html', 
         diagnostics=diagnostics,
-        claim_deeplink=claim_deeplink
+        claim_deeplink=claim_deeplink,
+        now=round(time.time())
     )
 
 
@@ -64,7 +66,8 @@ def do_restore():
         return render_template(
             'reconfigure_countdown.html',
             seconds=120,
-            next_url='/'
+            next_url='/',
+            now=round(time.time())
         )
     except Exception:
         return 'Error during restoration', 500