First Commit

src/filesystem_storage.py

@@ -0,0 +1,56 @@
+import os
+import shutil
+import logging
+
+logger = logging.getLogger(__name__)
+
+BASE_DIRECTORY = './embedding_data'
+
+class FileSystemStorage:
+    def __init__(self, base_directory=BASE_DIRECTORY):
+        self.base_dir = base_directory
+
+    def create_group(self, auth_token, group_id):
+        path = os.path.join(self.base_dir, group_id)
+        if os.path.exists(path):
+            raise FileExistsError(f"{group_id}")
+        else:
+            try:
+                os.makedirs(path)
+            except OSError as e:
+                logger.error(f"Error creating directory {path}: {e}")
+                raise
+
+    def get(self, auth_token, group_id, item_id):
+        try:
+            path = os.path.join(self.base_dir, group_id, item_id)
+            with open(path, 'rb') as f:
+                return f.read()
+        except Exception as e:
+            logger.error(f"Error reading from {path}: {e}")
+            raise
+
+    def save(self, auth_token, group_id, item_id, content):
+        try:
+            path = os.path.join(self.base_dir, group_id, item_id)
+            with open(path, 'wb') as f:
+                f.write(content)
+        except Exception as e:
+            logger.error(f"Error saving to {path}: {e}")
+            raise
+            
+    def remove(slef, auth_token, group_id, item_id):
+        try:
+            path = os.path.join(self.base_dir, group_id, item_id)
+            os.remove(path)
+        except Exception as e:
+            logger.error(f"Error remove item {path}: {e}")
+            raise
+            
+    def remove_group(self, auth_token, group_id):
+        try:
+            path = os.path.join(self.base_dir, group_id)
+            shutil.rmtree(path)
+        except Exception as e:
+            logger.error(f"Error remove group {path}: {e}")
+            raise