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 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) (limited to 'gn3/auth') 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]: -- cgit v1.2.3