Merge pull request #4 from ben-hudson/master

PDF metadata is set according to scraped paper data

arxiv_dl/helpers.py

@@ -1,6 +1,7 @@
 import json
 import logging
 import os
+import pypdf
 import re
 import shlex
 import string
@@ -89,6 +90,8 @@ def download_pdf(
         if out.is_file():
             logger.debug(f'[Done] Paper saved to "{download_path}"')
 
+    add_pdf_metadata(paper_data, download_path)
+
     return None
 
 
@@ -180,6 +183,30 @@ def command_exists(command: str) -> bool:
     return which(command) is not None
 
 
+def add_pdf_metadata(paper_data: PaperData, download_path: Path):
+    reader = pypdf.PdfReader(download_path)
+    writer = pypdf.PdfWriter()
+
+    # Add all pages to the writer
+    for page in reader.pages:
+        writer.add_page(page)
+
+    # Add the old metadata
+    metadata = reader.metadata
+    writer.add_metadata(metadata)
+
+    # Add the new metadata
+    writer.add_metadata({
+        "/Author": ", ".join(paper_data.authors),
+        "/Title": paper_data.title,
+        "/Subject": paper_data.abstract,
+    })
+
+    # Save the new PDF to a file
+    with open(download_path, "wb") as f:
+        writer.write(f)
+
+
 def add_to_paper_list(paper_data: PaperData, download_dir: Union[str, Path]) -> None:
     paper_list_path: Path = Path(download_dir) / "000_Paper_List.json"
     paper_dict = paper_data.dict()

requirements.txt

@@ -1,4 +1,5 @@
 beautifulsoup4
 colorlog
 pydantic
+pypdf
 requests

setup.py

@@ -36,6 +36,7 @@ def update_version_file():
         "colorlog>=4.1.0",
         "requests",
         "pydantic",
+        "pypdf>=3.10.0",
         "beautifulsoup4",
     ],
     extras_require={