Added missing return types to functions

[nominatim.git] / nominatim / db / async_connection.py

diff --git a/nominatim/db/async_connection.py b/nominatim/db/async_connection.py
index 1c550198ec053be6e70e0ae25cce194d66e70e96..07767e48ad8934c5dc831c16e9c0040e9a3a6d01 100644 (file)
--- a/nominatim/db/async_connection.py
+++ b/nominatim/db/async_connection.py
@@ -6,7 +6,7 @@
 # For a full list of authors see the git log.
 """ Non-blocking database connections.
 """
 # For a full list of authors see the git log.
 """ Non-blocking database connections.
 """

-from typing import Callable, Any, Optional, List, Iterator
+from typing import Callable, Any, Optional, Iterator, Sequence

 import logging
 import select
 import time
 import logging
 import select
 import time


@@ -22,7 +22,7 @@ try:
 except ImportError:
     __has_psycopg2_errors__ = False
 
 except ImportError:
     __has_psycopg2_errors__ = False
 

-from nominatim.typing import T_cursor
+from nominatim.typing import T_cursor, Query

 
 LOG = logging.getLogger()
 
 
 LOG = logging.getLogger()
 


@@ -65,12 +65,12 @@ class DBConnection:
                  ignore_sql_errors: bool = False) -> None:
         self.dsn = dsn
 
                  ignore_sql_errors: bool = False) -> None:
         self.dsn = dsn
 

-        self.current_query: Optional[str] = None
-        self.current_params: Optional[List[Any]] = None
+        self.current_query: Optional[Query] = None
+        self.current_params: Optional[Sequence[Any]] = None

         self.ignore_sql_errors = ignore_sql_errors
 
         self.ignore_sql_errors = ignore_sql_errors
 

-        self.conn: Optional['psycopg2.connection'] = None
-        self.cursor: Optional['psycopg2.cursor'] = None
+        self.conn: Optional['psycopg2._psycopg.connection'] = None
+        self.cursor: Optional['psycopg2._psycopg.cursor'] = None

         self.connect(cursor_factory=cursor_factory)
 
     def close(self) -> None:
         self.connect(cursor_factory=cursor_factory)
 
     def close(self) -> None:


@@ -78,7 +78,7 @@ class DBConnection:
         """
         if self.conn is not None:
             if self.cursor is not None:
         """
         if self.conn is not None:
             if self.cursor is not None:

-                self.cursor.close() # type: ignore[no-untyped-call]
+                self.cursor.close()

                 self.cursor = None
             self.conn.close()
 
                 self.cursor = None
             self.conn.close()
 


@@ -94,7 +94,8 @@ class DBConnection:
 
         # Use a dict to hand in the parameters because async is a reserved
         # word in Python3.
 
         # Use a dict to hand in the parameters because async is a reserved
         # word in Python3.

-        self.conn = psycopg2.connect(**{'dsn': self.dsn, 'async': True})
+        self.conn = psycopg2.connect(**{'dsn': self.dsn, 'async': True}) # type: ignore
+        assert self.conn

         self.wait()
 
         if cursor_factory is not None:
         self.wait()
 
         if cursor_factory is not None:


@@ -128,7 +129,7 @@ class DBConnection:
                 self.current_query = None
                 return
 
                 self.current_query = None
                 return
 

-    def perform(self, sql: str, args: Optional[List[Any]] = None) -> None:
+    def perform(self, sql: Query, args: Optional[Sequence[Any]] = None) -> None:

         """ Send SQL query to the server. Returns immediately without
             blocking.
         """
         """ Send SQL query to the server. Returns immediately without
             blocking.
         """