Spaces:
Running
Running
File size: 2,738 Bytes
4051191 |
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 |
import os
from io import BytesIO
from typing import List, BinaryIO
from pathlib import Path
from fpdf import FPDF
import re
from PIL import Image
def fld2pdf(folder: Path, out: str):
files = [file for file in folder.glob(r'*') if re.match(r'.*\.(jpg|png|jpeg|webp)', file.name)]
files.sort(key=lambda x: x.name)
pdf = folder / f'{out}.pdf'
img2pdf(files, pdf)
return pdf
def new_img(path: Path) -> Image.Image:
img = Image.open(path)
if img.mode != 'RGB':
img = img.convert('RGB')
return img
def old_img2pdf(files: List[Path], out: Path):
img_list = [new_img(img) for img in files]
img_list[0].save(out, resolution=100.0, save_all=True, append_images=img_list[1:])
for img in img_list:
img.close()
def pil_image(path: Path) -> (BytesIO, int, int):
img = new_img(path)
width, height = img.width, img.height
try:
membuf = BytesIO()
img.save(membuf, format='JPEG')
finally:
img.close()
return membuf, width, height
def unicode_to_latin1(s):
# Substitute the ' character
s = s.replace('\u2019', '\x92')
# Substitute the " character
s = s.replace('\u201d', '\x94')
# Substitute the - character
s = s.replace('\u2013', '\x96')
# Remove all other non-latin1 characters
s = s.encode('latin1', 'replace').decode('latin1')
return s
def img2pdf(files: List[Path], out: Path):
pdf = FPDF('P', 'pt')
for imageFile in files:
img_bytes, width, height = pil_image(imageFile)
pdf.add_page(format=(width, height))
pdf.image(img_bytes, 0, 0, width, height)
img_bytes.close()
pdf.set_title(unicode_to_latin1(out.stem))
pdf.output(out, "F")
def fld2thumb(folder: Path):
files = [file for file in folder.glob(r'*') if re.match(r'.*\.(jpg|png|jpeg|webp)', file.name)]
files.sort(key=lambda x: x.name)
thumb_path = make_thumb(folder, files)
return thumb_path
def make_thumb(folder, files):
aspect_ratio = 0.7
if len(files) > 1:
with Image.open(files[1]) as img:
aspect_ratio = img.width / img.height
thumbnail = Image.open(files[0]).convert('RGB')
tg_max_size = (300, 300)
thumbnail = crop_thumb(thumbnail, aspect_ratio)
thumbnail.thumbnail(tg_max_size)
thumb_path = folder / 'thumbnail' / f'thumbnail.jpg'
os.makedirs(thumb_path.parent, exist_ok=True)
thumbnail.save(thumb_path)
thumbnail.close()
return thumb_path
def crop_thumb(thumb: Image.Image, aspect_ratio):
w, h = thumb.width, thumb.height
if w * 2 <= h:
b = int(h - (w / aspect_ratio))
if b <= 0:
b = w
thumb = thumb.crop((0, 0, w, b))
return thumb
|