From 25da0232ac52509d6761e36ad80ed53b8dbbb64e Mon Sep 17 00:00:00 2001 From: Frederick Muriuki Muriithi Date: Wed, 16 Nov 2022 12:50:47 +0300 Subject: auth: fix bugs in the code * gn3/auth/authorisation/privileges.py: Set id to UUID type * gn3/auth/authorisation/roles.py: fix parameters to types that sqlite3 supports * gn3/auth/db.py: add logging for errors and re-raise the exception * tests/unit/auth/test_roles.py: fix test --- gn3/auth/authorisation/privileges.py | 2 +- gn3/auth/authorisation/roles.py | 7 ++++--- 2 files changed, 5 insertions(+), 4 deletions(-) (limited to 'gn3/auth/authorisation') diff --git a/gn3/auth/authorisation/privileges.py b/gn3/auth/authorisation/privileges.py index c60a58c..09439ad 100644 --- a/gn3/auth/authorisation/privileges.py +++ b/gn3/auth/authorisation/privileges.py @@ -21,4 +21,4 @@ def user_privileges(conn: db.DbConnection, user_id: UUID) -> Iterable[Privilege] (str(user_id),)) results = cursor.fetchall() - return (Privilege(row[0], row[1]) for row in results) + return (Privilege(UUID(row[0]), row[1]) for row in results) diff --git a/gn3/auth/authorisation/roles.py b/gn3/auth/authorisation/roles.py index 7c33ab3..8435c40 100644 --- a/gn3/auth/authorisation/roles.py +++ b/gn3/auth/authorisation/roles.py @@ -33,9 +33,10 @@ def create_role( cursor.execute( "INSERT INTO roles(role_id, role_name) VALUES (?, ?)", - (role.role_id, role.role_name)) - cursor.execute( + (str(role.role_id), role.role_name)) + cursor.executemany( "INSERT INTO role_privileges(role_id, privilege_id) VALUES (?, ?)", - ((role.role_id, priv.privilege_id) for priv in privileges)) + tuple((str(role.role_id), str(priv.privilege_id)) + for priv in privileges)) return role -- cgit 1.4.1