rick/LEDMatrix
1
0
Fork 0
mirror of https://github.com/ChuckBuilds/LEDMatrix.git synced 2026-04-11 21:33:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix datetime serialization and improve root user cache handling

Browse Source
This commit is contained in:
ChuckBuilds
2025-04-18 19:59:38 -05:00
parent d84a3e4635
commit edfacd6ca6
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/cache_manager.py
View File

@@ -7,6 +7,12 @@ from typing import Any, Dict, Optional
import logging import logging
import stat import stat
class DateTimeEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, datetime):
return obj.isoformat()
return super().default(obj)
class CacheManager: class CacheManager:
def __init__(self, cache_dir: str = "cache"): def __init__(self, cache_dir: str = "cache"):
self.logger = logging.getLogger(__name__) self.logger = logging.getLogger(__name__)
@@ -17,8 +23,12 @@ class CacheManager:
"""Ensure the cache directory exists with proper permissions.""" """Ensure the cache directory exists with proper permissions."""
if not os.path.exists(self.cache_dir): if not os.path.exists(self.cache_dir):
try: try:
# If running as root, use /tmp by default
if os.geteuid() == 0:
self.cache_dir = os.path.join("/tmp", "ledmatrix_cache")
self.logger.info(f"Running as root, using temporary cache directory: {self.cache_dir}")
# Try to create in user's home directory if current directory fails # Try to create in user's home directory if current directory fails
if not os.access(os.getcwd(), os.W_OK): elif not os.access(os.getcwd(), os.W_OK):
home_dir = os.path.expanduser("~") home_dir = os.path.expanduser("~")
self.cache_dir = os.path.join(home_dir, ".ledmatrix_cache") self.cache_dir = os.path.join(home_dir, ".ledmatrix_cache")
self.logger.info(f"Using cache directory in home: {self.cache_dir}") self.logger.info(f"Using cache directory in home: {self.cache_dir}")
@@ -69,7 +79,7 @@ class CacheManager:
# Create a temporary file first # Create a temporary file first
temp_path = cache_path + '.tmp' temp_path = cache_path + '.tmp'
with open(temp_path, 'w') as f: with open(temp_path, 'w') as f:
json.dump(data, f) json.dump(data, f, cls=DateTimeEncoder)
# Set proper permissions (644 - rw-r--r--) # Set proper permissions (644 - rw-r--r--)
os.chmod(temp_path, 0o644) os.chmod(temp_path, 0o644)