package

mrmp/local.py

@@ -0,0 +1,31 @@
+import datetime
+import os
+
+import arguments
+import repository
+
+
+class Local(repository.Repository):
+    def __init__(self):
+        super().__init__(arguments.directory)
+        self.temporary = arguments.temporary
+
+    def get_file(self, path):
+        with open(os.path.join(self.location, path), 'br') as f:
+            return f.read()
+
+    def get_files(self, path):
+        *_, files = next(os.walk(os.path.join(self.location, path)))
+        return files
+
+    def get_temporary(self):
+        return os.path.join(self.temporary,
+                            datetime.datetime.now()
+                            .strftime('%Y%m%d%H%M%S'))
+
+    def __str__(self):
+        lines = [
+            super().__str__(),
+            f'Temporary: {self.temporary}',
+        ]
+        return os.linesep.join(lines)