| |
| """ |
| File handling utilities for the DOCX to PDF converter |
| """ |
|
|
| import os |
| import tempfile |
| import shutil |
| import logging |
| from pathlib import Path |
| from typing import Optional |
|
|
| logger = logging.getLogger(__name__) |
|
|
| class FileHandler: |
| """Handle file operations for the converter""" |
| |
| def __init__(self, base_temp_dir: str = "/tmp/conversions"): |
| self.base_temp_dir = base_temp_dir |
| os.makedirs(self.base_temp_dir, exist_ok=True) |
| |
| def create_temp_directory(self) -> str: |
| """Create a temporary directory for file processing""" |
| try: |
| temp_dir = tempfile.mkdtemp(dir=self.base_temp_dir) |
| logger.info(f"Created temporary directory: {temp_dir}") |
| return temp_dir |
| except Exception as e: |
| logger.error(f"Failed to create temporary directory: {e}") |
| raise |
| |
| def save_uploaded_file(self, temp_dir: str, filename: str, content: bytes) -> str: |
| """Save uploaded file to temporary directory""" |
| try: |
| file_path = os.path.join(temp_dir, filename) |
| with open(file_path, "wb") as f: |
| f.write(content) |
| logger.info(f"Saved file: {file_path}") |
| return file_path |
| except Exception as e: |
| logger.error(f"Failed to save file {filename}: {e}") |
| raise |
| |
| def cleanup_temp_directory(self, temp_dir: str): |
| """Clean up temporary directory""" |
| try: |
| if os.path.exists(temp_dir): |
| shutil.rmtree(temp_dir) |
| logger.info(f"Cleaned up temporary directory: {temp_dir}") |
| except Exception as e: |
| logger.error(f"Failed to cleanup directory {temp_dir}: {e}") |
| |
| def get_file_size(self, file_path: str) -> int: |
| """Get file size in bytes""" |
| try: |
| return os.path.getsize(file_path) |
| except Exception as e: |
| logger.error(f"Failed to get file size for {file_path}: {e}") |
| return 0 |
| |
| def validate_file_extension(self, filename: str, allowed_extensions: list) -> bool: |
| """Validate file extension""" |
| try: |
| ext = Path(filename).suffix.lower() |
| return ext in allowed_extensions |
| except Exception as e: |
| logger.error(f"Failed to validate file extension for {filename}: {e}") |
| return False |
