Refactored App.py

This commit is contained in:
Kevin Veen-Birkenbach 2025-01-09 14:27:07 +01:00
parent c87c1df10a
commit d8ec067675
2 changed files with 77 additions and 50 deletions

View File

@ -5,54 +5,13 @@ import hashlib
import yaml
from utils.configuration_resolver import ConfigurationResolver
from pprint import pprint
from utils.cache_manager import CacheManager
# Verzeichnis mit Dateien, die gelöscht werden sollen
TEMP_DIR = "static/cache/"
# Initialize the CacheManager
cache_manager = CacheManager()
def delete_temp_files():
if os.path.exists(TEMP_DIR):
for filename in os.listdir(TEMP_DIR):
file_path = os.path.join(TEMP_DIR, filename)
if os.path.isfile(file_path):
os.remove(file_path)
print(f"Gelöscht: {file_path}")
else:
os.makedirs(TEMP_DIR) # Erstelle das Verzeichnis, falls es nicht existiert
print(f"Erstellt: {TEMP_DIR}")
# Löschen der Dateien beim App-Start
delete_temp_files()
def cache_file(file_url, cache_dir=TEMP_DIR):
"""Lädt ein Icon herunter und speichert es lokal, wenn es nicht existiert. Fügt einen Hash hinzu."""
# Erstelle das Verzeichnis, falls es nicht existiert
os.makedirs(cache_dir, exist_ok=True)
# Generiere einen 8-Zeichen-Hash basierend auf der URL
hash_object = hashlib.blake2s(file_url.encode('utf-8'), digest_size=8)
hash_suffix = hash_object.hexdigest()
splitted_file_url = file_url.split("/")
if splitted_file_url[-1] == "download":
# Erstelle den Dateinamen mit Hash
base_name = splitted_file_url[-2]
else:
base_name = splitted_file_url[-1]
filename = f"{base_name}_{hash_suffix}.png"
full_path = os.path.join(cache_dir, filename)
# Wenn die Datei existiert, überspringe den Download
if os.path.exists(full_path):
return full_path
# Lade die Datei herunter
response = requests.get(file_url, stream=True)
if response.status_code == 200:
with open(full_path, "wb") as f:
for chunk in response.iter_content(1024):
f.write(chunk)
return full_path
# Clear cache on startup
cache_manager.clear_cache()
def load_config(app):
"""Load and resolve the configuration."""
@ -80,12 +39,12 @@ def reload_config_in_dev():
else:
print("PRODUCTIVE ENVIRONMENT")
# Cachen der Icons
# Cache the icons
for card in app.config["cards"]:
card["icon"]["cache"] = cache_file(card["icon"]["source"])
card["icon"]["cache"] = cache_manager.cache_file(card["icon"]["source"])
app.config["company"]["logo"]["cache"] = cache_file(app.config["company"]["logo"]["source"])
app.config["company"]["favicon"]["cache"] = cache_file(app.config["company"]["favicon"]["source"])
app.config["company"]["logo"]["cache"] = cache_manager.cache_file(app.config["company"]["logo"]["source"])
app.config["company"]["favicon"]["cache"] = cache_manager.cache_file(app.config["company"]["favicon"]["source"])
@app.route('/')
def index():

View File

@ -0,0 +1,68 @@
import os
import hashlib
import requests
class CacheManager:
"""
A class to manage caching of files, including creating temporary directories
and caching files locally with hashed filenames.
"""
def __init__(self, cache_dir="static/cache"):
"""
Initialize the CacheManager with a cache directory.
:param cache_dir: The directory where cached files will be stored.
"""
self.cache_dir = cache_dir
self._ensure_cache_dir_exists()
def _ensure_cache_dir_exists(self):
"""
Ensure the cache directory exists. If it doesn't, create it.
"""
if not os.path.exists(self.cache_dir):
os.makedirs(self.cache_dir)
print(f"Created cache directory: {self.cache_dir}")
def clear_cache(self):
"""
Clear all files in the cache directory.
"""
if os.path.exists(self.cache_dir):
for filename in os.listdir(self.cache_dir):
file_path = os.path.join(self.cache_dir, filename)
if os.path.isfile(file_path):
os.remove(file_path)
print(f"Deleted: {file_path}")
def cache_file(self, file_url):
"""
Download a file and store it locally in the cache directory with a hashed filename.
:param file_url: The URL of the file to cache.
:return: The local path of the cached file.
"""
# Generate a hashed filename based on the URL
hash_object = hashlib.blake2s(file_url.encode('utf-8'), digest_size=8)
hash_suffix = hash_object.hexdigest()
# Determine the base name for the file
splitted_file_url = file_url.split("/")
base_name = splitted_file_url[-2] if splitted_file_url[-1] == "download" else splitted_file_url[-1]
# Construct the full path for the cached file
filename = f"{base_name}_{hash_suffix}.png"
full_path = os.path.join(self.cache_dir, filename)
# If the file already exists, return the cached path
if os.path.exists(full_path):
return full_path
# Download the file and save it locally
response = requests.get(file_url, stream=True)
if response.status_code == 200:
with open(full_path, "wb") as file:
for chunk in response.iter_content(1024):
file.write(chunk)
return full_path