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 = []