Merge pull request #2786 from lonvia/export-centroid-for-tokenizer

[nominatim.git] / nominatim / clicmd / freeze.py

diff --git a/nominatim/clicmd/freeze.py b/nominatim/clicmd/freeze.py
index 1b311e97f2f7ee1081c4d1706501b9591c08704c..5dfdd255c8736768713ee3f3532ffb3bf2b3dbb0 100644 (file)
--- a/nominatim/clicmd/freeze.py
+++ b/nominatim/clicmd/freeze.py
@@ -1,8 +1,16 @@
+# 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.

 """
 Implementation of the 'freeze' subcommand.
 """
 """
 Implementation of the 'freeze' subcommand.
 """

+import argparse

 
 

-from ..db.connection import connect
+from nominatim.db.connection import connect
+from nominatim.clicmd.args import NominatimArgs

 
 # Do not repeat documentation of subcommand classes.
 # pylint: disable=C0111
 
 # Do not repeat documentation of subcommand classes.
 # pylint: disable=C0111


@@ -21,16 +29,15 @@ class SetupFreeze:
     This command has the same effect as the `--no-updates` option for imports.
     """
 
     This command has the same effect as the `--no-updates` option for imports.
     """
 

-    @staticmethod
-    def add_args(parser):
+    def add_args(self, parser: argparse.ArgumentParser) -> None:

         pass # No options
 
         pass # No options
 

-    @staticmethod
-    def run(args):
+
+    def run(self, args: NominatimArgs) -> int:

         from ..tools import freeze
 
         with connect(args.config.get_libpq_dsn()) as conn:
             freeze.drop_update_tables(conn)
         from ..tools import freeze
 
         with connect(args.config.get_libpq_dsn()) as conn:
             freeze.drop_update_tables(conn)

-        freeze.drop_flatnode_file(args.config.FLATNODE_FILE)
+        freeze.drop_flatnode_file(args.config.get_path('FLATNODE_FILE'))

 
         return 0
 
         return 0