bb1d662ee1
Rebrand the Rust client crate (coven/ → hh/, package+binary "hack-house"), README, CLI strings, and branch (coven → hack-house). Gitea repo renamed cmd-chat → hack-house to match. Crypto/server logic unchanged; selftest + golden-vector test still green, binary is now `hack-house`. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
84 lines
2.3 KiB
Python
84 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import ItemsView, Iterator, KeysView, Mapping, ValuesView
|
|
from collections.abc import Mapping as MappingType
|
|
from typing import Any
|
|
|
|
|
|
class WorkerState(Mapping):
|
|
RESTRICTED = (
|
|
"health",
|
|
"pid",
|
|
"requests",
|
|
"restart_at",
|
|
"server",
|
|
"start_at",
|
|
"starts",
|
|
"state",
|
|
)
|
|
|
|
def __init__(self, state: dict[str, Any], current: str) -> None:
|
|
self._name = current
|
|
self._state = state
|
|
|
|
def __getitem__(self, key: str) -> Any:
|
|
return self._state[self._name][key]
|
|
|
|
def __setitem__(self, key: str, value: Any) -> None:
|
|
if key in self.RESTRICTED:
|
|
self._write_error([key])
|
|
self._state[self._name] = {
|
|
**self._state[self._name],
|
|
key: value,
|
|
}
|
|
|
|
def __delitem__(self, key: str) -> None:
|
|
if key in self.RESTRICTED:
|
|
self._write_error([key])
|
|
self._state[self._name] = {
|
|
k: v for k, v in self._state[self._name].items() if k != key
|
|
}
|
|
|
|
def __iter__(self) -> Iterator[Any]:
|
|
return iter(self._state[self._name])
|
|
|
|
def __len__(self) -> int:
|
|
return len(self._state[self._name])
|
|
|
|
def __repr__(self) -> str:
|
|
return repr(self._state[self._name])
|
|
|
|
def __eq__(self, other: object) -> bool:
|
|
return self._state[self._name] == other
|
|
|
|
def keys(self) -> KeysView[str]:
|
|
return self._state[self._name].keys()
|
|
|
|
def values(self) -> ValuesView[Any]:
|
|
return self._state[self._name].values()
|
|
|
|
def items(self) -> ItemsView[str, Any]:
|
|
return self._state[self._name].items()
|
|
|
|
def update(self, mapping: MappingType[str, Any]) -> None:
|
|
if any(k in self.RESTRICTED for k in mapping.keys()):
|
|
self._write_error(
|
|
[k for k in mapping.keys() if k in self.RESTRICTED]
|
|
)
|
|
self._state[self._name] = {
|
|
**self._state[self._name],
|
|
**mapping,
|
|
}
|
|
|
|
def pop(self) -> None:
|
|
raise NotImplementedError
|
|
|
|
def full(self) -> dict[str, Any]:
|
|
return dict(self._state)
|
|
|
|
def _write_error(self, keys: list[str]) -> None:
|
|
raise LookupError(
|
|
f"Cannot set restricted key{'s' if len(keys) > 1 else ''} on "
|
|
f"WorkerState: {', '.join(keys)}"
|
|
)
|