mirror of
https://github.com/RGBCube/GitHubWrapper
synced 2025-07-24 15:07:44 +00:00
PEP 8
This commit is contained in:
parent
706823da69
commit
66015d737a
7 changed files with 0 additions and 0 deletions
69
github/cache.py
Normal file
69
github/cache.py
Normal file
|
@ -0,0 +1,69 @@
|
|||
# == cache.py ==#
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from collections import deque
|
||||
from typing import Any, Deque, Tuple, TypeVar, Dict
|
||||
|
||||
__all__: Tuple[str, ...] = ('ObjectCache',)
|
||||
|
||||
|
||||
K = TypeVar('K')
|
||||
V = TypeVar('V')
|
||||
|
||||
|
||||
class _BaseCache(Dict[K, V]):
|
||||
"""This is a rough implementation of an LRU Cache using a deque and a dict."""
|
||||
|
||||
__slots__: Tuple[str, ...] = ('_max_size', '_lru_keys')
|
||||
|
||||
def __init__(self, max_size: int, *args: Any) -> None:
|
||||
self._max_size: int = max(min(max_size, 15), 0) # bounding max_size to 15 for now
|
||||
self._lru_keys: Deque[K] = deque(maxlen=self._max_size)
|
||||
super().__init__(*args)
|
||||
|
||||
def __getitem__(self, __k: K) -> V:
|
||||
index = self._lru_keys.index(__k)
|
||||
target = self._lru_keys[index]
|
||||
del self._lru_keys[index]
|
||||
|
||||
self._lru_keys.appendleft(target)
|
||||
return super().__getitem__(__k)
|
||||
|
||||
def __setitem__(self, __k: K, __v: V) -> None:
|
||||
if len(self) == self._max_size:
|
||||
self.__delitem__(self._lru_keys.pop())
|
||||
|
||||
self._lru_keys.appendleft(__k)
|
||||
return super().__setitem__(__k, __v)
|
||||
|
||||
def update(self, **kwargs: Any) -> None:
|
||||
for key, value in dict(**kwargs).items():
|
||||
key: K
|
||||
value: V
|
||||
|
||||
self.__setitem__(key, value)
|
||||
|
||||
|
||||
class ObjectCache(_BaseCache[K, V]):
|
||||
"""This adjusts the typehints to reflect Github objects."""
|
||||
|
||||
def __getitem__(self, __k: K) -> V:
|
||||
index = self._lru_keys.index(__k)
|
||||
target = self._lru_keys[index]
|
||||
self._lru_keys.appendleft(target)
|
||||
return super().__getitem__(__k)
|
||||
|
||||
def __setitem__(self, __k: K, __v: V) -> None:
|
||||
if self.__len__() == self._max_size:
|
||||
self.__delitem__(self._lru_keys.pop())
|
||||
|
||||
self._lru_keys.appendleft(__k)
|
||||
return super().__setitem__(__k, __v)
|
||||
|
||||
def update(self, **kwargs: Any) -> None:
|
||||
for key, value in dict(**kwargs).items():
|
||||
key: K
|
||||
value: V
|
||||
|
||||
self.__setitem__(key, value)
|
Loading…
Add table
Add a link
Reference in a new issue