Merge pull request #2770 from lonvia/typed-python

[nominatim.git] / nominatim / tokenizer / sanitizers / strip_brace_terms.py

diff --git a/nominatim/tokenizer/sanitizers/strip_brace_terms.py b/nominatim/tokenizer/sanitizers/strip_brace_terms.py
index caadc815edb8e71ec5bca387ee166c10bccd96cb..119d5693a58beee7ebfa2fa852b03cb312a5b441 100644 (file)
--- a/nominatim/tokenizer/sanitizers/strip_brace_terms.py
+++ b/nominatim/tokenizer/sanitizers/strip_brace_terms.py
@@ -1,14 +1,25 @@
+# SPDX-License-Identifier: GPL-2.0-only
+#
+# This file is part of Nominatim. (https://nominatim.org)
+#
+# Copyright (C) 2022 by the Nominatim developer community.
+# For a full list of authors see the git log.

 """
 This sanitizer creates additional name variants for names that have
 addendums in brackets (e.g. "Halle (Saale)"). The additional variant contains
 only the main name part with the bracket part removed.
 """
 """
 This sanitizer creates additional name variants for names that have
 addendums in brackets (e.g. "Halle (Saale)"). The additional variant contains
 only the main name part with the bracket part removed.
 """

+from typing import Callable

 
 

-def create(_):
+from nominatim.tokenizer.sanitizers.base import ProcessInfo
+from nominatim.tokenizer.sanitizers.config import SanitizerConfig
+
+
+def create(_: SanitizerConfig) -> Callable[[ProcessInfo], None]:

     """ Create a name processing function that creates additional name variants
         for bracket addendums.
     """
     """ Create a name processing function that creates additional name variants
         for bracket addendums.
     """

-    def _process(obj):
+    def _process(obj: ProcessInfo) -> None:

         """ Add variants for names that have a bracket extension.
         """
         if obj.names:
         """ Add variants for names that have a bracket extension.
         """
         if obj.names: