jpg png bmp图片批量转换为pdf格式文件python脚本工具

  • A+
所属分类：编程语言

转存源码：图片批量转换 图片转pdf格式文件，python版支持jpg转pdf，png转pdf以及bmp转pdf

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

发表评论

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