work round typing bug in pyosmium 4.0

[nominatim.git] / src / nominatim_db / tools / replication.py

diff --git a/src/nominatim_db/tools/replication.py b/src/nominatim_db/tools/replication.py
index 2b1d444f0cb1e3cd08b6540273ac9b64d136e43f..91171deb8cad626a4b64e1ba0bb5a9c95b5aa0ac 100644 (file)
--- a/src/nominatim_db/tools/replication.py
+++ b/src/nominatim_db/tools/replication.py
@@ -16,8 +16,6 @@ import time
 import types
 import urllib.request as urlrequest
 
 import types
 import urllib.request as urlrequest
 

-import requests
-

 from ..errors import UsageError
 from ..db import status
 from ..db.connection import Connection, connect, server_version_tuple
 from ..errors import UsageError
 from ..db import status
 from ..db.connection import Connection, connect, server_version_tuple


@@ -27,9 +25,10 @@ try:
     from osmium.replication.server import ReplicationServer
     from osmium import WriteHandler
     from osmium import version as pyo_version
     from osmium.replication.server import ReplicationServer
     from osmium import WriteHandler
     from osmium import version as pyo_version

-except ImportError as exc:
+    import requests
+except ModuleNotFoundError as exc:

     logging.getLogger().critical("pyosmium not installed. Replication functions not available.\n"
     logging.getLogger().critical("pyosmium not installed. Replication functions not available.\n"

-                                 "To install pyosmium via pip: pip3 install osmium")
+                                 "To install pyosmium via pip: pip install osmium")

     raise UsageError("replication tools not available") from exc
 
 LOG = logging.getLogger()
     raise UsageError("replication tools not available") from exc
 
 LOG = logging.getLogger()


@@ -126,7 +125,8 @@ def update(dsn: str, options: MutableMapping[str, Any],
     # Read updates into file.
     with _make_replication_server(options['base_url'], socket_timeout) as repl:
         outhandler = WriteHandler(str(options['import_file']))
     # Read updates into file.
     with _make_replication_server(options['base_url'], socket_timeout) as repl:
         outhandler = WriteHandler(str(options['import_file']))

-        endseq = repl.apply_diffs(outhandler, startseq + 1,
+        # tyoing: work around typing bug in pyosmium 4.0
+        endseq = repl.apply_diffs(outhandler, startseq + 1, # type: ignore[arg-type]

                                   max_size=options['max_diff_size'] * 1024)
         outhandler.close()
 
                                   max_size=options['max_diff_size'] * 1024)
         outhandler.close()