img2pdf_with_package.py 727 B

1234567891011121314151617181920212223
  1. import img2pdf
  2. from Image2PDF.config import SOURCE_DIR_PATH, TARGET_DIR_PATH
  3. import os
  4. def imgs_from_dir2pdf(source_dir_path, target_dir_path):
  5. # convert all files ending in .jpg inside a directory
  6. dirname = source_dir_path
  7. imgs = []
  8. for fname in os.listdir(dirname):
  9. if not fname.endswith(".jpg" or ".png"):
  10. continue
  11. path = os.path.join(dirname, fname)
  12. if os.path.isdir(path):
  13. continue
  14. imgs.append(path)
  15. for count,each_img in enumerate(imgs):
  16. with open(target_dir_path+r"\Img{}.pdf".format(count), "wb") as f:
  17. f.write(img2pdf.convert(each_img))
  18. if __name__=='__main__':
  19. imgs_from_dir2pdf(SOURCE_DIR_PATH, TARGET_DIR_PATH)