trezor_agent: use list comprehension

9 years ago · d0c5421a3d
parent 14b3b500cd
commit d0c5421a3d
1 changed files with 1 additions and 4 deletions
--- a/sshagent/trezor_agent.py
+++ b/sshagent/trezor_agent.py
@ -32,10 +32,7 @@ def main():
    logging.basicConfig(level=loglevel, format=fmt)

    with trezor.Client(factory=trezor.TrezorLibrary) as client:
-        public_keys = []
-        for label in args.identity:
-            ssh_public_key = client.get_public_key(label)
-            public_keys.append(ssh_public_key)
+        public_keys = [client.get_public_key(i) for i in args.identity]

        command = args.command
        if not command: