arg0

cd/__init__.py

@@ -86,8 +86,8 @@ def install_commands(path):
         os.symlink(path, os.path.join(user, f'{COMMANDS_PREFIX}{command}'))
 
 
-def set_ssh():
-    ssh_hosts = projects.environment.get('CD_SSH_HOSTS', None)
+def set_ssh(*arguments):
+    ssh_hosts = arguments[0]
     ssh_key = projects.environment.get('CD', None)
     ssh_type = projects.environment.get('CD_SSH_TYPE', 'ed25519')
     #

cd/__main__.py

@@ -10,7 +10,7 @@ if __name__ == '__main__':
     command, *arguments = sys.argv
     command = os.path.basename(command)
     if command == '__main__.py':
-        cd.set_ssh()
+        cd.set_ssh(*arguments)
         cd.install_commands(__file__)
     else:
         command = command.replace('-', '_')