diff --git a/download/main.py b/download/main.py index 6a8f85bb8..be808dfc6 100644 --- a/download/main.py +++ b/download/main.py @@ -10,7 +10,7 @@ from model.filing import Filings from model.transaction import Transactions -from gdrive_datastore.gdrive import test_data_pull +from gdrive_datastore.gdrive import pull_data def get_last_status(status_list): """ @@ -30,7 +30,7 @@ def main(): data_dir_path = '.local/downloads' # pull data from gdrive and put it in .local/downloads - test_data_pull(default_folder='OpenDisclosure') + pull_data(subfolder='main', default_folder='OpenDisclosure') #engine = create_engine('postgresql+psycopg2://localhost/disclosure-backend-v2', echo=True) diff --git a/download/requirements.txt b/download/requirements.txt index 3be9084a3..c90e61684 100644 --- a/download/requirements.txt +++ b/download/requirements.txt @@ -1,4 +1,4 @@ pandas~=2.0.3 SQLAlchemy~=2.0.20 psycopg2~=2.9.7 -gdrive-datastore==0.0.1.3 +gdrive-datastore==0.0.1.5 diff --git a/download/test_pull_from_gdrive.py b/download/test_pull_from_gdrive.py index e53102c48..6b4d83df4 100644 --- a/download/test_pull_from_gdrive.py +++ b/download/test_pull_from_gdrive.py @@ -1,4 +1,4 @@ import os -from gdrive_datastore.gdrive import test_data_pull +from gdrive_datastore.gdrive import pull_data -test_data_pull(default_folder='OpenDisclosure') +pull_data(subfolder='main', default_folder='OpenDisclosure')