pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix ingest_pe_numbers script

Browse Source
This commit is contained in:
Patrick Smith 2018-08-02 16:48:35 -04:00
parent 165a5bf374
commit a5c0d3dd92
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
script/ingest_pe_numbers.py
View File

@ -7,7 +7,7 @@ import sys
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.append(parent_dir) 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 from atst.domain.pe_numbers import PENumbers
@ -23,9 +23,11 @@ def make_pe_number_repo(config):
if __name__ == "__main__": if __name__ == "__main__":
config = make_config() config = make_config()
url = config["default"]['PE_NUMBER_CSV_URL'] url = config['PE_NUMBER_CSV_URL']
print("Fetching PE numbers from {}".format(url)) print("Fetching PE numbers from {}".format(url))
pe_numbers = get_pe_numbers(url) pe_numbers = get_pe_numbers(url)
print("Inserting {} PE numbers".format(len(pe_numbers)))
pe_numbers_repo = make_pe_number_repo(config) app = make_app(config)
pe_numbers_repo.create_many(pe_numbers) with app.app_context():
print("Inserting {} PE numbers".format(len(pe_numbers)))
PENumbers.create_many(pe_numbers)