@@ -33,7 +33,10 @@ def use_filestore_auth(user: User):
33
33
return None
34
34
verify , host = get_rest_host ()
35
35
user_login = get_user_login (user )
36
- return get_filestore_token (user_login , host , verify )
36
+ user_token , status = get_filestore_token (user_login , host , verify )
37
+ if not user_token :
38
+ return None
39
+ return user_token
37
40
38
41
39
42
def get_filestore_token (user_login , host , verify ):
@@ -43,8 +46,8 @@ def get_filestore_token(user_login, host, verify):
43
46
r_userlogin .raise_for_status ()
44
47
except (requests .exceptions .ConnectionError , requests .exceptions .HTTPError ) as err :
45
48
print ("{0}: " .format (err ))
46
- return None
47
- return r_userlogin .text
49
+ return None , r_userlogin . status_code
50
+ return r_userlogin .text , r_userlogin . status_code
48
51
49
52
50
53
def add_filestore_auth (user : User ):
@@ -53,7 +56,7 @@ def add_filestore_auth(user: User):
53
56
verify , host = get_rest_host ()
54
57
# get auth for setting new user
55
58
admin_login = get_admin_login ()
56
- admin_token = get_filestore_token (admin_login , host , verify )
59
+ admin_token , status = get_filestore_token (admin_login , host , verify )
57
60
if not admin_token :
58
61
return False
59
62
# get user defaults from global settings
@@ -90,19 +93,21 @@ def add_filestore_auth(user: User):
90
93
if user .is_staff : # admin and staff get root scope
91
94
set_filestore_scope (user )
92
95
93
- return use_filestore_auth (user )
96
+ fs_user_token = use_filestore_auth (user )
97
+ return fs_user_token
94
98
95
99
96
100
def set_filestore_scope (user : User ):
97
101
verify , host = get_rest_host ()
98
102
# get auth for setting new user
99
103
admin_login = get_admin_login ()
100
- admin_token = get_filestore_token (admin_login , host , verify )
104
+ admin_token , status = get_filestore_token (admin_login , host , verify )
101
105
if not admin_token :
102
106
return False
103
107
# find user
104
108
fs_user_token = use_filestore_auth (user )
105
- # TODO: dev1 jwt.exceptions.DecodeError: Invalid token type. Token must be a <class 'bytes'>
109
+ if not fs_user_token :
110
+ return False
106
111
payload = jwt .decode (fs_user_token , options = {"verify_signature" : False })
107
112
try :
108
113
r_user = requests .get (f"https://{ host } /storemng/api/users/{ payload ['user' ]['id' ]} " ,
@@ -142,11 +147,13 @@ def delete_filestore_user(user: User):
142
147
verify , host = get_rest_host ()
143
148
# get auth for removing user
144
149
admin_login = get_admin_login ()
145
- admin_token = get_filestore_token (admin_login , host , verify )
150
+ admin_token , status = get_filestore_token (admin_login , host , verify )
146
151
if not admin_token :
147
152
return False
148
153
# find user
149
154
fs_user_token = use_filestore_auth (user )
155
+ if not fs_user_token :
156
+ return False
150
157
payload = jwt .decode (fs_user_token , options = {"verify_signature" : False })
151
158
try :
152
159
r_user = requests .get (f"https://{ host } /storemng/api/users/{ payload ['user' ]['id' ]} " ,
0 commit comments