From 4b03b787c3a8e4e8891073f1930c395a8478535a Mon Sep 17 00:00:00 2001 From: Brandon Egger Date: Fri, 17 Mar 2023 18:59:29 -0500 Subject: [PATCH] add ability for path to be passed as import data in script --- scripts/import/main.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/scripts/import/main.py b/scripts/import/main.py index 25c87c9..77f03e6 100644 --- a/scripts/import/main.py +++ b/scripts/import/main.py @@ -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"]