Deployment: Wrap argument parsing in function

deployment_scripts/dump_attack_mitigations.py

@@ -2,6 +2,8 @@ import argparse
 
 import pymongo
 
+
+def parse_args():
     parser = argparse.ArgumentParser(description="Export attack mitigations from a database")
     parser.add_argument(
         "-host", "--mongo_host", default="localhost", help="URL for mongo database.", required=False
@@ -31,7 +33,7 @@ parser.add_argument(
         help="Which collection are we going to export",
         required=False,
     )
-args = parser.parse_args()
+    return parser.parse_args()
 
 
 def connect_to_mongo(mongo_host: str, mongo_port: int, database_name: str) -> pymongo.MongoClient:
@@ -51,6 +53,7 @@ def clean_collection(mongodb: pymongo.MongoClient, collection_name: str):
 
 
 def main():
+    args = parse_args()
     mongodb = connect_to_mongo(args.mongo_host, args.mongo_port, args.database_name)
 
     clean_collection(mongodb, args.collection_name)