From 3e67399d1aa4204156c98783aadee87962fa3440 Mon Sep 17 00:00:00 2001 From: Brandon Egger Date: Sun, 19 Mar 2023 22:37:52 -0500 Subject: [PATCH] update mongo client to use global mongo vars --- scripts/import/main.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/scripts/import/main.py b/scripts/import/main.py index 77f03e6..fda771e 100644 --- a/scripts/import/main.py +++ b/scripts/import/main.py @@ -4,6 +4,7 @@ import pandas as pd import numpy as np MONGO_URL = "mongodb://localhost:27017/atr_db?replicaSet=test" +MONGO_DB = "dev" def upload_to_mongo(resources, ar_col): ar_col.insert_many(resources) @@ -35,8 +36,8 @@ def get_platform_links(row): def main(path: str): df = pd.read_excel(path, header=[0,1]) - mongo_client = pymongo.MongoClient("mongodb://localhost:27017/atr_db?replicaSet=test") - mongo_db = mongo_client["atr_db"] + mongo_client = pymongo.MongoClient(MONGO_URL) + mongo_db = mongo_client[MONGO_DB] ar_col = mongo_db["AuditoryResource"] resources = []