diff --git a/script/ingest_pe_numbers.py b/script/ingest_pe_numbers.py index e3bbecb0..d8b22059 100644 --- a/script/ingest_pe_numbers.py +++ b/script/ingest_pe_numbers.py @@ -7,7 +7,7 @@ import sys parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.append(parent_dir) -from atst.app import make_deps, make_config +from atst.app import make_app, make_config from atst.domain.pe_numbers import PENumbers @@ -23,9 +23,11 @@ def make_pe_number_repo(config): if __name__ == "__main__": config = make_config() - url = config["default"]['PE_NUMBER_CSV_URL'] + url = config['PE_NUMBER_CSV_URL'] print("Fetching PE numbers from {}".format(url)) pe_numbers = get_pe_numbers(url) - print("Inserting {} PE numbers".format(len(pe_numbers))) - pe_numbers_repo = make_pe_number_repo(config) - pe_numbers_repo.create_many(pe_numbers) + + app = make_app(config) + with app.app_context(): + print("Inserting {} PE numbers".format(len(pe_numbers))) + PENumbers.create_many(pe_numbers)