| |
| from __future__ import annotations |
| from dataclasses import dataclass, asdict |
| from typing import Dict, List, Optional |
| from pathlib import Path |
| import json, time |
|
|
| PROFILE_DIR = Path("memory/.profiles") |
| PROFILE_DIR.mkdir(parents=True, exist_ok=True) |
|
|
| @dataclass |
| class Note: |
| key: str |
| value: str |
| created_at: float |
| updated_at: float |
| tags: List[str] |
|
|
| @dataclass |
| class Profile: |
| user_id: str |
| display_name: Optional[str] = None |
| notes: Dict[str, Note] = None |
|
|
| @classmethod |
| def load(cls, user_id: str) -> "Profile": |
| p = PROFILE_DIR / f"{user_id}.json" |
| if not p.exists(): |
| return Profile(user_id=user_id, notes={}) |
| data = json.loads(p.read_text(encoding="utf-8")) |
| notes = {k: Note(**v) for k, v in data.get("notes", {}).items()} |
| return Profile(user_id=data["user_id"], display_name=data.get("display_name"), notes=notes) |
|
|
| def save(self) -> None: |
| p = PROFILE_DIR / f"{self.user_id}.json" |
| data = { |
| "user_id": self.user_id, |
| "display_name": self.display_name, |
| "notes": {k: asdict(v) for k, v in (self.notes or {}).items()}, |
| } |
| p.write_text(json.dumps(data, ensure_ascii=False, indent=2), encoding="utf-8") |
|
|
| |
| def remember(self, key: str, value: str, tags: Optional[List[str]] = None) -> None: |
| now = time.time() |
| note = self.notes.get(key) |
| if note: |
| note.value, note.updated_at = value, now |
| if tags: note.tags = tags |
| else: |
| self.notes[key] = Note(key=key, value=value, tags=tags or [], created_at=now, updated_at=now) |
| self.save() |
|
|
| def forget(self, key: str) -> bool: |
| ok = key in self.notes |
| if ok: |
| self.notes.pop(key) |
| self.save() |
| return ok |
|
|
| def recall(self, key: str) -> Optional[str]: |
| n = self.notes.get(key) |
| return n.value if n else None |
|
|
| def list_notes(self) -> List[str]: |
| return sorted(self.notes.keys()) |
|
|
