34 lines
848 B
Python
34 lines
848 B
Python
from io import BytesIO
|
|
import os
|
|
|
|
|
|
class DiskCache:
|
|
def __init__(self, path=None, enabled=True):
|
|
if path is None:
|
|
self.path = os.path.join(os.path.dirname(__file__), "cache")
|
|
else:
|
|
self.path = path
|
|
|
|
self.enabled = enabled
|
|
if self.enabled is False:
|
|
return
|
|
|
|
os.makedirs(self.path, exist_ok=True)
|
|
|
|
def set(self, identifier, buf):
|
|
if self.enabled is False:
|
|
return
|
|
|
|
with open(self._build_path(identifier), "wb") as f:
|
|
f.write(buf.read())
|
|
|
|
def get(self, identifier):
|
|
if self.enabled is False:
|
|
raise FileNotFoundError
|
|
|
|
with open(self._build_path(identifier), "rb") as f:
|
|
return BytesIO(f.read())
|
|
|
|
def _build_path(self, identifier):
|
|
return os.path.join(self.path, identifier)
|