brandon/uiowaATR
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

update mongo client to use global mongo vars

Browse Source
This commit is contained in:
Brandon Egger 2023-03-19 22:37:52 -05:00
parent edc979debb
commit 3e67399d1a
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/import/main.py
View File

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