imp ssh control master connector
| 21 | |
| 22 | |
| 23 | class SshConnector: |
| 24 | """imp ssh control master connector""" |
| 25 | |
| 26 | ip = None |
| 27 | port = None |
| 28 | login_name = None |
| 29 | |
| 30 | def setup(self, login_name, ip, port): |
| 31 | self.ip = ip |
| 32 | self.login_name = login_name |
| 33 | self.port = port |
| 34 | |
| 35 | def copy(self, src_list, dst_dir): |
| 36 | assert isinstance(src_list, list), "code issue happened!!" |
| 37 | assert isinstance(dst_dir, str), "code issue happened!!" |
| 38 | for src in src_list: |
| 39 | cmd = 'rsync --progress -a -e "ssh -p {}" {} {}@{}:{}'.format( |
| 40 | self.port, src, self.login_name, self.ip, dst_dir |
| 41 | ) |
| 42 | logging.debug("ssh run cmd: {}".format(cmd)) |
| 43 | subprocess.check_call(cmd, shell=True) |
| 44 | |
| 45 | def cmd(self, cmd): |
| 46 | output = "" |
| 47 | assert isinstance(cmd, list), "code issue happened!!" |
| 48 | for sub_cmd in cmd: |
| 49 | p_cmd = 'ssh -p {} {}@{} "{}" '.format( |
| 50 | self.port, self.login_name, self.ip, sub_cmd |
| 51 | ) |
| 52 | logging.debug("ssh run cmd: {}".format(p_cmd)) |
| 53 | output = output + subprocess.check_output(p_cmd, shell=True).decode("utf-8") |
| 54 | return output |
| 55 | |
| 56 | |
| 57 | def get_finally_bench_resulut_from_log(raw_log) -> float: |