From b3d0b1fb867daeabdb2fd73d5427e6efc8c4e8d2 Mon Sep 17 00:00:00 2001
From: lonpatovaadelina <lonpatovaadelina@mail.ru>
Date: Thu, 28 May 2026 11:50:07 +0000
Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?=
 =?UTF-8?q?=D0=BD=20memory=5Fserver.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 memory_server.py | 170 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 170 insertions(+)
 create mode 100644 memory_server.py

diff --git a/memory_server.py b/memory_server.py
new file mode 100644
index 0000000..f95bf6c
--- /dev/null
+++ b/memory_server.py
@@ -0,0 +1,170 @@
+# memory_server.py
+
+from fastmcp import FastMCP
+import json
+from datetime import datetime
+from pathlib import Path
+from typing import Any, Optional, List
+import fnmatch
+
+
+class MemoryServer:
+    """
+    MCP‑сервер для управления памятью с поддержкой namespaces.
+    """
+
+    def __init__(self):
+        # Инициализируем FastMCP
+        self.mcp = FastMCP("Memory-Server")
+        # Путь к файлу хранения данных
+        self.storage_path = Path("./memory_data.json")
+
+    # ------------------------------------------------------------------
+    # Вспомогательные методы для работы с JSON‑файлом
+    # ------------------------------------------------------------------
+    def _load_memory(self) -> dict:
+        """Загружает память из JSON‑файла."""
+        if not self.storage_path.exists():
+            return {}
+        with open(self.storage_path, "r", encoding="utf-8") as f:
+            return json.load(f)
+
+    def _save_memory(self, data: dict):
+        """Сохраняет память в JSON‑файл."""
+        self.storage_path.parent.mkdir(parents=True, exist_ok=True)
+        with open(self.storage_path, "w", encoding="utf-8") as f:
+            json.dump(data, f, indent=2, ensure_ascii=False)
+
+    # ------------------------------------------------------------------
+    # Инструменты FastMCP
+    # ------------------------------------------------------------------
+
+    @self.mcp.tool()
+    def save(key: str, value: Any) -> bool:
+        """
+        Сохраняет значение по ключу в память сервера.
+
+        Args:
+            key: Идентификатор для сохранения (уникальный ключ).
+            value: Любое сериализуемое значение.
+
+        Returns:
+            True при успешном сохранении, False иначе.
+        """
+        data = self._load_memory()
+        # Проверяем наличие опасных символов в ключе
+        if "/" in key or "\\" in key:
+            return False
+        data[key] = {
+            "value": value,
+            "timestamp": datetime.utcnow().isoformat(),
+        }
+        self._save_memory(data)
+        return True
+
+    @self.mcp.tool()
+    def get(key: str) -> Optional[dict]:
+        """
+        Возвращает значение по ключу с метаданными.
+
+        Args:
+            key: Идентификатор для поиска.
+
+        Returns:
+            Словарь с полями {"key": ..., "value": ..., "timestamp": ...}
+            или None если ключ не найден.
+        """
+        data = self._load_memory()
+        if key not in data:
+            return None
+        entry = data[key]
+        return {
+            "key": key,
+            "value": entry["value"],
+            "timestamp": entry["timestamp"],
+        }
+
+    @self.mcp.tool()
+    def delete(key: str) -> bool:
+        """
+        Удаляет ключ из памяти сервера.
+
+        Args:
+            key: Идентификатор для удаления.
+
+        Returns:
+            True при успешном удалении, False если ключ не найден.
+        """
+        data = self._load_memory()
+        if key not in data:
+            return False
+        del data[key]
+        self._save_memory(data)
+        return True
+
+    @self.mcp.tool()
+    def list_keys(pattern: str = "*") -> List[str]:
+        """
+        Возвращает список всех ключей с поддержкой wildcard‑паттерна.
+
+        Args:
+            pattern: Паттерн для фильтрации (поддерживает * и ?).
+
+        Returns:
+            Список совпадающих ключей.
+        """
+        data = self._load_memory()
+        return [k for k in data.keys() if fnmatch.fnmatch(k, pattern)]
+
+    # ------------------------------------------------------------------
+    # Инструменты с поддержкой namespaces
+    # ------------------------------------------------------------------
+
+    @self.mcp.tool()
+    def save_with_namespace(key: str, value: Any, namespace: str = "default") -> bool:
+        """
+        Сохраняет значение с указанием пространства имён.
+
+        Args:
+            key: Идентификатор.
+            value: Значение для сохранения.
+            namespace: Пространство имён (по умолчанию 'default').
+
+        Returns:
+            True при успехе, False иначе.
+        """
+        full_key = f"{namespace}:{key}"
+        return self.save(full_key, value)
+
+    @self.mcp.tool()
+    def get_by_namespace(namespace: str = "default") -> List[dict]:
+        """
+        Возвращает все ключи из указанного namespace.
+
+        Args:
+            namespace: Пространство имён для чтения.
+
+        Returns:
+            Список словарей с метаданными всех ключей namespace.
+        """
+        data = self._load_memory()
+        prefix = f"{namespace}:"
+        result = []
+        for k, v in data.items():
+            if k.startswith(prefix):
+                result.append(
+                    {
+                        "key": k,
+                        "value": v["value"],
+                        "timestamp": v["timestamp"],
+                    }
+                )
+        return result
+
+
+# ----------------------------------------------------------------------
+# Запуск сервера
+# ----------------------------------------------------------------------
+if __name__ == "__main__":
+    server = MemoryServer()
+    server.mcp.run(transport="stdio", show_banner=False, log_level="ERROR")
\ No newline at end of file