p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Deployment: Wrap argument parsing in function

This commit is contained in:
Mike Salvatore 2021-09-28 12:01:16 -04:00
parent 7bcfc6d27a
commit 38f50641a5
1 changed files with 33 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
deployment_scripts/dump_attack_mitigations.py
View File

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