Python远程控制连接Linux操作(替代Jenkins)

import paramiko
import getpass

# 远程设备的IP 端口 用户名 密码
IP = "10.8.12.45"
port = 22
remote_username = "uos-0804"
remoet_password = "1"
# 包名
package_name = "0811.zip"
# 本机用户名
my_username = getpass.getuser()

# 建立连接
transport = paramiko.Transport((IP, port))
transport.connect(username=remote_username, password=remoet_password)

# 创建SSH对象
ssh = paramiko.SSHClient()
ssh._transport = transport


# 执行命令
def cmd(doit):
    # ssh.exec_command(doit)
    stdin, stdout, stderr = ssh.exec_command(doit)
    back_word = stdout.read().decode()
    return back_word


# 上传文件
def put_file():
    sftp = paramiko.SFTPClient.from_transport(transport)
    # from my_path to remote_path
    my_path = "/home/%s/Documents/%s" % (my_username, package_name)
    remote_path = "/home/%s/Documents/%s" % (remote_username, package_name)
    sftp.put(my_path, remote_path)


# 解压
def unzip():
    cmd("cd /home/%s/Documents/ && unzip %s" % (remote_username, package_name))


# 执行测试
def exec_test():
    cmd("cd /home/%s/Documents/media_automation_test/report/ && bash start_runner.sh" % remote_username)


# 关闭连接
def close():
    ssh.close()



def delete_package():
    doit = "cd /home/%s/Documents/ && rm -rf %s" % (remote_username, package_name)
    stdin, stdout, stderr = ssh.exec_command(doit)
    #stdin.write(remoet_password + r'
')
    #stdin.flush()

def delete_folder():
    doit = "cd /home/%s/Documents/ && rm -rf media_automation_test" % remote_username
    stdin, stdout, stderr = ssh.exec_command(doit, get_pty=True)
    #stdin.write(remoet_password + r'
')
    #stdin.flush()


# 上传项目包
put_file()
# 解压
unzip()
# 执行测试
exec_test()
# 删除项目包
delete_package()
# 删除项目文件夹
delete_folder()


close()

遇到的问题：

在远程批量执行测试用例的时候，测试用例执行失败，报错：KeyError："DISPLAY"

解决1：在python脚本里面添加  

import os 
os.environ[ "DISPLAY" ] = ":0"

 解决2：在shell脚本里面添加

export DISPLAY=:0