From e4b8116c98590f7e38383227eede537a1fdc44ed Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Tue, 8 Aug 2023 05:17:42 +0300
Subject: Fix type-checking errors

---
 gn_auth/auth/db/mariadb.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

(limited to 'gn_auth/auth/db/mariadb.py')

diff --git a/gn_auth/auth/db/mariadb.py b/gn_auth/auth/db/mariadb.py
index 69c1d8e..35f6947 100644
--- a/gn_auth/auth/db/mariadb.py
+++ b/gn_auth/auth/db/mariadb.py
@@ -3,11 +3,23 @@ import logging
 import traceback
 import contextlib
 from urllib.parse import urlparse
-from typing import Tuple, Iterator
+from typing import Any, Tuple, Protocol, Iterator
 
 import MySQLdb as mdb
 
-from .protocols import DbConnection
+class DbConnection(Protocol):
+    """Type annotation for a generic database connection object."""
+    def cursor(self, *args, **kwargs) -> Any:
+        """A cursor object"""
+        ...
+
+    def commit(self, *args, **kwargs) -> Any:
+        """Commit the transaction."""
+        ...
+
+    def rollback(self) -> Any:
+        """Rollback the transaction."""
+        ...
 
 def parse_db_url(sql_uri: str) -> Tuple:
     """Parse SQL_URI env variable note:there is a default value for SQL_URI so a
-- 
cgit v1.2.3