File size: 2,231 Bytes
5b14aa2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | """HTML file processor."""
import os
import logging
from typing import Dict, Any
from .base import BaseProcessor
from ..result import ConversionResult
from ..exceptions import ConversionError, FileNotFoundError
# Configure logging
logger = logging.getLogger(__name__)
class HTMLProcessor(BaseProcessor):
"""Processor for HTML files using markdownify for conversion."""
def can_process(self, file_path: str) -> bool:
"""Check if this processor can handle the given file.
Args:
file_path: Path to the file to check
Returns:
True if this processor can handle the file
"""
if not os.path.exists(file_path):
return False
# Check file extension - ensure file_path is a string
file_path_str = str(file_path)
_, ext = os.path.splitext(file_path_str.lower())
return ext in ['.html', '.htm']
def process(self, file_path: str) -> ConversionResult:
"""Process the HTML file and return a conversion result.
Args:
file_path: Path to the HTML file to process
Returns:
ConversionResult containing the processed content
Raises:
FileNotFoundError: If the file doesn't exist
ConversionError: If processing fails
"""
if not os.path.exists(file_path):
raise FileNotFoundError(f"File not found: {file_path}")
try:
try:
from markdownify import markdownify as md
except ImportError:
raise ConversionError("markdownify is required for HTML processing. Install it with: pip install markdownify")
metadata = self.get_metadata(file_path)
with open(file_path, 'r', encoding='utf-8') as f:
html_content = f.read()
content = md(html_content, heading_style="ATX")
return ConversionResult(content, metadata)
except Exception as e:
if isinstance(e, (FileNotFoundError, ConversionError)):
raise
raise ConversionError(f"Failed to process HTML file {file_path}: {str(e)}") |