Heroku/hikka/pointers.py

135 lines
3.3 KiB
Python

import typing
class PointerList(list):
"""Pointer to list saved in database"""
def __init__(
self,
db: "Database", # type: ignore
module: str,
key: str,
default: typing.Optional[typing.Any] = None,
):
self._db = db
self._module = module
self._key = key
self._default = default
super().__init__(db.get(module, key, default))
def __repr__(self):
return f"PointerList({list(self)})"
def __str__(self):
return f"PointerList({list(self)})"
def __delitem__(self, __i: typing.Union[typing.SupportsIndex, slice]) -> None:
a = super().__delitem__(__i)
self._save()
return a
def __setitem__(
self, __i: typing.Union[typing.SupportsIndex, slice], __v: typing.Any
) -> None:
a = super().__setitem__(__i, __v)
self._save()
return a
def __iadd__(self, __x: typing.Iterable) -> "Self": # type: ignore
a = super().__iadd__(__x)
self._save()
return a
def __imul__(self, __x: int) -> "Self": # type: ignore
a = super().__imul__(__x)
self._save()
return a
def append(self, value: typing.Any):
super().append(value)
self._save()
def extend(self, value: typing.Iterable):
super().extend(value)
self._save()
def insert(self, index: int, value: typing.Any):
super().insert(index, value)
self._save()
def remove(self, value: typing.Any):
super().remove(value)
self._save()
def pop(self, index: int = -1) -> typing.Any:
a = super().pop(index)
self._save()
return a
def clear(self) -> None:
super().clear()
self._save()
def _save(self):
self._db.set(self._module, self._key, list(self))
class PointerDict(dict):
"""Pointer to dict saved in database"""
def __init__(
self,
db: "Database", # type: ignore
module: str,
key: str,
default: typing.Optional[typing.Any] = None,
):
self._db = db
self._module = module
self._key = key
self._default = default
super().__init__(db.get(module, key, default))
def __repr__(self):
return f"PointerDict({dict(self)})"
def __bool__(self) -> bool:
return bool(self._db.get(self._module, self._key, self._default))
def __setitem__(self, key: str, value: typing.Any):
super().__setitem__(key, value)
self._save()
def __delitem__(self, key: str):
super().__delitem__(key)
self._save()
def __str__(self):
return f"PointerDict({dict(self)})"
def update(self, __m: dict) -> None:
super().update(__m)
self._save()
def setdefault(self, key: str, default: typing.Any = None) -> typing.Any:
a = super().setdefault(key, default)
self._save()
return a
def pop(self, key: str, default: typing.Any = None) -> typing.Any:
a = super().pop(key, default)
self._save()
return a
def popitem(self) -> tuple:
a = super().popitem()
self._save()
return a
def clear(self) -> None:
super().clear()
self._save()
def _save(self):
self._db.set(self._module, self._key, dict(self))