B t\@sddlmZmZmZddlmZddlZddlmZej ddej dde dd ej d d e d d ej d de dd ej dde dd ej dde dej ddejdddd ej dde dd ej dde dd ej dde dd ej d dd!d"ej d#dejddd$d ej d%de d&d ej d'd(d)d*ej d+e dd,d-ej d.e dd/d-ej d0de d1d ejd2d3Zd4d5Zed6kredS)7)print_functiondivisionabsolute_import) SSHClusterN)check_python_3aLaunch a distributed cluster over SSH. A 'dask-scheduler' process will run on the first host specified in [HOSTNAMES] or in the hostfile (unless --scheduler is specified explicitly). One or more 'dask-worker' processes will be run each host in [HOSTNAMES] or in the hostfile. Use command line flags to adjust how many dask-worker process are run on each host (--nprocs) and how many cpus are used by each dask-worker process (--nthreads).)helpz --schedulerz3Specify scheduler node. Defaults to first address.)defaulttyperz--scheduler-portiR"z6Specify scheduler port number. Defaults to port 8786.z --nthreadsznNumber of threads per worker process. Defaults to number of cores divided by the number of processes per host.z--nprocsz6Number of worker processes per host. Defaults to one. hostnames)nargsr z --hostfileT)existsz$Textfile with hostnames/IP addressesz--ssh-usernamez2Username to use when establishing SSH connections.z --ssh-portz Port to use for SSH connections.z--ssh-private-keyz,Private key file to use for SSH connections.z--nohostz'Do not pass the hostname to the worker.)Zis_flagrz--log-directoryz`Directory to use on all cluster nodes for the output of dask-scheduler and dask-worker commands.z--remote-pythonzPath to Python on remote nodes.z--memory-limitautozBytes of memory that the worker can use. This can be an integer (bytes), float (fraction of total system memory), string (like 5GB or 5000M), 'auto', or zero for no memory management)rrz --worker-portz,Serving computation port, defaults to random)r rrz --nanny-portz&Serving nanny port, defaults to randomz--remote-dask-workerz6Worker to run. Defaults to distributed.cli.dask_workercCs0yFt|}|r8t|}|}WdQRX|||sD|d}Wn(tk rnt|tdYnXt |||||||| | | | | |||}ddl }tdtdj |j dtdj t |dx&t|D]\}}tdj ||d qWtd j ||d td |td |tddS)Nrr z@ ---------------------------------------------------------------z- Dask.distributed v{version} )versionz Worker nodes:)nz {num}: {host})Znumhostz scheduler node: {addr}:{port})ZaddrZportzA--------------------------------------------------------------- zG [ dask-ssh ]: Shutting down remote processes (this may take a moment).z=[ dask-ssh ]: Remote processes have been terminated. Exiting.)listopenreadsplitextend IndexErrorprintZget_helpexitr distributedformat __version__len enumerateZmonitor_remote_processesZshutdown)ZctxZ schedulerZscheduler_portr ZhostfileZnthreadsZnprocsZ ssh_usernameZssh_portZssh_private_keyZnohostZ log_directoryZ remote_pythonZ memory_limitZ worker_portZ nanny_portZremote_dask_workerfZhostscrirr$7lib/python3.7/site-packages/distributed/cli/dask_ssh.pymain s4/       r&cCsttdS)N)rr&r$r$r$r%go\sr'__main__)Z __future__rrrZdistributed.deploy.sshrZclickZdistributed.cli.utilsrZcommandZoptionstrintZargumentZPathZ pass_contextr&r'__name__r$r$r$r%sP             0(