import platform
import os
import fitz  # pip install PyMuPDF
from PIL import Image
import shutil
import logging


# 将word文件转换成pdf文件
def word2pdf(word_file):
    from win32com import client  # pip install pywin32

    # 获取word格式处理对象
    word = client.Dispatch('Word.Application')
    # 以Doc对象打开文件
    doc_ = word.Documents.Open(word_file)
    # 另存为pdf文件
    pdf_file = word_file.replace(os.path.basename(word_file).split('.')[1], "pdf")
    doc_.SaveAs(pdf_file, FileFormat=17)
    logging.info(f'{word_file} ----转pdf成功')
    # 关闭doc对象
    doc_.Close()
    # 退出word对象
    word.Quit()
    return pdf_file


# 将word文件转换成pdf文件（Linux）
def word2pdf_linux(word_file):
    word_path = os.path.dirname(word_file)
    os.system(f"libreoffice --headless --language=zh-CN --convert-to pdf {word_file} --outdir {word_path}")
    logging.info(f'{word_file} ----转pdf成功')
    pdf_file = word_file.replace(os.path.basename(word_file).split('.')[1], "pdf")
    return pdf_file


# pdf转图片
def pdf2png(pdf_file):
    image_path = os.path.abspath(f'{os.path.dirname(pdf_file)}/tmp_pdf2png')
    try:
        # 创建一个空白图片，用于拼接内容
        width, height = 0, 0
        images = []

        pdf_doc = fitz.open(pdf_file)
        for pg in range(pdf_doc.page_count):
            page = pdf_doc[pg]
            rotate = int(0)
            # 每个尺寸的缩放系数为1.3，这将为我们生成分辨率提高2.6的图像。
            # 此处若是不做设置，默认图片大小为：792X612, dpi=96
            zoom_x = 1.33333333  # (1.33333333-->1056x816)   (2-->1584x1224)
            zoom_y = 1.33333333
            mat = fitz.Matrix(zoom_x, zoom_y).prerotate(rotate)
            pix = page.get_pixmap(matrix=mat, alpha=False)

            if not os.path.exists(image_path):  # 判断存放图片的文件夹是否存在
                os.makedirs(image_path)  # 若图片文件夹不存在就创建
            pix.save(image_path + '/' + 'tmp%s.png' % pg)  # 将图片写入指定的文件夹内

            img = Image.open(image_path + '/' + 'tmp%s.png' % pg)
            img_width, img_height = img.size

            # 更新拼接图片的宽度和高度
            width = max(width, img_width)
            height += img_height

            # 添加图片到拼接列表
            images.append(img)

        # 创建一个空白长图
        long_image = Image.new('RGB', (width, height), (255, 255, 255))
        y_offset = 0

        # 将每张图片拼接到长图中
        for img in images:
            long_image.paste(img, (0, y_offset))
            y_offset += img.height

        # 保存拼接后的长图
        png_file = pdf_file.replace(os.path.basename(pdf_file).split('.')[1], "png")
        long_image.save(png_file)

        # 删除中间临时保存的图片
        shutil.rmtree(image_path)
    except IOError as error:
        logging.error('pdf转png失败')
        raise error
    else:
        logging.info("pdf转png成功")
        return png_file


def word_to_long_image(word_file_path):
    try:
        _file = os.path.abspath(word_file_path)  # os.path.abspath('input.docx')

        if platform.system().lower() == 'windows':
            pdf_file = word2pdf(_file)
        else:
            pdf_file = word2pdf_linux(_file)

        png_file = pdf2png(pdf_file)

        # 删除中间保存的pdf文件
        os.remove(pdf_file)

        return png_file
    except Exception as error:
        logging.error('word转长图出错：{}'.format(error))
        raise error