diff --git a/institutions/institutions.py b/institutions/institutions.py index 01309004..0e48887d 100644 --- a/institutions/institutions.py +++ b/institutions/institutions.py @@ -10,10 +10,12 @@ def setup(): Institution.init() + def remove(): es = Elasticsearch(ELASTIC_URI, request_timeout=ELASTIC_TIMEOUT) es.indices.delete(ELASTIC_INST_INDEX) + class Institution(DocType): name = String() established = String() diff --git a/tasks.py b/tasks.py index 3dfa4a82..51014428 100644 --- a/tasks.py +++ b/tasks.py @@ -310,16 +310,18 @@ def institutions(grid_file='institutions/grid_2015_10_09.json', ipeds_file='inst grid(grid_file) ipeds(ipeds_file) + @task def remove_institutions(skip=False): if not skip: resp = six.moves.input('You are about to delete the institutions index. Are you sure? (y, n)\n') if resp not in ('y', 'Y', 'Yes', 'yes'): - print ('Remove institutions stopped.') + print('Remove institutions stopped.') return from institutions.institutions import remove remove() + def grid(grid_file): from institutions import institutions, grid institutions.setup()