| 12345678910111213141516171819202122232425262728 |
- from __future__ import annotations
- import collections
- from typing import Any
- class CappedDict(collections.OrderedDict):
- default_max_size = 50
- def __init__(self, max_size: int | None = None) -> None:
- self.max_size = max_size or self.default_max_size
- super().__init__()
- def __setitem__(self, key: str, val: Any) -> None:
- if key not in self:
- max_size = self.max_size - 1
- self._prune_dict(max_size)
- super().__setitem__(key, val)
- def update(self, **kwargs: Any) -> None: # type: ignore[override]
- super().update(**kwargs)
- self._prune_dict(self.max_size)
- def _prune_dict(self, max_size: int) -> None:
- if len(self) >= max_size:
- diff = len(self) - max_size
- for k in list(self.keys())[:diff]:
- del self[k]
|
