File size: 4,156 Bytes
2b3b710 |
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 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 |
"""
Images to PDF Converter Module
This module provides functionality to convert multiple images into a single PDF file.
"""
import os
from PIL import Image
from pathlib import Path
from custom_logger import logger_config
class ImagesToPDF:
"""
Convert multiple images into a single PDF file.
"""
def __init__(self):
self.current_path = os.path.abspath(os.path.join(os.path.dirname(__file__)))
self.input_dir = os.path.join(self.current_path, "input")
self.output_dir = os.path.join(self.current_path, "output")
self.supported_formats = ['jpg', 'jpeg', 'png', 'webp', 'bmp', 'gif', 'tiff', 'tif']
# Create directories if they don't exist
os.makedirs(self.input_dir, exist_ok=True)
os.makedirs(self.output_dir, exist_ok=True)
def _validate_image(self, filename: str) -> bool:
"""Check if file is a supported image format."""
ext = Path(filename).suffix.lower().lstrip('.')
return ext in self.supported_formats
def convert_to_pdf(self, image_ids: list, output_name: str = "output.pdf") -> str:
"""
Convert multiple images to a single PDF.
Args:
image_ids: List of image file IDs in the input directory
output_name: Name for the output PDF file
Returns:
Path to the generated PDF file, or None on failure
"""
try:
if not image_ids:
raise ValueError("No images provided for conversion")
images = []
first_image = None
for img_id in image_ids:
img_path = os.path.join(self.input_dir, img_id)
if not os.path.exists(img_path):
logger_config.warning(f"Image not found: {img_path}")
continue
# Open and convert image to RGB (PDF requires RGB)
img = Image.open(img_path)
# Handle transparency by converting to RGB with white background
if img.mode in ('RGBA', 'LA', 'P'):
# Create white background
background = Image.new('RGB', img.size, (255, 255, 255))
if img.mode == 'P':
img = img.convert('RGBA')
background.paste(img, mask=img.split()[-1] if img.mode == 'RGBA' else None)
img = background
elif img.mode != 'RGB':
img = img.convert('RGB')
if first_image is None:
first_image = img
else:
images.append(img)
if first_image is None:
raise ValueError("No valid images found for conversion")
# Generate output path
output_path = os.path.join(self.output_dir, output_name)
# Save as PDF
if images:
first_image.save(
output_path,
save_all=True,
append_images=images,
resolution=100.0
)
else:
first_image.save(output_path, resolution=100.0)
logger_config.success(f"PDF created successfully: {output_path}")
# Cleanup input files
for img_id in image_ids:
img_path = os.path.join(self.input_dir, img_id)
if os.path.exists(img_path):
os.unlink(img_path)
return output_path
except Exception as e:
logger_config.error(f"PDF conversion failed: {str(e)}")
return None
def upload(self, file_id: str, file_content: bytes):
"""Save uploaded file to input directory."""
file_path = os.path.join(self.input_dir, file_id)
with open(file_path, 'wb') as f:
f.write(file_content)
logger_config.info(f"File uploaded: {file_path}")
|