python3 将图片转换至PDF，批量转换漫画文档等

  • A+
所属分类：python基础入门

将图片转PDF，适用于图片批量转换。如漫画。图片文档等。

自动对图片大小进行处理，适用与kindle、iphone、ipad等终端。

支持多重文件路径检索进行转换。

支持jpg、png、bmp

转载源码：

  1. # -*- coding: utf-8 -*-
  2. # -------------------------------------------------------------------------------
  3. # @File     : Convert2PDF.py
  4. # @Created  : 2017/12/25 下午3:03
  5. # @Software : PyCharm
  6. # 
  7. # @Author   : Liu.Qi
  8. # @Contact  : liuqi_0725#aliyun.com
  9. # -------------------------------------------------------------------------------
  11. import os
  13. from reportlab.platypus import SimpleDocTemplate,Image, PageBreak
  14. from reportlab.lib.pagesizes import A4, landscape
  16. import time
  18. from PIL import Image as pilImage
  20. # 支持的图片类型
  21. __allow_type = [".jpg", ".jpeg", ".bmp", ".png"]
  23. __rootDir = ""
  25. def convert_images2PDF_one_dir(file_dir,save_name=None ,filename_sort_fn=None):
  26.     '''
  27.     转换一个目录文件夹下的图片至 PDF
  28.     :param file_dir:
  29.     :param file_name: 如果为空,则以当前文件夹的名称命名, 必须是.pdf结尾
  30.     :param filename_sort_fn:
  31.     文件名排序的回调函数,当此回调函数有值时,在文件名排序时,会回调,并将 file 的完整路径返回。
  32.     回调函数需要返回一个可转换整形的内容,函数根据此回调函数的返回值,对文件名排序
  33.     比如:
  34.         现实中,文件名会是
  35.         test_01_doc_0.png、
  36.         test_01_doc_1.png、
  37.         test_01_doc_2.png、
  38.         test_01_doc_3.png、
  39.         test_01_doc_11.png、
  40.         test_01_doc_21.png
  41.         等等,我们也希望读取出来的顺序如此,但是 mac、win 下,包括sort 排序出来的结果都不理想。
  42.         结果为
  43.         test_01_doc_0.png、
  44.         test_01_doc_1.png、
  45.         test_01_doc_11.png、
  46.         test_01_doc_2.png、
  47.         test_01_doc_21.png、
  48.         test_01_doc_3.png
  49.         不是我们想要得到的。
  50.         通过 filename_sort_fn(filename) 返回的整形数字,对齐正确的排序
  51.     '''
  52.     book_pages = []
  54.     for parent, dirnames ,filenames in os.walk(file_dir):
  56.         # 只遍历最顶层
  57.         if parent != file_dir :
  58.             continue
  60.         # 过滤文件中所有的图片
  61.         for file_name in filenames:
  62.             file_path = os.path.join(parent, file_name)
  63.             # 是否图片
  64.             if __isAllow_file(file_path) :
  65.                 book_pages.append(file_path)
  67.         # 取当前目录的文件名为书名
  68.         if save_name == None :
  69.             save_name = os.path.join(file_dir,(os.path.basename(file_dir) + ".pdf"))
  70.         else :
  71.             save_name = os.path.join(file_dir,save_name)
  73.         if len(book_pages) > 0 :
  74.             # 开始转换
  75.             print("[*][转换PDF] : 开始. [保存路径] > [%s]" % (save_name))
  76.             beginTime = time.clock()
  77.             __converted(save_name , book_pages ,filename_sort_fn)
  78.             endTime = time.clock()
  79.             print("[*][转换PDF] : 结束. [保存路径] > [%s] , 耗时 %f s " % (save_name, (endTime - beginTime)))
  80.         else :
  81.             print("该目录下没有找到任何图片文件.如果是多重目录,尝试使用 convert_images2PDF_more_dirs 函数")
  84. def convert_images2PDF_more_dirs(dirPath):
  85.     """
  86.     转换一个目录文件夹下的图片至 PDF
  87.     :param file_dir:
  88.     :param filename_sort_fn:
  89.     """
  91.     # 已经找到目录
  92.     __dirs = {}
  94.     for parent, dirnames, filenames in os.walk(dirPath):
  96.         for dirname in dirnames:
  97.             # 假设每个文件夹下都有图片，都是一本书
  98.             dirData = {"name": "", "pages": [], "isBook": False}
  99.             dirName = dirname.split('/')[0]
  100.             dirData['name'] = dirName
  101.             __dirs[dirName] = dirData
  103.         # 查找有无图片
  104.         for filename in filenames:
  106.             real_filename = os.path.join(parent, filename)
  107.             # 取父文件夹名称为书名
  108.             parentDirName = real_filename.split('/')[-2]
  110.             if parentDirName in __dirs.keys():
  111.                 dirJsonData = __dirs[parentDirName]
  112.             else:
  113.                 continue
  115.             # 检查是否图片
  116.             if __isAllow_file(real_filename) :
  117.                 # 将图片添加至书本
  118.                 dirJsonData['pages'].append(real_filename)
  120.                 # 如果该书的isbook 是false 改为true
  121.                 if not dirJsonData['isBook']:
  122.                     dirJsonData['isBook'] = True
  124.     index = 1
  125.     for dirName in __dirs.keys():
  127.         dirData = __dirs[dirName]
  129.         if dirData['isBook']:
  130.             print("[*][转换PDF] : 开始. [名称] > [%s]" % (dirName))
  131.             beginTime = time.clock()
  132.             __converted(os.path.join(dirPath,(dirData['name'] + ".pdf")) , dirData['pages'])
  133.             endTime = time.clock()
  134.             print("[*][转换PDF] : 结束. [名称] > [%s] , 耗时 %f s " % (dirName, (endTime - beginTime)))
  135.             index += 1
  137.     print("[*][所有转换完成] : 本次转换检索目录数 %d 个，共转换的PDF %d 本 " % (len(__dirs), index - 1))
  140. def __isAllow_file(filepath):
  141.     """
  142.     是否允许的文件
  143.     :param file:
  144.     :return:
  145.     """
  146.     if filepath and (os.path.splitext(filepath)[1] in __allow_type):
  147.         return True
  149.     return False
  151. def __converted(save_book_name,book_pages=[],filename_sort_fn=None):
  152.     """
  153.     开始转换
  154.     :param book_name: 保存的文件名(包含路径)
  155.     :param book_pages: 图片数组
  156.     :param filename_sort_fn: 文件名排序规则
  157.     :return:
  158.     """
  160.     # A4 纸的宽高
  161.     __a4_w, __a4_h = landscape(A4)
  163.     # 对数据进行排序
  164.     if (filename_sort_fn == None):
  165.         # 将按图片按名称的ASCII排序以避免错乱问题
  166.         book_pages.sort()
  167.     else:
  168.         # lambda 匿名函数, 第一个参数定义函数入参,第二个为参数的处理表达式
  169.         # 这样理解
  170.         # def getName(name):
  171.         #   return name.split("-")[1]
  172.         #
  173.         # 用匿名函数就这么表示
  174.         # name = lambda name:name.split("-")[1]
  175.         # xname = name("haha-dd-23")
  176.         #
  177.         # 支持多个参数
  178.         #
  179.         book_pages = sorted(book_pages, key=lambda name: int(filename_sort_fn(name)))
  181.     bookPagesData = []
  183.     bookDoc = SimpleDocTemplate(save_book_name, pagesize=A4, rightMargin=72, leftMargin=72, topMargin=72, bottomMargin=18)
  185.     for page in book_pages:
  187.         img_w, img_h = ImageTools().getImageSize(page)
  189.         # img_w = img.imageWidth
  190.         # img_h = img.imageHeight
  193.         if __a4_w / img_w < __a4_h / img_h:
  194.             ratio = __a4_w / img_w
  195.         else:
  196.             ratio = __a4_h / img_h
  198.         data = Image(page, img_w * ratio, img_h * ratio)
  199.         bookPagesData.append(data)
  200.         bookPagesData.append(PageBreak())
  202.     try:
  203.         bookDoc.build(bookPagesData)
  204.         # print("已转换 >>>> " + bookName)
  205.     except Exception as err:
  206.         print("[*][转换PDF] : 错误. [名称] > [%s]" % (save_book_name))
  207.         print("[*] Exception >>>> ", err)
  210. class ImageTools:
  211.     def getImageSize(self, imagePath):
  212.         img = pilImage.open(imagePath)
  213.         return img.size
weinxin
我的微信公众号
爱真理，得永生！　　　　　　　　　　爱在灵灵久博客，网罗天下，福利大家！

发表评论

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: