def import_naptan(previous_result=None, url=None, force_update=False): if previous_result in (None, True): try: app = create_app() with app.blueprint_context(BLUEPRINT_NAME): url = url or app.config['NAPTAN_IMPORT_URL'] naptan = get_resource(url, force_update) if naptan: archive = zipfile.ZipFile(open(naptan)) f = archive.open('NaPTAN.xml') naptan = NaPTANImporter(search_importer, 10, f, ['340'], 'identifiers') naptan.run() return True else: logger.info("Naptan hasn't been imported - resource not loaded") except: logger.error("Error running NaPTAN importer", exc_info=True) return False
def test_search_called_each_result(self): naptan_importer = NaPTANImporter(self.mock_solr, 10, file(self.naptan_file), ['639'], 'identifiers') naptan_importer.run() self.assertEqual(self.mock_solr.search_for_ids.call_count, len(naptan_importer.handler.stop_points)+len(naptan_importer.handler.stop_areas))