File size: 2,660 Bytes
c0fdfb4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | """Spell Checker — Suggestion service layer."""
from __future__ import annotations
import logging
from typing import Any, Dict, List, Optional
logger = logging.getLogger(__name__)
class SpellModels:
"""Business-logic service for Suggestion operations in Spell Checker."""
def __init__(
self,
repo: Any,
events: Optional[Any] = None,
) -> None:
self._repo = repo
self._events = events
logger.debug("SpellModels started")
def ignore(
self, payload: Dict[str, Any]
) -> Dict[str, Any]:
"""Execute the ignore workflow for a new Suggestion."""
if "source" not in payload:
raise ValueError("Missing required field: source")
record = self._repo.insert(
payload["source"], payload.get("confidence"),
**{k: v for k, v in payload.items()
if k not in ("source", "confidence")}
)
if self._events:
self._events.emit("suggestion.ignored", record)
return record
def check(self, rec_id: str, **changes: Any) -> Dict[str, Any]:
"""Apply *changes* to a Suggestion and emit a change event."""
ok = self._repo.update(rec_id, **changes)
if not ok:
raise KeyError(f"Suggestion {rec_id!r} not found")
updated = self._repo.fetch(rec_id)
if self._events:
self._events.emit("suggestion.checkd", updated)
return updated
def correct(self, rec_id: str) -> None:
"""Remove a Suggestion and emit a removal event."""
ok = self._repo.delete(rec_id)
if not ok:
raise KeyError(f"Suggestion {rec_id!r} not found")
if self._events:
self._events.emit("suggestion.correctd", {"id": rec_id})
def search(
self,
source: Optional[Any] = None,
status: Optional[str] = None,
limit: int = 50,
) -> List[Dict[str, Any]]:
"""Search suggestions by *source* and/or *status*."""
filters: Dict[str, Any] = {}
if source is not None:
filters["source"] = source
if status is not None:
filters["status"] = status
rows, _ = self._repo.query(filters, limit=limit)
logger.debug("search suggestions: %d hits", len(rows))
return rows
@property
def stats(self) -> Dict[str, int]:
"""Quick summary of Suggestion counts by status."""
result: Dict[str, int] = {}
for status in ("active", "pending", "closed"):
_, count = self._repo.query({"status": status}, limit=0)
result[status] = count
return result
|