Main handler

Author: robertnovac1

filenameprocessor/src/file_name_processor.py

@@ -6,6 +6,7 @@
 (ODS code has multiple lengths)
 """
 
+import argparse
 from uuid import uuid4
 from utils_for_filenameprocessor import get_created_at_formatted_string, move_file, invoke_filename_lambda
 from file_key_validation import validate_file_key
@@ -170,16 +171,20 @@ def lambda_handler(event: dict, context) -> None:  # pylint: disable=unused-argu
 
 
 if __name__ == "__main__":
+    parser = argparse.ArgumentParser("file_name_processor")
+    parser.add_argument("--bucket", required=True, help="Bucket name.", type=str)
+    parser.add_argument("--key", required=True, help="Object key.", type=str)
+    args = parser.parse_args()
 
     event = {
         "Records": [
             {
                 "s3": {
                     "bucket": {
-                        "name": "immunisation-batch-internal-dev-data-sources"
+                        "name": args.bucket
                     },
                     "object": {
-                        "key": "FLU_Vaccinations_v5_YGM41_20000101T00000001.csv"
+                        "key": args.key
                     }
                 }
             }