mirror of
https://github.com/kevinveenbirkenbach/homepage.veen.world.git
synced 2025-07-23 17:31:10 +02:00
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
import os
|
|
import hashlib
|
|
import requests
|
|
import mimetypes
|
|
|
|
class CacheManager:
|
|
def __init__(self, cache_dir="static/cache"):
|
|
self.cache_dir = cache_dir
|
|
self._ensure_cache_dir_exists()
|
|
|
|
def _ensure_cache_dir_exists(self):
|
|
if not os.path.exists(self.cache_dir):
|
|
os.makedirs(self.cache_dir)
|
|
|
|
def clear_cache(self):
|
|
if os.path.exists(self.cache_dir):
|
|
for filename in os.listdir(self.cache_dir):
|
|
path = os.path.join(self.cache_dir, filename)
|
|
if os.path.isfile(path):
|
|
os.remove(path)
|
|
|
|
def cache_file(self, file_url):
|
|
# generate a short hash for filename
|
|
hash_suffix = hashlib.blake2s(file_url.encode('utf-8'), digest_size=8).hexdigest()
|
|
parts = file_url.rstrip("/").split("/")
|
|
base = parts[-2] if parts[-1] == "download" else parts[-1]
|
|
|
|
try:
|
|
resp = requests.get(file_url, stream=True, timeout=5)
|
|
resp.raise_for_status()
|
|
except requests.RequestException:
|
|
return None
|
|
|
|
content_type = resp.headers.get('Content-Type', '')
|
|
ext = mimetypes.guess_extension(content_type.split(";")[0].strip()) or ".png"
|
|
filename = f"{base}_{hash_suffix}{ext}"
|
|
full_path = os.path.join(self.cache_dir, filename)
|
|
|
|
if not os.path.exists(full_path):
|
|
with open(full_path, "wb") as f:
|
|
for chunk in resp.iter_content(1024):
|
|
f.write(chunk)
|
|
|
|
# return path relative to /static/
|
|
return f"cache/{filename}"
|