Spaces:
Build error
Build error
File size: 1,420 Bytes
ab9ff53 |
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 |
import shelve
import shutil
from pathlib import Path
from typing import Any, Generic, TypeVar
T = TypeVar("T")
class ShelveDB(Generic[T]):
dir_path: Path
def __init__(self, db_name: str, init: bool) -> None:
self.db_path = self.dir_path / db_name
if init:
self.dir_path.mkdir(parents=True, exist_ok=True)
for file_path in self.dir_path.glob(f"{db_name}*"):
if file_path.is_file():
file_path.unlink()
elif file_path.is_dir():
shutil.rmtree(file_path)
@classmethod
def from_table(cls, table: str) -> "ShelveDB":
return cls(table, False)
def save(self, key: str, value: Any) -> None:
with shelve.open(str(self.db_path)) as db:
db[key] = value
def fetch(self, key: str) -> T | None:
with shelve.open(str(self.db_path)) as db:
return db.get(key, None)
def delete(self, key: str) -> bool:
with shelve.open(str(self.db_path)) as db:
if key in db:
del db[key]
return True
return False
def clear(self) -> None:
with shelve.open(str(self.db_path)) as db:
for key in list(db.keys()):
del db[key]
def list_keys(self) -> list[str]:
with shelve.open(str(self.db_path)) as db:
return list(db.keys())
|