package

mrmp/catalog.py

@@ -0,0 +1,36 @@
+import io
+import os
+import tarfile
+
+import package
+
+CATALOG = '.files'
+FILES = 'files'
+PACKAGE = 'desc'
+
+
+class Catalog:
+    def __init__(self, subsystem):
+        self.subsystem = subsystem
+        self.path = os.path.join(self.subsystem.path,
+                                 f'{self.subsystem.name}{CATALOG}')
+        self.load()
+
+    def load(self):
+        binary = self.subsystem.architecture.repository.get_file(self.path)
+        f = io.BytesIO(binary)
+        archive = tarfile.open(fileobj=f)
+        m = {}
+        packages = {}
+        for member in archive.getmembers():
+            directory, *file = member.name.split(os.sep)
+            if file:
+                d = m[directory]
+                d[file[0]] = archive.extractfile(member).read()
+                if len(d) == 2:
+                    p = package.Package(d[PACKAGE], d[FILES])
+                    packages[p.name] = p
+            else:
+                m[directory] = {}
+        archive.close()
+        self.packages = packages