move country_info into data submodule

This commit is contained in:
Sarah Hoffmann 2022-07-06 11:08:36 +02:00
parent bce93d60bd
commit cbbcbb1fd7
10 changed files with 18 additions and 16 deletions

View File

@ -63,8 +63,9 @@ class SetupAll:
@staticmethod
def run(args):
from ..tools import database_import, refresh, postcodes, freeze, country_info
def run(args): # pylint: disable=too-many-statements
from ..data import country_info
from ..tools import database_import, refresh, postcodes, freeze
from ..indexer.indexer import Indexer
country_info.setup_country_config(args.config)

View File

@ -11,7 +11,7 @@ format.
import re
from nominatim.errors import UsageError
from nominatim.tools import country_info
from nominatim.data import country_info
class CountryPostcodeMatcher:
""" Matches and formats a postcode according to a format definition

View File

@ -17,7 +17,7 @@ from nominatim.db.properties import set_property, get_property
from nominatim.errors import UsageError
from nominatim.tokenizer.place_sanitizer import PlaceSanitizer
from nominatim.tokenizer.icu_token_analysis import ICUTokenAnalysis
import nominatim.tools.country_info
import nominatim.data.country_info
LOG = logging.getLogger()
@ -46,7 +46,7 @@ class ICURuleLoader:
config='TOKENIZER_CONFIG')
# Make sure country information is available to analyzers and sanitizers.
nominatim.tools.country_info.setup_country_config(config)
nominatim.data.country_info.setup_country_config(config)
self.normalization_rules = self._cfg_to_icu_rules(rules, 'normalization')
self.transliteration_rules = self._cfg_to_icu_rules(rules, 'transliteration')

View File

@ -30,7 +30,7 @@ Arguments:
any analyzer tagged) is retained. (default: replace)
"""
from nominatim.tools import country_info
from nominatim.data import country_info
class _AnalyzerByLanguage:
""" Processor for tagging the language of names in a place.

View File

@ -10,7 +10,7 @@ Tests for import command of the command-line interface wrapper.
import pytest
import nominatim.tools.database_import
import nominatim.tools.country_info
import nominatim.data.country_info
import nominatim.tools.refresh
import nominatim.tools.postcodes
import nominatim.indexer.indexer
@ -37,7 +37,7 @@ class TestCliImportWithDb:
def test_import_full(self, mock_func_factory, with_updates, place_table, property_table):
mocks = [
mock_func_factory(nominatim.tools.database_import, 'setup_database_skeleton'),
mock_func_factory(nominatim.tools.country_info, 'setup_country_tables'),
mock_func_factory(nominatim.data.country_info, 'setup_country_tables'),
mock_func_factory(nominatim.tools.database_import, 'import_osm_data'),
mock_func_factory(nominatim.tools.refresh, 'import_wikipedia_articles'),
mock_func_factory(nominatim.tools.database_import, 'truncate_data_tables'),
@ -46,7 +46,7 @@ class TestCliImportWithDb:
mock_func_factory(nominatim.tools.database_import, 'create_table_triggers'),
mock_func_factory(nominatim.tools.database_import, 'create_partition_tables'),
mock_func_factory(nominatim.tools.database_import, 'create_search_indices'),
mock_func_factory(nominatim.tools.country_info, 'create_country_names'),
mock_func_factory(nominatim.data.country_info, 'create_country_names'),
mock_func_factory(nominatim.tools.refresh, 'load_address_levels_from_config'),
mock_func_factory(nominatim.tools.postcodes, 'update_postcodes'),
mock_func_factory(nominatim.indexer.indexer.Indexer, 'index_full'),
@ -76,7 +76,7 @@ class TestCliImportWithDb:
mock_func_factory(nominatim.tools.database_import, 'truncate_data_tables'),
mock_func_factory(nominatim.tools.database_import, 'load_data'),
mock_func_factory(nominatim.tools.database_import, 'create_search_indices'),
mock_func_factory(nominatim.tools.country_info, 'create_country_names'),
mock_func_factory(nominatim.data.country_info, 'create_country_names'),
mock_func_factory(nominatim.tools.postcodes, 'update_postcodes'),
mock_func_factory(nominatim.indexer.indexer.Indexer, 'index_full'),
mock_func_factory(nominatim.tools.refresh, 'setup_website'),
@ -94,7 +94,7 @@ class TestCliImportWithDb:
temp_db_conn):
mocks = [
mock_func_factory(nominatim.tools.database_import, 'create_search_indices'),
mock_func_factory(nominatim.tools.country_info, 'create_country_names'),
mock_func_factory(nominatim.data.country_info, 'create_country_names'),
mock_func_factory(nominatim.indexer.indexer.Indexer, 'index_full'),
mock_func_factory(nominatim.tools.refresh, 'setup_website'),
mock_func_factory(nominatim.db.properties, 'set_property')
@ -115,7 +115,7 @@ class TestCliImportWithDb:
def test_import_continue_postprocess(self, mock_func_factory):
mocks = [
mock_func_factory(nominatim.tools.database_import, 'create_search_indices'),
mock_func_factory(nominatim.tools.country_info, 'create_country_names'),
mock_func_factory(nominatim.data.country_info, 'create_country_names'),
mock_func_factory(nominatim.tools.refresh, 'setup_website'),
mock_func_factory(nominatim.db.properties, 'set_property')
]

View File

@ -10,7 +10,7 @@ Tests for function that handle country properties.
from textwrap import dedent
import pytest
from nominatim.tools import country_info
from nominatim.data import country_info
@pytest.fixture
def loaded_country(def_config):

View File

@ -11,7 +11,7 @@ import pytest
from nominatim.tokenizer.place_sanitizer import PlaceSanitizer
from nominatim.data.place_info import PlaceInfo
from nominatim.tools import country_info
from nominatim.data import country_info
@pytest.fixture
def sanitize(def_config, request):

View File

@ -11,7 +11,7 @@ import pytest
from nominatim.data.place_info import PlaceInfo
from nominatim.tokenizer.place_sanitizer import PlaceSanitizer
from nominatim.tools.country_info import setup_country_config
from nominatim.data.country_info import setup_country_config
class TestWithDefaults:

View File

@ -11,7 +11,8 @@ import subprocess
import pytest
from nominatim.tools import postcodes, country_info
from nominatim.tools import postcodes
from nominatim.data import country_info
import dummy_tokenizer
class MockPostcodeTable: