From 5e185a6498a9059b4dbd719a11368bb605298a6b Mon Sep 17 00:00:00 2001
From: Quang <nguyenvanquang247@gmail.com>
Date: Wed, 22 Jan 2025 17:33:30 +0700
Subject: [PATCH] Add command-line interface for PDF upload script

---
 dsst_etl/upload_pdfs_title_is_pmid.py | 37 +++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/dsst_etl/upload_pdfs_title_is_pmid.py b/dsst_etl/upload_pdfs_title_is_pmid.py
index 9546f9f..2979d84 100644
--- a/dsst_etl/upload_pdfs_title_is_pmid.py
+++ b/dsst_etl/upload_pdfs_title_is_pmid.py
@@ -1,3 +1,4 @@
+import argparse
 import hashlib
 
 import boto3
@@ -111,3 +112,39 @@ def _create_document_entries(self, key, file_content, file_hash, provenance):
             self.db_session.flush()
         except Exception as e:
             logger.error(f"Error running Oddpub analysis: {str(e)}")
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Upload PDFs to S3 where the title is the PMID."
+    )
+
+    parser.add_argument(
+        "--db-url",
+        required=True,
+        help="The database connection URL. This should be a valid SQLAlchemy database URL.",
+    )
+    parser.add_argument(
+        "--pdf-path",
+        required=True,
+        help="The path to the PDF file to upload. This should be a valid file path on your system.",
+    )
+
+    args = parser.parse_args()
+
+    # Set up the database session
+    engine = sqlalchemy.create_engine(args.db_url)
+    Session = sqlalchemy.orm.sessionmaker(bind=engine)
+    db_session = Session()
+
+    try:
+        uploader = UploadPDFsTitleIsPMID(db_session)
+        uploader.process_s3_inventory(args.pdf_path)
+    except Exception as e:
+        logger.error(f"An error occurred: {str(e)}")
+    finally:
+        db_session.close()
+
+
+if __name__ == "__main__":
+    main()