From bf8296c5c3093aef32a5cd91fb008ad257e7a2f1 Mon Sep 17 00:00:00 2001 From: Munyoki Kilyungi Date: Tue, 5 Sep 2023 14:40:35 +0300 Subject: Remove un-necessary ellipsis Signed-off-by: Munyoki Kilyungi --- gn3/auth/db.py | 10 ++-------- gn3/db_utils.py | 1 - 2 files changed, 2 insertions(+), 9 deletions(-) diff --git a/gn3/auth/db.py b/gn3/auth/db.py index 2ba6619..5cd230f 100644 --- a/gn3/auth/db.py +++ b/gn3/auth/db.py @@ -10,40 +10,34 @@ class DbConnection(Protocol): """Type annotation for a generic database connection object.""" def cursor(self) -> Any: """A cursor object""" - ... def commit(self) -> Any: """Commit the transaction.""" - ... def rollback(self) -> Any: """Rollback the transaction.""" - ... + class DbCursor(Protocol): """Type annotation for a generic database cursor object.""" def execute(self, *args, **kwargs) -> Any: """Execute a single query""" - ... def executemany(self, *args, **kwargs) -> Any: """ Execute parameterized SQL statement sql against all parameter sequences or mappings found in the sequence parameters. """ - ... def fetchone(self, *args, **kwargs): """Fetch single result if present, or `None`.""" - ... def fetchmany(self, *args, **kwargs): """Fetch many results if present or `None`.""" - ... def fetchall(self, *args, **kwargs): """Fetch all results if present or `None`.""" - ... + @contextlib.contextmanager def connection(db_path: str, row_factory: Callable = sqlite3.Row) -> Iterator[DbConnection]: diff --git a/gn3/db_utils.py b/gn3/db_utils.py index 5fd250d..e4dc81f 100644 --- a/gn3/db_utils.py +++ b/gn3/db_utils.py @@ -21,7 +21,6 @@ class Connection(Protocol): """Type Annotation for MySQLdb's connection object""" def cursor(self, *args, **kwargs) -> Any: """A cursor in which queries may be performed""" - ... @contextlib.contextmanager -- cgit v1.2.3