Added base backup manager

iosfu/backup.py

@@ -0,0 +1,70 @@
+from os import listdir
+from os.path import join as join_paths, basename, isdir, isfile
+
+from .conf import BACKUPS_PATH
+
+
+class BackupManager(object):
+    # Path to backups
+    path = None
+
+    # Backups loaded
+    backups = {}
+
+    def __init__(self):
+        self.path = BACKUPS_PATH
+
+    def lookup(self):
+        """
+        Look for backup folders on PATH
+        """
+        folders = listdir(self.path)
+        for dirname in folders:
+            path = join_paths(self.path, dirname)
+            if isdir(path):
+                backup = Backup(path)
+                self.backups[backup.id] = backup
+
+
+class Backup(object):
+    """
+    Backup object
+    """
+    id = None
+    name = None
+    path = None
+    files = []
+
+    valid = True
+
+    _required_files = [
+        'Info.plist', 'Manifest.mbdb', 'Manifest.plist', 'Status.plist'
+    ]
+
+    def __init__(self, path):
+        self.path = path
+        self.get_info()
+        self.check()
+
+    def get_info(self):
+        self.id = basename(self.path)
+        self.name = self.id
+        self.files = [
+            x for x in listdir(self.path) if isfile(join_paths(self.path, x))
+        ]
+
+    def check(self):
+        for required_file in self._required_files:
+            if required_file not in self.files:
+                self.valid = False
+
+    def exists(self, filename):
+        return filename in self.files
+
+
+### TEST
+# manager = BackupManager()
+# manager.lookup()
+
+# for k, backup in manager.backups.items():
+#     print(backup.id, backup.valid, backup.exists('Info.plist'))

iosfu/conf.py

@@ -0,0 +1,9 @@
+from os import getcwd
+from os.path import join as join_paths
+
+
+# Main
+ROOT_PATH = getcwd()
+
+# Backups folder name
+BACKUPS_PATH = join_paths(ROOT_PATH, '_backups')