Merge pull request #199 from britive/fix/click-exception-aws-process

fix/click-exception-aws-process

Author: theborch

CHANGELOG.md

@@ -3,6 +3,28 @@
 > As of v1.4.0, release candidates will be published in an effort to get new features out faster while still allowing
 > time for full QA testing before moving the release candidate to a full release.
 
+## v2.1.3 [2025-03-31]
+
+__What's New:__
+
+* None
+
+__Enhancements:__
+
+* None
+
+__Bug Fixes:__
+
+* Fixed missing catch for `ClickException` in `pybritive-aws-cred-process` after previous catch update change in `9bf6738f`
+
+__Dependencies:__
+
+* None
+
+__Other:__
+
+* None
+
 ## v2.1.2 [2025-03-14]
 
 __What's New:__

src/pybritive/__init__.py

@@ -1 +1 @@
-__version__ = '2.1.2'
+__version__ = '2.1.3'

src/pybritive/helpers/aws_credential_process.py

@@ -4,6 +4,8 @@
 import sys
 from sys import argv
 
+from click.exceptions import ClickException
+
 
 def _fallback_input(prompt='', stream=None):
     if not stream:
@@ -231,7 +233,9 @@ def main():
 
         try:
             perform_checkout(b, args)
-        except exceptions.StepUpAuthRequiredButNotProvided:
+        except (exceptions.StepUpAuthRequiredButNotProvided, ClickException) as e:
+            if 'step up authentication required' not in str(e).lower():
+                raise SystemExit(e) from e
             perform_checkout(b, args, otp=get_input(prompt='(pybritive) Enter OTP:'))
         except (
             exceptions.ApprovalRequiredButNoJustificationProvided,