Python合并多張圖片成PDF

更新時(shí)間：2021年06月09日 11:46:17 作者：臨時(shí)營(yíng)地

最近需要將記的筆記整理成一個(gè)pdf進(jìn)行保存，所以就研究了一下如何利用Python代碼將拍下來(lái)的照片整個(gè)合并成一個(gè)pdf，感興趣的可以了解一下

前言

最近需要將記的筆記整理成一個(gè) pdf 進(jìn)行保存，所以就研究了一下如何利用 Python 代碼將拍下來(lái)的照片整個(gè)合并成一個(gè) pdf

過(guò)程

拿到一個(gè)需求最重要的就是將大塊任務(wù)拆分成一個(gè)個(gè)小模塊，逐個(gè)擊破。

拍照

這一步首先是將所有的書(shū)頁(yè)拍好，需要注意的是要按照書(shū)的頁(yè)碼來(lái)拍，因?yàn)楹竺娴呐判蚴前凑瘴募M(jìn)行排序的，拍照的文件名基本上是按照時(shí)間生成的，如果拍的時(shí)候亂了，到時(shí)候生成的 pdf 里面的頁(yè)碼也會(huì)亂掉。

用到的Python 操作庫(kù)

Python 最好的地方就是有大量的第三方庫(kù)能幫我們快速實(shí)現(xiàn)我們想要的方法，搜索到了兩個(gè)庫(kù)，
PyFPDF 和img2pdf，我們這里選擇img2pdf來(lái)完成我們的需求

pip install img2pdf

Python遍歷文件夾獲取圖片

    dirname = "f:/wlzcool"
    imgs = []
    for fname in os.listdir(dirname):
        if not fname.endswith(".jpg"):
            continue
        path = os.path.join(dirname, fname)
        if os.path.isdir(path):
            continue
        imgs.append(path)

需要注意圖片的文件名如果是純數(shù)字且位數(shù)不一樣，排序會(huì)為1之后是10而不是2，需要進(jìn)行一個(gè)排序，如果是手機(jī)拍的文件就沒(méi)有這個(gè)問(wèn)題。

files.sort(key=lambda x: int(x[:-4]))

旋轉(zhuǎn)圖片展示方向并壓縮像素

有的時(shí)候手機(jī)拍出來(lái)的圖片是水平的，需要將其改為豎直的
用rotate旋轉(zhuǎn)方向的時(shí)候需要注意加上expand=True 這個(gè)參數(shù)，否則會(huì)有黑邊出現(xiàn)。
手機(jī)的照片像素太高，有的需要進(jìn)行壓縮以保證最后生成的pdf的大小適中。

    img = Image.open(path)    
    if img.size[0] > img.size[1]:
        im_rotate = img.rotate(90, expand=True)
        size = (int(im_rotate.size[0] / 3), int(im_rotate.size[1] / 3))
        im_rotate = im_rotate.resize(size)
        im_rotate.save(savepath, quality=95)
    else:
        size = (int(img.size[0] / 3), int(img.size[1] / 3))
        img = img.resize(size)
        img.save(savepath, quality=95)

整體代碼

寫(xiě)成腳本需要考慮的有很多，為了方便使用，需要將各種參數(shù)改為允許用戶輸入的。比如圖片文件夾所在的路徑，壓縮比之類的

from PIL import Image
import os
import img2pdf

flag = False
while not flag:
    dirname = input("請(qǐng)輸入圖片文件夾所在路徑(例如d:/wlzcool)：")
    flag = os.path.exists(dirname)
    if not flag:
        print("圖片文件夾所在路徑不存在！")
saveflag = False
while not saveflag:
    savedirname = input("請(qǐng)輸入目標(biāo)圖片文件夾所在路徑(例如d:/wlzcool2)：")
    saveflag = os.path.exists(savedirname)
    if not saveflag:
        print("圖片文件夾所在路徑不存在！")
        automakedir = input("是否自動(dòng)創(chuàng)建對(duì)應(yīng)文件夾？(是Y/否N):")
        if automakedir.strip().upper() == "Y":
            os.makedirs(savedirname)
            saveflag = True
files = os.listdir(dirname)
reductionFactor = int(input("請(qǐng)輸入長(zhǎng)寬壓縮比(例如3):"))
if reductionFactor <= 0:
    reductionFactor = 3
isConvertBlack = input("是否輸出黑白版本？(是Y/否N):").strip().upper() == "Y"
for fname in files:
    if not fname.endswith(".jpg"):
        continue
    path = os.path.join(dirname, fname)
    savePath = os.path.join(savedirname, fname)
    if os.path.isdir(path):
        continue
    img = Image.open(path)    
    if img.size[0] > img.size[1]:
        im_rotate = img.rotate(90, expand=True)
        size = (int(im_rotate.size[0] / reductionFactor), int(im_rotate.size[1] / reductionFactor))
        im_rotate = im_rotate.resize(size)
        if isConvertBlack:
            im_rotate = im_rotate.convert("L")
        im_rotate.save(savePath, quality=95)
    else:
        size = (int(img.size[0] / reductionFactor), int(img.size[1] / reductionFactor))
        img = img.resize(size)
        if isConvertBlack:
            img = img.convert("L")
        img.save(savePath, quality=95)
filename = input("請(qǐng)輸入輸出文件名(例如：第一章)：")
with open(filename + ".pdf", "wb") as f:
    imgs = []
    files = os.listdir(savedirname)
    for fname in files:
        if not fname.endswith(".jpg"):
            continue
        path = os.path.join(savedirname, fname)
        if os.path.isdir(path):
            continue
        imgs.append(path)
    f.write(img2pdf.convert(imgs))