# -*- coding: utf-8 -*-
"""Convert file format.
"""
import os
import re
import PIL.Image
import PyPDF2
from .core import Common
from .exceptions import InvalidImageFileFormat
from .utils.logging import get_logger
logger = get_logger("image2pdf")
[docs]class Image2PDF(Common):
"""Make pdf file for using e-books.
"""
def __init__(self, digits, extension, directory_path=None):
"""Constructor
Args:
digits (str): Regex target digit.
extension (str): Target file extension.
directory_path (str): Target directory path.
"""
super().__init__()
self.__digits = digits
self.__extension = self._convert_extension_with_dot(extension)
self.__regex_ext = re.compile(self.__extension)
self.__file_writer = PyPDF2.PdfFileWriter()
if directory_path is not None:
self.__directory_path = directory_path
else:
self.__directory_path = os.getcwd()
logger.debug("Current Directory: {cwd}".format(cwd=self.__directory_path))
os.chdir(self.__directory_path)
[docs] def make_pdf(self, filename, remove_flag=False):
"""Make pdf file take in some image files.
Make pdf file which you use e-books by take in some image
files such as jpeg, png and gif.
Args:
filename (str): pdf file name
remove_flag (bool): If true, original image file is deleted
Returns:
bool: If success, return true.
"""
if self.__extension != ".jpg" and self.__extension != ".png" and self.__extension != ".gif":
raise InvalidImageFileFormat()
files = os.listdir(self.__directory_path)
files.sort()
logger.debug("files: {files}".format(files=files))
page_count = 0
remove_files = []
for file in files:
num = self._check_serial_number(file, self.__digits)
if not num:
logger.debug("Skip(No number): {filename}".format(filename=file))
elif not self.__regex_ext.search(file):
logger.debug("Skip(No target extension): {filename}".format(filename=file))
else:
logger.debug(file)
pdf_file = self._convert_image_to_pdf(file)
result_merge_pdf = self._merge_pdf_file(pdf_file, filename)
if result_merge_pdf:
logger.info("Success write pdf for {page} page.".format(page=page_count + 1))
page_count += 1
if remove_flag:
remove_files.append(file)
logger.info("-" * 55)
if page_count == 0:
logger.warn("Target file doesn't exist... Finish.")
return False
logger.info("All image file are converted. Filename: {filename}".format(filename=filename))
if self._remove_file_bulk(remove_files):
logger.info("Post possess is finished")
return True
def _convert_image_to_pdf(self, file, resolution=100.0):
"""Convert Image file to pdf file format.
Args:
file (str): Image file
resolution (float): Pdf file resolution, default 100.
Returns:
str: Convert pdf file name.
"""
image = PIL.Image.open(file).convert("RGB")
pdf_file_name = file.replace(self.__extension, ".pdf")
image.save(pdf_file_name, "PDF", resolution=resolution)
return pdf_file_name
def _merge_pdf_file(self, pdf_file, filename):
"""Marge pdf files
Args:
pdf_file (str): 1 page pdf file
filename (str): Merge target pdf file name
Returns:
bool: If success, return true.
"""
logger.debug(type(pdf_file))
with open(pdf_file, "rb") as f:
file_reader = PyPDF2.PdfFileReader(f)
self.__file_writer.addPage(file_reader.getPage(0))
logger.debug("Merge {pdf_file}".format(pdf_file=pdf_file))
self._write_pdf(filename)
self._remove_file(pdf_file, assume_yes=True)
return True
def _write_pdf(self, file_name):
"""Write pdf file
Args:
file_name (str): pdf file name.
Returns:
If success, return true.
"""
with open(file_name, "wb") as f:
self.__file_writer.write(f)
return True
[docs] def move_file(self, file, dst, assume_yes):
"""Move file
Args:
file (str): Target file name
dst (str): Target destination path
assume_yes (bool): If true, no verify users
Returns:
bool: If success, return true. Nothing target, return false.
"""
destination = os.path.join(dst, file)
return self._move_file(file=file, dst=destination, assume_yes=assume_yes)