hans/Nominatim
1
0
Fork 1
mirror of https://github.com/osm-search/Nominatim.git synced 2026-02-26 11:08:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix type issues with calls to pyosmium

Browse Source
This commit is contained in:
Sarah Hoffmann
2022-11-09 20:45:56 +01:00
parent 846ecff0c5
commit 64c591da7f
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
nominatim/tools/replication.py
View File

@@ -156,25 +156,25 @@ def _make_replication_server(url: str, timeout: int) -> ContextManager[Replicati
""" Download a resource from the given URL and return a byte sequence """ Download a resource from the given URL and return a byte sequence
of the content. of the content.
""" """
get_params = { headers = {"User-Agent" : f"Nominatim (pyosmium/{pyo_version.pyosmium_release})"}
'headers': {"User-Agent" : f"Nominatim (pyosmium/{pyo_version.pyosmium_release})"},
'timeout': timeout or None,
'stream': True
}
if self.session is not None: if self.session is not None:
return self.session.get(url.get_full_url(), **get_params) return self.session.get(url.get_full_url(),
headers=headers, timeout=timeout or None,
stream=True)
@contextmanager @contextmanager
def _get_url_with_session() -> Iterator[requests.Response]: def _get_url_with_session() -> Iterator[requests.Response]:
with requests.Session() as session: with requests.Session() as session:
request = session.get(url.get_full_url(), **get_params) # type: ignore request = session.get(url.get_full_url(),
headers=headers, timeout=timeout or None,
stream=True)
yield request yield request
return _get_url_with_session() return _get_url_with_session()
repl = ReplicationServer(url) repl = ReplicationServer(url)
repl.open_url = types.MethodType(patched_open_url, repl) setattr(repl, 'open_url', types.MethodType(patched_open_url, repl))
return cast(ContextManager[ReplicationServer], repl) return cast(ContextManager[ReplicationServer], repl)