add ability for path to be passed as import data in script

This commit is contained in:
Brandon Egger 2023-03-17 18:59:29 -05:00
parent 4455d61795
commit 4b03b787c3

View File

@ -4,7 +4,6 @@ import pandas as pd
import numpy as np
MONGO_URL = "mongodb://localhost:27017/atr_db?replicaSet=test"
DATA_PATH = "/home/beggr/Documents/at_overview.xlsx"
def upload_to_mongo(resources, ar_col):
ar_col.insert_many(resources)
@ -34,7 +33,7 @@ def get_platform_links(row):
return [{"link": platforms[key], "platform": key} for key in platforms.index]
def main(path: str):
df = pd.read_excel(DATA_PATH, header=[0,1])
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"]