python3图片裁剪+转换pdf+压缩

本地大量长图，要发送给别人，所以要对图片进行裁剪+转换pdf+压缩

import zipfile
import os
from concurrent.futures import ThreadPoolExecutor
from reportlab.lib.pagesizes import A4, landscape
from reportlab.pdfgen import canvas
from PIL import Image
import time

# 设置全局变量
filepath = "../jpg/cartoon/"
outzipfilepath = "../jpg/cartoonzip/"
outpdffilepath = "../jpg/cartoonpdf/"
cropimgpath = "../jpg/image/"


# 找出文件夹下所有.xml后缀的文件
def listfiles(rootdir, prefix='.xml'):
    file = []
    for parent, dirnames, filenames in os.walk(rootdir):
        if parent == rootdir:
            for filename in filenames:
                if filename.endswith(prefix):
                    file.append(filename)
            return file
        else:
            pass


# 创建文件夹
def createjia(path):
    try:
        os.makedirs(path)
    except:
        pass


# 这里是裁剪图片
# 传入的是图片的名字
def cropimg_tranpdf(imgname):
    # 新生成的文件夹的名字
    tempfilename = str(imgname.replace(".png", ""))

    # 创建保存裁剪后的图片的文件夹
    createpath = "../jpg/image/" + str(tempfilename)
    createjia(createpath)

    # 打开图片
    imgpath = filepath + imgname
    img = Image.open(imgpath)

    # 获得图片的宽高
    width = int(img.size[0])
    height = int(img.size[1])

    # A4的宽1240，高1754
    # 制作裁剪坐标
    countheight = (height // 1240)

    # 设置初始高度为1240
    newheight = 1240

    # 将要保存的pdf的位置和名字
    pdfname = str(outpdffilepath) + str(tempfilename) + ".pdf"
    # A4的大小
    # 发现A4会把长图压缩成1张A4的大小
    (w, h) = landscape(A4)
    # 保存pdf
    c = canvas.Canvas(str(pdfname), pagesize=landscape(A4))
    # number of page
    for i in range(0, countheight):
        newheight = newheight + 1240
        # 裁剪的位置
        # (起始宽的位置，起始高的位置，裁剪宽度，裁剪高度)
        # 左上角的坐标为(0,0)
        region = (0, newheight - 1240, width, newheight)
        # 裁剪
        cropImg = img.crop(region)
        # 保存
        jpgname = str(i) + ".jpg"
        # 保存裁剪后的图片
        cropImg.save(str(createpath) + "/" + str(jpgname))

        # 写入的jpg将其组合成pdf
        filepath_jpgname = str(createpath) + "/" + str(jpgname)
        c.drawImage(filepath_jpgname, 0, 0, w, h)
        c.showPage()
    c.save()
    print("完成PDF：" + str(tempfilename))


# 这里是裁剪和转化pdf的多进程
# 开启多进程
def threadingcrop_pdf(number):
    # 进程数
    pool = ThreadPoolExecutor(int(number))
    # 读取文件夹名字
    namelist = listfiles(filepath, "png")

    # 进程开跑
    for name in namelist:
        print(name)
        pool.submit(cropimg_tranpdf, name)
        # 太快电脑受不了
        time.sleep(1)



# 这里传入的是pdf的名字
# 写入压缩文件
def zipfiles(names):
    # 需要压缩到的文件目录和名字
    zipname = str(outzipfilepath) + str(names.replace(".pdf", "")) + ".zip"
    # 需要压缩的文件位置和名字
    name = outpdffilepath + names
    files = zipfile.ZipFile(zipname, 'w', zipfile.ZIP_DEFLATED)

    # 写入压缩包
    files.write(name)
    files.close()
    print("完成压缩：" + str(zipname))


# 这里是压缩zip的多进程
# 开启多进程
def threadingzip(number):
    # 先转化为pdf
    threadingcrop_pdf(number)
    # 进程数
    pool = ThreadPoolExecutor(int(number))
    # 读取文件名字
    namelist = listfiles(outpdffilepath, "pdf")
    # 进程开跑
    for name in namelist:
        print(name)
        pool.submit(zipfiles, name)
        # 太快电脑受不了
        time.sleep(1)


if __name__ == '__main__':
    # 多进程
    #number = 1
    #threadingzip(number)

    # 写入pdf
    namelistpdf = listfiles(filepath, "png")
    for name in namelistpdf:
        print(name)
        cropimg_tranpdf(name)

    # 写入zip
    namelistzip = listfiles(outpdffilepath, "pdf")
    for name in namelistzip:
        print(name)
        zipfiles(name)