TonieToolbox 0.2.3__py3-none-any.whl → 0.4.0__py3-none-any.whl

TonieToolbox/constants.py

@@ -9,7 +9,82 @@ TOO_MANY_SEGMENTS = -4
 TIMESTAMP_DEDUCT = 0x50000000
 OPUS_TAGS = [
     bytearray(
-        b"\x4F\x70\x75\x73\x54\x61\x67\x73\x0D\x00\x00\x00\x4C\x61\x76\x66\x35\x38\x2E\x32\x30\x2E\x31\x30\x30\x03\x00\x00\x00\x26\x00\x00\x00\x65\x6E\x63\x6F\x64\x65\x72\x3D\x6F\x70\x75\x73\x65\x6E\x63\x20\x66\x72\x6F\x6D\x20\x6F\x70\x75\x73\x2D\x74\x6F\x6F\x6C\x73\x20\x30\x2E\x31\x2E\x31\x30\x2A\x00\x00\x00\x65\x6E\x63\x6F\x64\x65\x72\x5F\x6F\x70\x74\x69\x6F\x6E\x73\x3D\x2D\x2D\x71\x75\x69\x65\x74\x20\x2D\x2D\x62\x69\x74\x72\x61\x74\x65\x20\x39\x36\x20\x2D\x2D\x76\x62\x72\x3B\x01\x00\x00\x70\x61\x64\x3D\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30"),
+        b"\x4F\x70\x75\x73\x54\x61\x67\x73\x0D\x00\x00\x00\x4C\x61\x76\x66\x35\x38\x2E\x32\x30\x2E\x31\x30\x30\x03\x00\x00\x00\x26\x00\x00\x00\x65\x6E\x63\x6F\x64\x65\x72\x3D\x6F\x70\x75\x73\x65\x6E\x63\x20\x66\x72\x6F\x6D\x20\x6F\x70\x75\x73\x2D\x74\x6F\x6F\x6C\x73\x20\x30\x2E\x31\x2E\x31\x30\x2A\x00\x00\x00\x65\x6E\x63\x6F\x64\x65\x72\x5F\x6F\x70\x74\x69\x6F\x6E\x73\x3D\x2D\x2D\x71\x75\x69\x65\x74\x20\x2D\x2D\x62\x69\x74\x72\x61\x74\x65\x20\x39\x36\x20\x2D\x2D\x76\x62\x72\x3B\x01\x00\x00\x70\x61\x64\x3D\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30"),
     bytearray(
         b"\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30\x30")
-]
+]
+
+# Mapping of language tags to ISO codes
+LANGUAGE_MAPPING = {
+    # Common language names to ISO codes
+    'deutsch': 'de-de',
+    'german': 'de-de',
+    'english': 'en-us',
+    'englisch': 'en-us',
+    'français': 'fr-fr',
+    'french': 'fr-fr',
+    'franzosisch': 'fr-fr',
+    'italiano': 'it-it',
+    'italian': 'it-it',
+    'italienisch': 'it-it',
+    'español': 'es-es',
+    'spanish': 'es-es',
+    'spanisch': 'es-es',
+    # Two-letter codes
+    'de': 'de-de',
+    'en': 'en-us',
+    'fr': 'fr-fr',
+    'it': 'it-it',
+    'es': 'es-es',
+}
+
+# Mapping of genre tags to tonie categories
+GENRE_MAPPING = {
+    # Standard Tonie category names from tonies.json
+    'hörspiel': 'Hörspiele & Hörbücher',
+    'hörbuch': 'Hörspiele & Hörbücher',
+    'hörbücher': 'Hörspiele & Hörbücher',
+    'hörspiele': 'Hörspiele & Hörbücher',
+    'audiobook': 'Hörspiele & Hörbücher',
+    'audio book': 'Hörspiele & Hörbücher',
+    'audio play': 'Hörspiele & Hörbücher',
+    'audio-play': 'Hörspiele & Hörbücher',
+    'audiospiel': 'Hörspiele & Hörbücher',
+    'geschichte': 'Hörspiele & Hörbücher',
+    'geschichten': 'Hörspiele & Hörbücher',
+    'erzählung': 'Hörspiele & Hörbücher',
+    
+    # Music related genres
+    'musik': 'music',
+    'lieder': 'music',
+    'songs': 'music',
+    'music': 'music',
+    'lied': 'music',
+    'song': 'music',
+    
+    # More specific categories
+    'kinder': 'Hörspiele & Hörbücher',
+    'children': 'Hörspiele & Hörbücher',
+    'märchen': 'Hörspiele & Hörbücher',
+    'fairy tale': 'Hörspiele & Hörbücher',
+    'märche': 'Hörspiele & Hörbücher',
+    
+    'wissen': 'Wissen & Hörmagazine',
+    'knowledge': 'Wissen & Hörmagazine',
+    'sachbuch': 'Wissen & Hörmagazine',
+    'learning': 'Wissen & Hörmagazine',
+    'educational': 'Wissen & Hörmagazine',
+    'bildung': 'Wissen & Hörmagazine',
+    'information': 'Wissen & Hörmagazine',
+    
+    'schlaf': 'Schlaflieder & Entspannung',
+    'sleep': 'Schlaflieder & Entspannung',
+    'meditation': 'Schlaflieder & Entspannung',
+    'entspannung': 'Schlaflieder & Entspannung',
+    'relaxation': 'Schlaflieder & Entspannung',
+    'schlaflied': 'Schlaflieder & Entspannung',
+    'einschlafhilfe': 'Schlaflieder & Entspannung',
+    
+    # Default to standard format for custom
+    'custom': 'Hörspiele & Hörbücher',
+}

