aboutsummaryrefslogtreecommitdiff
path: root/gn3/authentication.py
diff options
context:
space:
mode:
Diffstat (limited to 'gn3/authentication.py')
-rw-r--r--gn3/authentication.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/gn3/authentication.py b/gn3/authentication.py
index 892aa8f..7bc7b77 100644
--- a/gn3/authentication.py
+++ b/gn3/authentication.py
@@ -3,9 +3,9 @@ import functools
import json
from urllib.parse import urljoin
from enum import Enum, unique
-from typing import Dict
+from typing import Dict, Union
-import redis
+from redis import Redis
import requests
@@ -40,7 +40,7 @@ class AdminRole(OrderedEnum):
EDIT_ADMINS = "edit-admins"
-def get_user_membership(conn: redis.Redis, user_id: str,
+def get_user_membership(conn: Redis, user_id: str,
group_id: str) -> Dict:
"""Return a dictionary that indicates whether the `user_id` is a
member or admin of `group_id`.
@@ -84,7 +84,7 @@ def get_highest_user_access_role(
A dict indicating the highest access role the user has.
"""
- role_mapping = {}
+ role_mapping: Dict[str, Union[DataRole, AdminRole]] = {}
for data_role, admin_role in zip(DataRole, AdminRole):
role_mapping.update({data_role.value: data_role, })
role_mapping.update({admin_role.value: admin_role, })