mirror of
https://github.com/spalencsar/bearhub.git
synced 2026-07-07 00:04:15 +02:00
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
import json
|
|
import os
|
|
from pathlib import Path
|
|
from threading import Thread, Lock
|
|
from typing import List, Dict
|
|
|
|
from bauh.core.model import Application
|
|
from bauh.util.cache import Cache
|
|
|
|
|
|
class DiskCacheLoader(Thread):
|
|
|
|
def __init__(self, enabled: bool, cache_map: Dict[type, Cache], apps: List[Application] = []):
|
|
super(DiskCacheLoader, self).__init__(daemon=True)
|
|
self.apps = apps
|
|
self.stop = False
|
|
self.lock = Lock()
|
|
self.cache_map = cache_map
|
|
self.enabled = enabled
|
|
|
|
def run(self):
|
|
|
|
if self.enabled:
|
|
while True:
|
|
if self.apps:
|
|
self.lock.acquire()
|
|
app = self.apps[0]
|
|
del self.apps[0]
|
|
self.lock.release()
|
|
self.fill_cached_data(app)
|
|
elif self.stop:
|
|
break
|
|
|
|
def add(self, app: Application):
|
|
if self.enabled:
|
|
if app and app.supports_disk_cache():
|
|
self.lock.acquire()
|
|
self.apps.append(app)
|
|
self.lock.release()
|
|
|
|
def fill_cached_data(self, app: Application):
|
|
if self.enabled:
|
|
if os.path.exists(app.get_disk_data_path()):
|
|
with open(app.get_disk_data_path()) as f:
|
|
cached_data = json.loads(f.read())
|
|
app.fill_cached_data(cached_data)
|
|
self.cache_map.get(app.__class__).add_non_existing(app.base_data.id, cached_data)
|
|
|
|
|
|
class DiskCacheLoaderFactory:
|
|
|
|
def __init__(self, disk_cache: bool, cache_map: Dict[type, Cache]):
|
|
self.disk_cache = disk_cache
|
|
self.cache_map = cache_map
|
|
|
|
def new(self):
|
|
return DiskCacheLoader(enabled=self.disk_cache, cache_map=self.cache_map)
|
|
|
|
|
|
def save(app: Application, icon_bytes: bytes = None, only_icon: bool = False):
|
|
|
|
if app.supports_disk_cache():
|
|
|
|
if not only_icon:
|
|
Path(app.get_disk_cache_path()).mkdir(parents=True, exist_ok=True)
|
|
data = app.get_data_to_cache()
|
|
|
|
with open(app.get_disk_data_path(), 'w+') as f:
|
|
f.write(json.dumps(data))
|
|
|
|
if icon_bytes:
|
|
Path(app.get_disk_cache_path()).mkdir(parents=True, exist_ok=True)
|
|
|
|
with open(app.get_disk_icon_path(), 'wb+') as f:
|
|
f.write(icon_bytes)
|