TonieToolbox/dependency_manager.py

@@ -466,7 +466,7 @@ def ensure_dependency(dependency_name, auto_download=False):
     Returns:
         str: Path to the binary if available, None otherwise
     """
-    logger.info("Ensuring dependency: %s", dependency_name)
+    logger.debug("Ensuring dependency: %s", dependency_name)
     system = get_system()
     
     if system not in ['windows', 'linux', 'darwin']:
@@ -496,7 +496,7 @@ def ensure_dependency(dependency_name, auto_download=False):
             existing_binary = find_binary_in_extracted_dir(dependency_dir, binary_path)
             if existing_binary and os.path.exists(existing_binary):
                 # Verify that the binary works
-                logger.info("Found previously downloaded %s: %s", dependency_name, existing_binary)
+                logger.debug("Found previously downloaded %s: %s", dependency_name, existing_binary)
                 try:
                     if os.access(existing_binary, os.X_OK) or system == 'windows':
                         if system in ['linux', 'darwin']:
@@ -509,14 +509,14 @@ def ensure_dependency(dependency_name, auto_download=False):
                             try:
                                 result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, timeout=5)
                                 if result.returncode == 0:
-                                    logger.info("Using previously downloaded %s: %s", dependency_name, existing_binary)
+                                    logger.debug("Using previously downloaded %s: %s", dependency_name, existing_binary)
                                     return existing_binary
                             except:
                                 # If --version fails, try without arguments
                                 try:
                                     result = subprocess.run([existing_binary], stdout=subprocess.PIPE, stderr=subprocess.PIPE, timeout=5)
                                     if result.returncode == 0:
-                                        logger.info("Using previously downloaded %s: %s", dependency_name, existing_binary)
+                                        logger.debug("Using previously downloaded %s: %s", dependency_name, existing_binary)
                                         return existing_binary
                                 except:
                                     pass
@@ -525,7 +525,7 @@ def ensure_dependency(dependency_name, auto_download=False):
                             try:
                                 result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, timeout=5)
                                 if result.returncode == 0:
-                                    logger.info("Using previously downloaded %s: %s", dependency_name, existing_binary)
+                                    logger.debug("Using previously downloaded %s: %s", dependency_name, existing_binary)
                                     return existing_binary
                             except:
                                 pass

TonieToolbox/logger.py

@@ -5,6 +5,8 @@ Logging configuration for the TonieToolbox package.
 import logging
 import os
 import sys
+from datetime import datetime
+from pathlib import Path
 
 # Define log levels and their names
 TRACE = 5  # Custom level for ultra-verbose debugging
@@ -19,20 +21,39 @@ def trace(self, message, *args, **kwargs):
 # Add trace method to the Logger class
 logging.Logger.trace = trace
 
-def setup_logging(level=logging.INFO):
+def get_log_file_path():
+    """
+    Get the path to the log file in the .tonietoolbox folder with timestamp.
+    
+    Returns:
+        Path: Path to the log file
+    """
+    # Create .tonietoolbox folder in user's home directory if it doesn't exist
+    log_dir = Path.home() / '.tonietoolbox'
+    log_dir.mkdir(exist_ok=True)
+    
+    # Create timestamp string for the filename
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    
+    # Define log file path with timestamp
+    log_file = log_dir / f'tonietoolbox_{timestamp}.log'
+    
+    return log_file
+
+def setup_logging(level=logging.INFO, log_to_file=False):
     """
     Set up logging configuration for the entire application.
     
     Args:
         level: Logging level (default: logging.INFO)
