Inverts pdf colors

invert-pdf-colors.py

import pdf2image
from PIL import ImageOps
import img2pdf
import os
import sys
from pathlib import Path
from random import randint


def invert_color(filepath):
    print("Converting to images...")
    images = pdf2image.convert_from_path(filepath)

    idx_counter = []
    temp_dir_path = Path(f"temp_{randint(1000000, 9999999)}")
    temp_dir_path.mkdir(exist_ok=True)
    try:
        print("Inverting colors...")
        for idx, i in enumerate(images):
            temp_fp = str(temp_dir_path / f"temp_{idx}.jpeg")
            idx_counter.append(temp_fp)
            i = ImageOps.invert(i)
            i.save(temp_fp)
        print("Saving output to output.pdf...")
        with open("output.pdf", "wb") as f:
            f.write(img2pdf.convert(idx_counter))
    finally:
        print("Deleting temp directory...")
        for fp in idx_counter:
            os.remove(fp)
        temp_dir_path.rmdir()
    print("Done!")


#
if __name__ == "__main__":
    args = sys.argv
    fp = args[-1]
    if not fp.endswith("pdf"):
        print("USAGE: python3 script.py <filepath>")
        exit(1)
    invert_color(filepath=fp)