class Job: def __init__(self, hostname, port, username, pkey, command, token=None, **kwargs): self.ssh_cli = SSH(hostname, port, username, pkey) self.key = f'{hostname}:{port}' self.command = command self.token = token self.rds_cli = None def _send(self, message, with_expire=False): if self.rds_cli is None: self.rds_cli = get_redis_connection() self.rds_cli.lpush(self.token, json.dumps(message)) if with_expire: self.rds_cli.expire(self.token, 300) def send(self, data): message = {'key': self.key, 'type': 'info', 'data': data} self._send(message) def send_system(self, data): message = {'key': self.key, 'type': 'system', 'data': data} self._send(message) def send_error(self, data): message = {'key': self.key, 'type': 'error', 'data': data} self._send(message) def send_status(self, code): message = {'key': self.key, 'status': code} self._send(message, True) def run(self): if not self.token: return self.ssh_cli.exec_command(self.command) self.send_system('### Executing') code = -1 try: for code, out in self.ssh_cli.exec_command_with_stream( self.command): self.send(out) except socket.timeout: code = 130 self.send_error('### Time out') except Exception as e: code = 131 self.send_error(f'{e}') finally: self.send_status(code)
class Job: def __init__(self, key, name, hostname, port, username, pkey, command, interpreter, token=None): self.ssh = SSH(hostname, port, username, pkey) self.key = key self.command = self._handle_command(command, interpreter) self.token = token self.rds_cli = None self.env = dict(SPUG_HOST_ID=str(self.key), SPUG_HOST_NAME=name, SPUG_HOST_HOSTNAME=hostname, SPUG_SSH_PORT=str(port), SPUG_SSH_USERNAME=username, SPUG_INTERPRETER=interpreter) def _send(self, message, with_expire=False): if self.rds_cli is None: self.rds_cli = get_redis_connection() self.rds_cli.lpush(self.token, json.dumps(message)) if with_expire: self.rds_cli.expire(self.token, 300) def _handle_command(self, command, interpreter): if interpreter == 'python': attach = 'INTERPRETER=python\ncommand -v python3 &> /dev/null && INTERPRETER=python3' return f'{attach}\n$INTERPRETER << EOF\n# -*- coding: UTF-8 -*-\n{command}\nEOF' return command def send(self, data): message = {'key': self.key, 'data': data} self._send(message) def send_status(self, code): message = {'key': self.key, 'status': code} self._send(message, True) def run(self): if not self.token: with self.ssh: return self.ssh.exec_command(self.command, self.env) self.send('\r\n\x1b[36m### Executing ...\x1b[0m\r\n') code = -1 try: with self.ssh: for code, out in self.ssh.exec_command_with_stream( self.command, self.env): self.send(out) except socket.timeout: code = 130 self.send('\r\n\x1b[31m### Time out\x1b[0m') except Exception as e: code = 131 self.send(f'\r\n\x1b[31m### Exception {e}\x1b[0m') raise e finally: self.send_status(code)