blob: ae4ed885d961e72909d72c6afe54628d220a8d5b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
"""Handle privileges"""
from typing import Any, Iterable, NamedTuple
from gn3.auth import db
from gn3.auth.authentication.users import User
class Privilege(NamedTuple):
"""Class representing a privilege: creates immutable objects."""
privilege_id: str
privilege_description: str
def dictify(self) -> dict[str, Any]:
"""Return a dict representation of `Privilege` objects."""
return {
"privilege_id": self.privilege_id,
"privilege_description": self.privilege_description
}
def user_privileges(conn: db.DbConnection, user: User) -> Iterable[Privilege]:
"""Fetch the user's privileges from the database."""
with db.cursor(conn) as cursor:
cursor.execute(
("SELECT p.privilege_id, p.privilege_description "
"FROM user_roles AS ur "
"INNER JOIN role_privileges AS rp ON ur.role_id=rp.role_id "
"INNER JOIN privileges AS p ON rp.privilege_id=p.privilege_id "
"WHERE ur.user_id=?"),
(str(user.user_id),))
results = cursor.fetchall()
return (Privilege(row[0], row[1]) for row in results)
|
