main: refactor shell flag

8 years ago · 2b047f0525
parent 64776fd294
commit 2b047f0525
1 changed files with 3 additions and 3 deletions
--- a/trezor_agent/main.py
+++ b/trezor_agent/main.py
@ -106,13 +106,13 @@ def run_agent(client_factory):

        public_key = conn.get_public_key(label=label)

-        use_shell = False
        if args.connect:
            command = ssh_args(label) + args.command
            log.debug('SSH connect: %r', command)

-        if args.shell:
-            command, use_shell = os.environ['SHELL'], True
+        use_shell = bool(args.shell)
+        if use_shell:
+            command = os.environ['SHELL']
            log.debug('using shell: %r', command)

        if not command: