#!/usr/bin/python3

import paramiko

import os

import sys

import subprocess

curPath = os.path.abspath(os.path.dirname(__file__))

rootPath = os.path.split(curPath)[0]

sys.path.append(rootPath)

from src.logutils import logger

log=logger("root",rootstdout=True,handlerList=['I','E'])

class SshConnect(object):

    def __init__(self,ip,user,pwd,port):

        self.port=port

        self.pwd=pwd

        self.user=user

        self.ip=ip

    def get_connected_client(self):

        client=paramiko.SSHClient()

        try:

            client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

            client.connect(self.ip,self.port,self.user,self.pwd)

            if client:

                return client

        except ConnectionError as e:

            log.error(e)

    def exe_cmd(self,shell):

        client=self.get_connected_client()

        try:

            input, out, err =client.exec_command(shell)

            res,err = out.read(),err.read()

            result = res if res else err

            if type(result).__name__ == 'bytes':

                return result.decode('utf-8')

            elif type(result).__name__ == 'str':

                return result

        except Exception as e:

            log.error(e)

        finally:

            if client:

                client.close()

            else:

                log.info("ssh client is null not need close")

def run():

    ip='192.168.81.129'

    port=22

    pwd='admin'

    user='root'

    client=SshConnect(ip,user,pwd,port)

    res=client.exe_cmd('ps -ef|grep python&&df -m') #type should be str

    log.info(res)

# def deal_res(res):

#     std=''

#     for i  in res:

#         output=i.decode("utf-8")

#         std +=output

#     return std

if __name__ == '__main__':

    run()