Fix token authentication bugs (#27)

Co-authored-by: Stephen Finucane <[email protected]>
getpatchwork · Nov 17, 2023 · 6120b39 · 6120b39
1 parent 88adfe2
commit 6120b39
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/pwclient/api.py b/pwclient/api.py
@@ -457,9 +457,8 @@ def _generate_headers(self, additional_headers=None):
         }
 
         if self._token:
-            pass
-
-        if self._username:
+            headers['Authorization'] = f"Token {self._token}"
+        elif self._username and self._password:
             credentials = base64.b64encode(
                 f'{self._username}:{self._password}'.encode('ascii')
             ).decode('ascii')

diff --git a/pwclient/shell.py b/pwclient/shell.py
@@ -92,9 +92,12 @@ def main(argv=sys.argv[1:]):
     if action in auth_actions:
         if backend == 'rest':
             if not (
-                config.has_option(project_str, 'username')
-                and config.has_option(project_str, 'password')
-            ) or config.has_option(project_str, 'token'):
+                (
+                    config.has_option(project_str, 'username')
+                    and config.has_option(project_str, 'password')
+                )
+                or config.has_option(project_str, 'token')
+            ):
                 sys.stderr.write(
                     "The %s action requires authentication, but no "
                     "username/password or\n"