Create a new argument (reverting) for the BackupPersistent constructor.

It is necessary to differentiate between calling the BackupPersistent class
for backup purposes or for reverting purposes.
Also the grub revert function and the firstaidkitrevert file changed to
address the new changes.

1 files changed, 10 insertions, 0 deletions

diff --git a/pyfirstaidkit/utils/backup.py b/pyfirstaidkit/utils/backup.py
index 244e223..be086fb 100644
--- a/pyfirstaidkit/utils/backup.py
+++ b/pyfirstaidkit/utils/backup.py
@@ -238,6 +238,16 @@ class FileBackupStore(BackupStoreInterface):
             return False
 
     class BackupPersistent(Backup):
+        def __init__(self, id, path, reverting = False):
+            self._id = id
+            self._path = path
+            self._metafile = "__meta.pickle"
+            self._data = {} # name -> (stored as, origin)
+            self._origin = {} # origin -> name
+
+            if not reverting:
+                os.makedirs(self._path)
+
         def cleanup(self):
             self.saveMeta()
             return False