Guest User

Untitled

a guest
Dec 18th, 2018
68
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.89 KB | None | 0 0
  1. node_list = {'cn000': 28, 'cn001': 28}
  2. children, pid = [], 0
  3. for node, ntasks in nnode_list.items():
  4. pid_range = range(pid, pid + ntasks)
  5. pid += ntasks
  6. for worker in pid_range:
  7. chile = pexpect.spawn(worker_command % worker)
  8. children.append(child)
  9.  
  10. from pexpect import pxssh
  11. from socket import gethostname
  12. from getpass import getuser
  13. connections, pid = [], 0
  14. for node, ntasks in node_list.items():
  15. pid_range = range(pid, pid + ntasks)
  16. pid += ntasks
  17. if node == gethostname():
  18. children = []
  19. for worker in pid_range:
  20. child = pexpect.spawn(worker_command % worker)
  21. children.append(child)
  22. connections.append(children)
  23. if node != gethostname():
  24. ssh = pxssh.pxssh()
  25. ssh.login(node, getuser())
  26. for worker in pid_range:
  27. ssh.sendline(worker_command % worker)
  28. connections.append(ssh)
Add Comment
Please, Sign In to add comment