hans/Nominatim
1
0
Fork 1
mirror of https://github.com/osm-search/Nominatim.git synced 2026-03-14 06:44:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

add type annotation to DB utils

Browse Source 
As a cursor is needed as type, make this a public type.
This commit is contained in:
Sarah Hoffmann
2022-07-05 10:46:55 +02:00
parent e6775e713c
commit 26f30bff28
2 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
nominatim/db/connection.py
View File

@@ -22,7 +22,7 @@ from nominatim.errors import UsageError
LOG = logging.getLogger()
class _Cursor(psycopg2.extras.DictCursor):
class Cursor(psycopg2.extras.DictCursor):
""" A cursor returning dict-like objects and providing specialised
execution functions.
"""
@@ -82,18 +82,18 @@ class Connection(psycopg2.extensions.connection):
adds convenience functions for administrating the database.
"""
@overload # type: ignore[override]
def cursor(self) -> _Cursor:
def cursor(self) -> Cursor:
...
@overload
def cursor(self, name: str) -> _Cursor:
def cursor(self, name: str) -> Cursor:
...
@overload
def cursor(self, cursor_factory: Callable[..., T_cursor]) -> T_cursor:
...
def cursor(self, cursor_factory = _Cursor, **kwargs): # type: ignore
def cursor(self, cursor_factory = Cursor, **kwargs): # type: ignore
""" Return a new cursor. By default the specialised cursor is returned.
"""
return super().cursor(cursor_factory=cursor_factory, **kwargs)