美文网首页python
修改PDF字体颜色和背景(反转色)

修改PDF字体颜色和背景(反转色)

作者: Noza_ea8f | 来源:发表于2021-10-11 15:36 被阅读0次

这个需求是这样;
有这样一个PDF文件


image.png

文件需要打印;
但是打印出来比较费墨;
所以需要变成白底黑字的;
我百度了一下;
没有直接能够操作PDF文字和背景的库;
当然,也许是我没找到;
怎么办呢?

思路一:

我尝试转换成WORD再修改文字及背景;
但效果让我大跌眼镜;

思路二:

先把PDF文件逐页转换成图片;
图片按顺序命名;
第二步把图片进行颜色反转;
第三步按照原顺序把图片组合成PDF文档;
这个方案可行!
代码:


image.png

把PDF文件逐页转换成图片

import os
import fitz

file_dir = 'pdf'


def get_file_name(file_dir):
    '''
    获取指定目录下所有文件名称
    :param file_dir:指定目录
    :return:返回文件名列表
    '''
    for root, dirs, files in os.walk(file_dir):
        # return root#当前目录路径
        # return dirs#当前路径下所有子目录
        return files  # 当前路径下所有非目录子文件


def conver_img():
    for filename in filenames:
        # pdf_name = os.path.splitext(pdf)[0]
        print(filename)

        doc = fitz.open(f'{file_dir}/{filename}')

        for pg in range(doc.pageCount):
            page = doc[pg]
            rotate = int(0)
            # 每个尺寸的缩放系数为2,这将为我们生成分辨率提高四倍的图像。
            zoom_x = 2.0
            zoom_y = 2.0
            trans = fitz.Matrix(zoom_x, zoom_y).preRotate(rotate)
            pm = page.getPixmap(matrix=trans, alpha=False)
            # pm.writePNG('%s.png' % pg)
            pm.writePNG(f'jpg/{pg}.jpg')


if __name__ == '__main__':
    filenames = get_file_name(file_dir)
    print(filenames)
    conver_img()

把图片进行颜色反转

import cv2
import numpy as np


# 读取一张图片并保存
def read_img_output2save(path, num):
    img = cv2.imread(path, 1)  # 读取一张图片,彩色
    cha = img.shape
    height, width, deep = cha
    dst = np.zeros((height, width, 3), np.uint8)
    for i in range(height):  # 色彩反转
        for j in range(width):
            b, g, r = img[i, j]
            dst[i, j] = (255 - b, 255 - g, 255 - r)
    # cv2.imshow('img', img)
    # cv2.imshow('dst', dst)
    cv2.imwrite(f"save/{num}.jpg", dst)
    # cv2.waitKey()


for i in range(60):
    path = f'jpg/{i}.jpg'
    read_img_output2save(path, num=i)
    print(f'保存第{i}张图片成功!')

按照原顺序把图片组合成PDF文档

from PIL import Image
import os


def combine2pdf(folderPath, pdfFilePath):
    # files = os.listdir(folderPath)

    pngFiles = []
    sources = []
    for file in range(60):
        print(file)
        file = f'{file}.jpg'
        if 'jpg' in file:
            pngFiles.append(folderPath + file)
    # pngFiles.sort()  # 这里不要选择排序,这种排序是按头一个数字进行排序,会打乱原有顺序
    # print(pngFiles[0])
    output = Image.open(pngFiles[0])
    pngFiles.pop(0)
    for file in pngFiles:
        pngFile = Image.open(file)
        if pngFile.mode == "RGB":
            pngFile = pngFile.convert("RGB")
        sources.append(pngFile)
    output.save(pdfFilePath, "pdf", save_all=True, append_images=sources)


if __name__ == "__main__":
    folder = "save/"
    pdfFile = "jpg2pdf/contract.pdf"
    combine2pdf(folder, pdfFile)

最终效果

image.png

相关文章

网友评论

    本文标题:修改PDF字体颜色和背景(反转色)

    本文链接:https://www.haomeiwen.com/subject/sofholtx.html