import os
import paramiko
import time
import xlrd
import pandas as pd


class ServerSsh:

    def __int__(self):
        pass

    def ssh_server(self, ip, user, user_passwd):
        """
        远程服务器模块
        :param ip: 服务器ip
        :param user: 登录用户
        :param user_passwd: 密码
        :return: 返回登录会话对象
        """
        ssh_client = paramiko.SSHClient()
        ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        try:
            ssh_client.connect(hostname=ip, port="22", username=user, password=user_passwd)
            return ssh_client
        except Exception as e:
            print(ip + '密码错误，登录失败')
            ssh_client.close()
            return '登录失败'

    def read_xsxl(self, file_path, sheet_name):
        """
        读excel文件，返回sheet页内容
        :param file_path: 文件路径
        :param sheet_name: sheet名称
        :return: 返回excel结果
        """
        try:
            df = pd.read_excel(file_path, sheet_name=sheet_name)
            return df
        except Exception as e:
            return "文件读取失败"

    def ssh_cmd(self, server_ip, user, user_passwd, cmd, root_passwd):
        """
        执行命令
        :param server_ip: 服务器ip
        :param user: 登录用户
        :param user_passwd: 登录密码
        :param cmd: 执行命令
        :param root_passwd: root密码
        :return:返回执行结果
        """
        ssh_client = self.ssh_server(server_ip, user, user_passwd)
        if ssh_client != '登录失败':
            if user != 'root':
                ssh = ssh_client.invoke_shell()
                time.sleep(0.1)
                # 切换root用户
                ssh.send('su - root\n')
                buff = ''
                # 拼出返回结果
                while not buff.endswith('Password: '):
                    resp = ssh.recv(9999)
                    buff += resp.decode('utf-8')
                ssh.send(root_passwd + '\n')
                # 拼出返回结果
                buff = ''
                while "root@" not in buff:
                    resp = ssh.recv(9999)
                    buff += resp.decode('utf-8')
                # 拼出返回结果
                ssh.send(cmd + '\n')
                buff = ''
                while "root@" not in buff:
                    resp = ssh.recv(9999)
                    buff += resp.decode('utf-8')
                result = buff
                ssh.close()
                return result
            else:
                stdin, stdout, stderr = ssh_client.exec_command(cmd)
                result = stdout.read()
            ssh_client.close()
            return result
        else:
            return server_ip +'命令执行失败'


if __name__ == '__main__':
    a = ServerSsh()
    flie_path = 'C:\\Users\\admin\\PycharmProjects\\pythonProject\\venv\\Scripts\\zznode\\serveripinfo.xlsx'
    sheet_name = 'Sheet1'
    user = 'test'
    passwd = '123456'
    cmd = 'touch 1.txt'
    root_passwd = '12345678'
    if '文件读取失败' == date_obj:
        print('文件读取失败')
    else:
        date_obj = a.read_xsxl(flie_path, sheet_name)
        for ip in date_obj['服务器信息']:
            result = a.ssh_cmd(server_ip=ip, user=user, user_passwd=passwd, cmd=cmd, root_passwd=root_passwd)
            print(ip + result)