src/nominatim_db/tools/special_phrases/sp_csv_loader.py

   1 # SPDX-License-Identifier: GPL-3.0-or-later
   2 #
   3 # This file is part of Nominatim. (https://nominatim.org)
   4 #
   5 # Copyright (C) 2024 by the Nominatim developer community.
   6 # For a full list of authors see the git log.
   7 """
   8     Module containing the SPCsvLoader class.
   9
  10     The class allows to load phrases from a csv file.
  11 """
  12 from typing import Iterable
  13 import csv
  14 import os
  15
  16 from ...errors import UsageError
  17 from .special_phrase import SpecialPhrase
  18
  19
  20 class SPCsvLoader:
  21     """
  22         Handles loading of special phrases from external csv file.
  23     """
  24     def __init__(self, csv_path: str) -> None:
  25         self.csv_path = csv_path
  26
  27     def generate_phrases(self) -> Iterable[SpecialPhrase]:
  28         """ Open and parse the given csv file.
  29             Create the corresponding SpecialPhrases.
  30         """
  31         self._check_csv_validity()
  32
  33         with open(self.csv_path, encoding='utf-8') as fd:
  34             reader = csv.DictReader(fd, delimiter=',')
  35             for row in reader:
  36                 yield SpecialPhrase(row['phrase'], row['class'], row['type'], row['operator'])
  37
  38     def _check_csv_validity(self) -> None:
  39         """
  40             Check that the csv file has the right extension.
  41         """
  42         _, extension = os.path.splitext(self.csv_path)
  43
  44         if extension != '.csv':
  45             raise UsageError(f'The file {self.csv_path} is not a csv file.')