47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
from fastmcp import MCP
|
|
import json, os, time
|
|
|
|
# Simple in-memory store with JSON persistence
|
|
DATA_FILE = "data.json"
|
|
store = {}
|
|
if os.path.exists(DATA_FILE):
|
|
with open(DATA_FILE) as f:
|
|
store = json.load(f)
|
|
|
|
@MCP.tool()
|
|
def save(key: str, value: str):
|
|
store[key] = {"value": value, "timestamp": time.time()}
|
|
with open(DATA_FILE, "w") as f:
|
|
json.dump(store, f)
|
|
return f"Saved {key}"
|
|
|
|
@MCP.tool()
|
|
def get(key: str):
|
|
return store.get(key, {}).get("value", None)
|
|
|
|
@MCP.tool()
|
|
def delete(key: str):
|
|
if key in store:
|
|
del store[key]
|
|
with open(DATA_FILE, "w") as f:
|
|
json.dump(store, f)
|
|
return f"Deleted {key}"
|
|
return f"{key} not found"
|
|
|
|
@MCP.tool()
|
|
def list_keys():
|
|
return list(store.keys())
|
|
|
|
# Namespace helpers
|
|
@MCP.tool()
|
|
def save_with_namespace(namespace: str, key: str, value: str):
|
|
full = f"{namespace}:{key}"
|
|
return save(full, value)
|
|
|
|
@MCP.tool()
|
|
def get_by_namespace(namespace: str, key: str):
|
|
full = f"{namespace}:{key}"
|
|
return get(full)
|
|
|
|
if __name__ == "__main__":
|
|
MCP.run() |