main: Ask for login when required

Signed-off-by: Mathieu Dubois-Briand <[email protected]>

Author: mbriand

swattool/main.py

@@ -64,12 +64,28 @@ def parse_filters(kwargs) -> dict[str, Any]:
 
 @click.group()
 @click.option('-v', '--verbose', count=True, help="Increase verbosity")
-def main(verbose: int):
+def maingroup(verbose: int):
     """Handle triage of Yocto autobuilder failures."""
     utils.setup_logging(verbose)
 
 
-@main.command()
+def main():
+    """Handle triage of Yocto autobuilder failures."""
+    try:
+        maingroup()  # pylint: disable=no-value-for-parameter
+    except utils.LoginRequiredException as e:
+        if e.service == "swatbot":
+            logger.warning("Login required to swatbot server")
+            user = click.prompt('swatbot user')
+            password = click.prompt('swatbot password', hide_input=True)
+            success = swatbotrest.login(user, password)
+            if success:
+                maingroup()  # pylint: disable=no-value-for-parameter
+        else:
+            raise
+
+
+@maingroup.command()
 @click.option('--user', '-u', prompt=True)
 @click.option('--password', '-p', prompt=True, hide_input=True)
 def login(user: str, password: str):
@@ -145,7 +161,7 @@ def format_field(build, userinfo, field):
     return (table, headers)
 
 
-@main.command()
+@maingroup.command()
 @_add_options(failures_list_options)
 @click.option('--open-url', '-u', is_flag=True,
               help="Open the autobuilder url in web browser")
@@ -194,7 +210,7 @@ def show_pending_failures(refresh: str, open_url: str,
                       if b.status == swatbuild.Status.CANCELLED]))
 
 
-@main.command()
+@maingroup.command()
 @_add_options(failures_list_options)
 @click.option('--open-autobuilder-url', '-u', is_flag=True,
               help="Open the autobuilder url in web browser")
@@ -235,7 +251,7 @@ def review_pending_failures(refresh: str, open_autobuilder_url: bool,
     userinfos.save()
 
 
-@main.command()
+@maingroup.command()
 @click.option('--dry-run', '-n', is_flag=True,
               help="Only shows what would be done")
 def publish_new_reviews(dry_run: bool):

swattool/swatbotrest.py

@@ -100,7 +100,7 @@ def _get_csrftoken() -> str:
     return Session().session.cookies['csrftoken']
 
 
-def login(user: str, password: str):
+def login(user: str, password: str) -> bool:
     """Login to the swatbot Django interface."""
     session = Session()
 
@@ -120,11 +120,13 @@ def login(user: str, password: str):
             raise error
     else:
         logger.warning("Unexpected reply, login probably failed")
-        return
+        return False
 
     session.save_cookies()
     logger.info("Logging success")
 
+    return True
+
 
 def _get_json(path: str, max_cache_age: int = -1):
     data = Session().get(f"{REST_BASE_URL}{path}", max_cache_age)
@@ -133,7 +135,8 @@ def _get_json(path: str, max_cache_age: int = -1):
     except json.decoder.JSONDecodeError as err:
         Session().invalidate_cache(f"{REST_BASE_URL}{path}")
         if "Please login to see this page." in data:
-            raise utils.SwattoolException("Not logged in swatbot") from err
+            raise utils.LoginRequiredException("Not logged in swatbot",
+                                               "swatbot") from err
         raise utils.SwattoolException("Failed to parse server reply") from err
     return json_data
 

swattool/utils.py

@@ -60,6 +60,15 @@ class SwattoolException(Exception):
     """A generic swattool error."""
 
 
+class LoginRequiredException(SwattoolException):
+    """An exception for operations requiring login."""
+
+    def __init__(self, message, service):
+        super().__init__(message)
+
+        self.service = service
+
+
 class _LogFormatter(logging.Formatter):
     colors = {
         logging.DEBUG: Color.WHITE,