Server-py/sri-pysdk/xclient/xssh.py

# update: 2021-3-26-13
"""
https://www.jianshu.com/p/512a5641b501

# --- 上传脚本 ---
sftp = s.open_sftp()
sftp.put('../test.sh', '/data/test.sh')
sftp.close()

# --- 执行脚本并删除 ---

docker exec zelu_8891 bash -c "python /home/server/projects/python-admin/zelu/libraries/protocol_l4/ssh/client.py"
"""
import paramiko


def test():
    host_ip = '192.168.30.13'
    username = 'fish'
    password = 'admin'

    client = paramiko.SSHClient()
    try:

        # --- password login ---
        client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        client.connect(hostname=host_ip, port=22, username=username, password=password)

        # --- test ---
        stdin, stdout, stderr = client.exec_command('ls -all /home')
        # stdin, stdout, stderr = client.exec_command('ls /home/ShellScripts')
        print('--- --- --- ---')
        print(stdout.read().decode('utf-8'))

    except Exception as e:
        print(e)
    finally:
        client.close()


class Client(paramiko.SSHClient):

    def __init__(self, ipv4='172.18.0.1', port=22, username='fish', password='admin'):

        super().__init__()
        self.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        self.connect(hostname=ipv4, port=port, username=username, password=password)

    def run_script(self, script_path, host_ip, username, password, command='sh /root/1.sh && cat /home/out.txt'):
        """执行上传脚本"""
        # script_path = '/home/server/projects/user-dashboard/operation/builtin_script/linux基线检查脚本.sh'
        # script_path = r'D:\share\gitee\secdeer.user-dashboard-online\operation\builtin_script\linux基线检查脚本.sh'
        # host_ip = '47.104.160.37'
        # username = 'root'
        # password = 'Secdeer_01!'
        self.connect(hostname=host_ip, port=22, username=username, password=password)

        sftp = self.open_sftp()
        sftp.put(script_path, '/root/1.sh')
        sftp.close()

        # stdin, stdout, stderr = self.exec_command('ls -all /home')
        # stdin, stdout, stderr = self.exec_command('sh /root/1.sh')
        # stdin, stdout, stderr = self.exec_command('cat /home/out.txt')
        stdin, stdout, stderr = self.exec_command(command)
        output = stdout.read().decode('utf-8')
        print(output)
        print('--- --- --- ---')
        return output

    def run_command(self, command):
        """执行命令"""
        try:
            stdin, stdout, stderr = self.exec_command(command)
            output = stdout.read().decode('utf-8')
            return output
        except Exception as exception:
            import traceback
            print(f"SSHClient.run_command.exception: {exception.__class__.__name__}")
            print(f"SSHClient.run_command.traceback: {traceback.format_exc()}")

    # finally:
    # 	self.close()

    def run_sudo_command(self, command, password):
        """执行命令"""
        try:
            stdin, stdout, stderr = self.exec_command(f"echo \"{password}\" | sudo -S {command}")
            output = stdout.read().decode('utf-8')
            return output
        except Exception as exception:
            import traceback
            print(f"SSHClient.run_sudo_command.exception: {exception.__class__.__name__}")
            print(f"SSHClient.run_sudo_command.traceback: {traceback.format_exc()}")

    # finally:
    # 	self.close()

    def get_disk_utilization_rate(self, path='/'):
        """获取根目录磁盘所用空间占比"""
        try:
            stdin, stdout, stderr = self.exec_command("df -h")
            output = stdout.read().decode('utf-8')

            # --- get utilization_rate by '/' ---
            utilization_rate = str()
            for row in output.split('\n'):
                if not row:
                    continue
                if row[-1] != '/':
                    continue
                for val in row.split(' '):
                    if '%' in val:
                        utilization_rate = val
                        break

            # --- transform ---
            if '%' in utilization_rate:
                utilization_rate = int(utilization_rate.strip('%'))
            else:
                utilization_rate = 0
            return utilization_rate

        except Exception as exception:
            import traceback
            print(f"SSHClient.show_root_disk.exception: {exception.__class__.__name__}")
            print(f"SSHClient.show_root_disk.traceback: {traceback.format_exc()}")

    def close(self):
        self.close()


if __name__ == '__main__':
    # --- init ---
    # ssh = Client('192.168.30.13', 22, 'fish', 'admin')
    # ssh = Client('172.18.0.1', 22, 'fish', 'admin')
    # ssh = Client('192.168.1.45', 22, 'server', 'server')
    # ssh = Client('192.168.30.13', 22, 'server', 'server')
    ssh = Client('10.10.10.89', 22, 'nvidia', '123456')

    # --- test ---
    out = ssh.get_disk_utilization_rate()
    print(out)

    # --- test ---
    # out = ssh.run_command('date +%Y-%m-%d-%H-%M-%S')
    # date +"%Y-%m-%d %H:%M:%S"
    # print(out)

    # --- test ---
    # ssh.run_sudo_command('echo -e "\n#x1" >> /etc/network/interfaces', 'admin')
    # ssh.run_sudo_command('echo -e "\n#x1" >> /etc/network/interfaces', 'server')
    # ssh.run_command('echo -e "\n#x2" >> /etc/network/interfaces')
    # ssh.run_command('sudo echo -e "\n#x3" >> /etc/network/interfaces')
    # out = ssh.run_command("sudo chmod 777 /etc/network/interfaces")
    # out = ssh.run_command("cat /etc/network/interfaces")
    # print(out)

    # --- test ---
    # out = ssh.run_command("ifconfig")
    # for row in out.split('\n\n'):
    # 	if row[:4] != 'eth0':
    # 		continue
    # 	for one in row.split('\n'):
    # 		one = one.strip()
    # 		if one[:4] != 'inet':
    # 			continue
    # 		if one[:5] == 'inet6':
    # 			continue
    # 		one = [i for i in one.split(' ') if i]
    # 		ipv4, netmask = one[1], one[3]
    # 		print(ipv4, netmask)

    # --- test ---
    # out = ssh.run_command('ls /home/server/images')
    # out = ssh.run_command('rm -rf /home/server/images/*')
    # out = ssh.run_sudo_command('rm -rf /home/server/images/*', 'admin')
    # print(out)

    # --- test ---
    # out0 = ssh.run_sudo_command('chmod 777 /etc/network/interfaces', 'admin')
    # out1 = ssh.run_command('echo -e "\n#x2" >> /etc/network/interfaces')
    # out2 = ssh.run_command("cat /etc/network/interfaces")
    # print(out2)

    # --- test ---
    # out = c.run_command("ls -all /home")
    # out = c.run_command("df -h | grep 234G | awk \'{print $5}\'")
    # out = c.run_command("df -h | grep \"/dev/vda1\" | awk \'{print $5}\'")
    # out = c.run_command("df -h ")
    # print(out)

    # --- test ---
    # out = c.run_script(1, 2, 3, 4)
    # print(repr(out))

    # --- test ---
    # out = c.run_script('/home/UserFiles/hello.sh', '118.190.217.96', 'root', 'Secdeer_01!', 'sh /root/1.sh')
    # print(repr(out))