From c5d4a6f8104af1eaa0c9dfd8ea7025a673de731e Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Thu, 2 Nov 2023 15:38:49 +0300
Subject: mypy: Fix typing errors.

---
 gn_auth/auth/authorisation/resources/models.py | 6 +++---
 gn_auth/commands.py                            | 4 +++-
 2 files changed, 6 insertions(+), 4 deletions(-)

(limited to 'gn_auth')

diff --git a/gn_auth/auth/authorisation/resources/models.py b/gn_auth/auth/authorisation/resources/models.py
index a9d1378..15bb72f 100644
--- a/gn_auth/auth/authorisation/resources/models.py
+++ b/gn_auth/auth/authorisation/resources/models.py
@@ -298,7 +298,7 @@ def attach_resources_data(
     with db.cursor(conn) as cursor:
         return tuple(
             resource for categories in
-            (resource_data_function[category.resource_category_key](
+            (resource_data_function[category.resource_category_key](# type: ignore[operator]
                 cursor, rscs)
              for category, rscs in organised.items())
             for resource in categories)
@@ -377,7 +377,7 @@ def save_resource(
 
 def user_roles_on_resources(conn: db.DbConnection,
                             user: User,
-                            resource_ids: tuple[UUID] = tuple()) -> dict:
+                            resource_ids: tuple[UUID, ...] = tuple()) -> dict:
     """Get roles on resources for a particular user."""
     def __setup_roles__(old_roles, row):
         roles = {role.role_id: role for role in old_roles}
@@ -409,7 +409,7 @@ def user_roles_on_resources(conn: db.DbConnection,
         "INNER JOIN role_privileges AS rp ON r.role_id=rp.role_id "
         "INNER JOIN privileges AS p ON rp.privilege_id=p.privilege_id "
         "WHERE ur.user_id=?")
-    params = (str(user.user_id),)
+    params: tuple[str, ...] = (str(user.user_id),)
 
     if len(resource_ids) > 0:
         pholders = ", ".join(["?"] * len(resource_ids))
diff --git a/gn_auth/commands.py b/gn_auth/commands.py
index cc00413..02bed10 100644
--- a/gn_auth/commands.py
+++ b/gn_auth/commands.py
@@ -44,7 +44,9 @@ Returns the name of the specific redis hash for the specific task.
         conn.hset(name=unique_id, key="env", value=json.dumps(env))
     return unique_id
 
-def run_cmd(cmd: str, success_codes: Tuple = (0,), env: str = None) -> Dict:
+def run_cmd(cmd: str,
+            success_codes: Tuple = (0,),
+            env: Optional[str] = None) -> Dict:
     """Run CMD and return the CMD's status code and output as a dict"""
     parsed_cmd = json.loads(cmd)
     parsed_env = (json.loads(env) if env is not None else None)
-- 
cgit v1.2.3