+        log_to_file: Whether to log to a file (default: False)
         
     Returns:
         logging.Logger: Root logger instance
     """
-    # Configure root logger
-    logging.basicConfig(
-        level=level,
-        format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    # Create formatter
+    formatter = logging.Formatter(
+        '%(asctime)s - %(name)s - %(levelname)s - %(message)s',
         datefmt='%Y-%m-%d %H:%M:%S'
     )
     
@@ -40,6 +61,31 @@ def setup_logging(level=logging.INFO):
     root_logger = logging.getLogger('TonieToolbox')
     root_logger.setLevel(level)
     
+    # Remove any existing handlers to avoid duplicate logs
+    for handler in root_logger.handlers[:]:
+        root_logger.removeHandler(handler)
+    
+    # Console handler
+    console_handler = logging.StreamHandler(sys.stdout)
+    console_handler.setFormatter(formatter)
+    console_handler.setLevel(level)
+    root_logger.addHandler(console_handler)
+    
+    # File handler (if enabled)
+    if log_to_file:
+        try:
+            log_file = get_log_file_path()
+            file_handler = logging.FileHandler(
+                log_file,
+                encoding='utf-8'
+            )
+            file_handler.setFormatter(formatter)
+            file_handler.setLevel(level)
+            root_logger.addHandler(file_handler)
+            root_logger.info(f"Log file created at: {log_file}")
+        except Exception as e:
+            root_logger.error(f"Failed to set up file logging: {e}")
+    
     return root_logger
 
 def get_logger(name):

TonieToolbox/media_tags.py

@@ -8,6 +8,9 @@ which can be used to enhance Tonie file creation with proper track information.
 import os
 from typing import Dict, Any, Optional, List
 import logging
+import tempfile
+import base64
+from mutagen.flac import Picture
 from .logger import get_logger
 from .dependency_manager import is_mutagen_available, ensure_mutagen
 
@@ -206,12 +209,28 @@ def get_file_tags(file_path: str) -> Dict[str, Any]:
         # Process different file types
         if isinstance(audio, ID3) or hasattr(audio, 'ID3'):
             # MP3 files
-            id3 = audio if isinstance(audio, ID3) else audio.ID3
-            for tag_key, tag_value in id3.items():
-                tag_name = tag_key.split(':')[0]  # Handle ID3 tags with colons
-                if tag_name in TAG_MAPPING:
-                    tag_value_str = str(tag_value)
-                    tags[TAG_MAPPING[tag_name]] = normalize_tag_value(tag_value_str)
+            try:
+                id3 = audio if isinstance(audio, ID3) else audio.ID3
+                for tag_key, tag_value in id3.items():
+                    tag_name = tag_key.split(':')[0]  # Handle ID3 tags with colons
+                    if tag_name in TAG_MAPPING:
+                        tag_value_str = str(tag_value)
+                        tags[TAG_MAPPING[tag_name]] = normalize_tag_value(tag_value_str)
+            except (AttributeError, TypeError) as e:
+                logger.debug("Error accessing ID3 tags: %s", e)
+                # Try alternative approach for ID3 tags
+                try:
+                    if hasattr(audio, 'tags') and audio.tags:
+                        for tag_key in audio.tags.keys():
+                            if tag_key in TAG_MAPPING:
+                                tag_value = audio.tags[tag_key]
+                                if hasattr(tag_value, 'text'):
+                                    tag_value_str = str(tag_value.text[0]) if tag_value.text else ''
+                                else:
+                                    tag_value_str = str(tag_value)
+                                tags[TAG_MAPPING[tag_key]] = normalize_tag_value(tag_value_str)
+                except Exception as e:
+                    logger.debug("Alternative ID3 tag reading failed: %s", e)
         elif isinstance(audio, (FLAC, OggOpus, OggVorbis)):
             # FLAC and OGG files
             for tag_key, tag_values in audio.items():
@@ -468,4 +487,167 @@ def format_metadata_filename(metadata: Dict[str, str], template: str = "{tracknu
         return result
     except Exception as e:
         logger.error("Error formatting metadata: %s", str(e))
-        return ""
+        return ""
+
+def extract_artwork(file_path: str, output_path: Optional[str] = None) -> Optional[str]:
+    """
+    Extract artwork from an audio file.
+    
+    Args:
+        file_path: Path to the audio file
+        output_path: Path where to save the extracted artwork.
+                     If None, a temporary file will be created.
+    
+    Returns:
+        Path to the extracted artwork file, or None if no artwork was found
+    """
+    if not MUTAGEN_AVAILABLE:
+        logger.debug("Mutagen not available - cannot extract artwork")
+        return None
+        
+    if not os.path.exists(file_path):
+        logger.error("File not found: %s", file_path)
+        return None
+    
+    try:
+        file_ext = os.path.splitext(file_path.lower())[1]
+        artwork_data = None
+        mime_type = None
+        
+        # Extract artwork based on file type
+        if file_ext == '.mp3':
+            audio = mutagen.File(file_path)
+            
+            # Try to get artwork from APIC frames
+            if audio.tags:
+                for frame in audio.tags.values():
+                    if frame.FrameID == 'APIC':
+                        artwork_data = frame.data
+                        mime_type = frame.mime
+                        break
+                        
+        elif file_ext == '.flac':
+            audio = FLAC(file_path)
+            
+            # Get pictures from FLAC
+            if audio.pictures:
+                artwork_data = audio.pictures[0].data
+                mime_type = audio.pictures[0].mime
+                
+        elif file_ext in ['.m4a', '.mp4', '.aac']:
+            audio = MP4(file_path)
+            
+            # Check 'covr' atom
+            if 'covr' in audio:
+                artwork_data = audio['covr'][0]
+                # Determine mime type based on data format
+                if isinstance(artwork_data, mutagen.mp4.MP4Cover):
+                    if artwork_data.format == mutagen.mp4.MP4Cover.FORMAT_JPEG:
+                        mime_type = 'image/jpeg'
+                    elif artwork_data.format == mutagen.mp4.MP4Cover.FORMAT_PNG:
+                        mime_type = 'image/png'
+                    else:
+                        mime_type = 'image/jpeg'  # Default guess
+                    
+        elif file_ext == '.ogg':
+            try:
+                audio = OggVorbis(file_path)
+            except:
+                try:
+                    audio = OggOpus(file_path)
+                except:
+                    logger.debug("Could not determine OGG type for %s", file_path)
+                    return None
+            
+            # For OGG files, metadata pictures are more complex to extract
+            if 'metadata_block_picture' in audio:
+                picture_data = base64.b64decode(audio['metadata_block_picture'][0])
+                flac_picture = Picture(data=picture_data)
+                artwork_data = flac_picture.data
+                mime_type = flac_picture.mime
+                
+        # If we found artwork data, save it to a file
+        if artwork_data:
+            # Determine file extension from mime type
+            if mime_type == 'image/jpeg':
+                ext = '.jpg'
+            elif mime_type == 'image/png':
+                ext = '.png'
+            else:
+                ext = '.jpg'  # Default to jpg
+                
+            # Create output path if not provided
+            if not output_path:
+                # Create a temporary file
+                temp_file = tempfile.NamedTemporaryFile(suffix=ext, delete=False)
+                output_path = temp_file.name
+                temp_file.close()
+            elif not os.path.splitext(output_path)[1]:
+                # Add extension if not in the output path
+                output_path += ext
+                
+            # Write artwork to file
+            with open(output_path, 'wb') as f:
+                f.write(artwork_data)
+                
+            logger.info("Extracted artwork saved to %s", output_path)
+            return output_path
+        else:
+            logger.debug("No artwork found in file: %s", file_path)
+            return None
+            
+    except Exception as e:
+        logger.debug("Error extracting artwork: %s", e)
+        return None
+
+def find_cover_image(source_dir):
+    """
+    Find a cover image in the source directory.
+    
+    Args:
+        source_dir: Path to the directory to search for cover images
+        
+    Returns:
+        str: Path to the found cover image, or None if not found
+    """
+    if not os.path.isdir(source_dir):
+        return None
+        
+    # Common cover image file names
+    cover_names = [
+        'cover', 'folder', 'album', 'front', 'artwork', 'image', 
+        'albumart', 'albumartwork', 'booklet'
+    ]
+    
+    # Common image extensions
+    image_extensions = ['.jpg', '.jpeg', '.png', '.bmp', '.gif']
+    
+    # Try different variations
+    for name in cover_names:
+        for ext in image_extensions:
+            # Try exact name match
+            cover_path = os.path.join(source_dir, name + ext)
+            if os.path.exists(cover_path):
+                logger.debug("Found cover image: %s", cover_path)
+                return cover_path
+                
+            # Try case-insensitive match
+            for file in os.listdir(source_dir):
+                if file.lower() == (name + ext).lower():
+                    cover_path = os.path.join(source_dir, file)
+                    logger.debug("Found cover image: %s", cover_path)
+                    return cover_path
+    
+    # If no exact matches, try finding any file containing the cover names
+    for file in os.listdir(source_dir):
+        file_lower = file.lower()
+        file_ext = os.path.splitext(file_lower)[1]
+        if file_ext in image_extensions:
+            for name in cover_names:
+                if name in file_lower:
+                    cover_path = os.path.join(source_dir, file)
+                    logger.debug("Found cover image: %s", cover_path)
+                    return cover_path
+    
+    logger.debug("No cover image found in directory: %s", source_dir)
+    return None