Remove 'http://localhost' from all links in a PDF file.

remove_localhost_from_pdf.py

#!/usr/bin/env python

import re
import sys

import fitz

def remove_localhost_from_pdf(
    pdf_filename: str, pattern: re.Pattern = re.compile(r"^http://localhost:\d+/")
):
    """
    Remove 'http://localhost' from all links in a PDF file.

    Args:
        pdf_filename (str): The filename of the PDF file to process.
        pattern (re.Pattern, optional): A regular expression pattern used to identify links to be removed.
            Defaults to a pattern that matches 'http://localhost:port/'.
    """
    with fitz.open(pdf_filename) as doc:
        for page in doc:
            for link in page.get_links():
                if (uri := link.get("uri")) and pattern.match(uri):
                    link["uri"] = pattern.sub("", uri)
                    page.update_link(link)
        doc.saveIncr()

if __name__ == "__main__":
    for pdf_filename in sys.argv[1:]:
        remove_localhost_from_pdf(pdf_filename)