python文件处理——图片去水印

目录

一、要求

二、代码实现

三、结果输出

一、要求

        PLT==8.4.0

二、代码实现

from itertools import product
from PIL import Image
import os


def drop_watermark(filepath):
    '''
    :param filepath: filepath
    :return: drop watermark img
    '''
    new_pdfpath = filepath[:-4] + '_drop_watermark' + filepath[-4:]
    try:
        img = Image.open(filepath)
        width, height = img.size
        for pos in product(range(width), range(height)):
            if sum(img.getpixel(pos)[:3]) > 600:
                img.putpixel(pos, (255,255,255))
        print("n")
    except Exception as e:
        print(f"pdf decode error , info : {e}")

    else:
        img.save(new_pdfpath)
        print(f"img drop watermark successful , check your file in {os.path.dirname(new_pdfpath)}")




if __name__ == '__main__':
    drop_watermark('pdf文件路径')

三、结果输出

        在原文件的同级目录生成一个加_drop_watermark的图片文件