Marco Ricci commited on 2024-06-30 16:37:52
Zeige 1 geänderte Dateien mit 8 Einfügungen und 8 Löschungen.
| ... | ... |
@@ -134,11 +134,11 @@ def _get_suitable_ssh_keys( |
| 134 | 134 |
passphrase derivation. |
| 135 | 135 |
|
| 136 | 136 |
Raises: |
| 137 |
- RuntimeError: |
|
| 138 |
- There was an error communicating with the SSH agent. |
|
| 139 |
- RuntimeError: |
|
| 137 |
+ LookupError: |
|
| 140 | 138 |
No keys usable for passphrase derivation are loaded into the |
| 141 | 139 |
SSH agent. |
| 140 |
+ RuntimeError: |
|
| 141 |
+ There was an error communicating with the SSH agent. |
|
| 142 | 142 |
|
| 143 | 143 |
""" |
| 144 | 144 |
client: ssh_agent_client.SSHAgentClient |
| ... | ... |
@@ -166,7 +166,7 @@ def _get_suitable_ssh_keys( |
| 166 | 166 |
if dpp.Vault._is_suitable_ssh_key(key): |
| 167 | 167 |
yield pair |
| 168 | 168 |
if not suitable_keys: # pragma: no cover |
| 169 |
- raise RuntimeError('No usable SSH keys were found')
|
|
| 169 |
+ raise IndexError('No usable SSH keys were found')
|
|
| 170 | 170 |
|
| 171 | 171 |
|
| 172 | 172 |
def _prompt_for_selection( |
| ... | ... |
@@ -263,11 +263,11 @@ def _select_ssh_key( |
| 263 | 263 |
IndexError: |
| 264 | 264 |
The user made an invalid or empty selection, or requested an |
| 265 | 265 |
abort. |
| 266 |
- RuntimeError: |
|
| 267 |
- There was an error communicating with the SSH agent. |
|
| 268 |
- RuntimeError: |
|
| 266 |
+ LookupError: |
|
| 269 | 267 |
No keys usable for passphrase derivation are loaded into the |
| 270 | 268 |
SSH agent. |
| 269 |
+ RuntimeError: |
|
| 270 |
+ There was an error communicating with the SSH agent. |
|
| 271 | 271 |
""" |
| 272 | 272 |
suitable_keys = list(_get_suitable_ssh_keys(conn)) |
| 273 | 273 |
key_listing: list[str] = [] |
| ... | ... |
@@ -825,7 +825,7 @@ def derivepassphrase( |
| 825 | 825 |
_select_ssh_key()).decode('ASCII')
|
| 826 | 826 |
except IndexError: |
| 827 | 827 |
ctx.fail('no valid SSH key selected')
|
| 828 |
- except RuntimeError as e: |
|
| 828 |
+ except (LookupError, RuntimeError) as e: |
|
| 829 | 829 |
ctx.fail(str(e)) |
| 830 | 830 |
elif use_phrase: |
| 831 | 831 |
maybe_phrase = _prompt_for_passphrase() |
| 832 | 832 |