136 lines
4.8 KiB
Python
136 lines
4.8 KiB
Python
import os
|
||
import sys
|
||
import re
|
||
import datetime
|
||
import paramiko
|
||
import shutil
|
||
from paramiko import channel
|
||
|
||
host_ip = "10.0.1.111"
|
||
host_name = "daxiong"
|
||
host_passwd = "5555"
|
||
host_cmd = """
|
||
export PATH=$PATH:/opt/riscv_env/bin/;
|
||
cd share/kunlun-private/iotelic/kunlun/Mainline/build/;
|
||
git checkout master_dtest;
|
||
./build_dtest_kl3.sh
|
||
"""
|
||
# 这里不适用“git rev-parse HEAD”，获取到的是本地合并之后的commit id，与库上的不符合，
|
||
# 需要抽取merge id，这里考虑是内部邮件，直接打印近两天的提交
|
||
host_cmd_commit_id = """
|
||
cd share/kunlun-private/iotelic/kunlun/Mainline/build/;
|
||
git log --since=2.days
|
||
"""
|
||
# 目前仅拷贝xxx.bin文件，
|
||
# 约定拷贝的文件存放于脚本所在目录下的dtest_file目录下，以单独的bin文件存在
|
||
path_dtest_kl3 = "/home/daxiong/share/kunlun-private/iotelic/kunlun/Mainline/dtest/dtest3/"
|
||
path_dtest_build = "/.output/lib/"
|
||
list_dtest_kl3 = [
|
||
"unit_test_sfc",
|
||
"unit_test_smc",
|
||
]
|
||
|
||
# -------------------------------建立ssh连接------------------------------------
|
||
print("开始ssh连接服务器")
|
||
# 建立一个sshclient对象
|
||
ssh = paramiko.SSHClient()
|
||
ssh.load_system_host_keys()
|
||
try:
|
||
# 允许将信任的主机自动加入到host_allow 列表，此方法必须放在connect方法的前面
|
||
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
# 调用connect方法连接服务器
|
||
ssh.connect(hostname=host_ip, port=22, username=host_name,
|
||
password=host_passwd, timeout=5)
|
||
except paramiko.SSHException as e_ssh:
|
||
print(e_ssh)
|
||
print("%s@%s SSH连接失败" % (host_ip, host_name))
|
||
sys.exit(1)
|
||
except Exception as e:
|
||
print(e)
|
||
print("%s@%s 连接失败" % (host_ip, host_name))
|
||
sys.exit(2)
|
||
print("%s@%s 连接成功" % (host_ip, host_name))
|
||
|
||
# -------------------------------远程执行命令-----------------------------------
|
||
print("开始执行指令：", host_cmd)
|
||
try:
|
||
build_fail = 0
|
||
# 执行命令,结果放到stdout中，如果有错误将放到stderr中
|
||
stdin, stdout, stderr = ssh.exec_command(host_cmd, get_pty=False)
|
||
for line in iter(stdout.readline,""):
|
||
if re.search("failed", line):
|
||
build_fail = 1
|
||
print(line, end="")
|
||
except paramiko.SSHException as e_ssh:
|
||
print(e_ssh)
|
||
print("命令[%s]执行失败" % (host_cmd))
|
||
sys.exit(3)
|
||
except Exception as e:
|
||
print(e)
|
||
print("命令[%s]执行异常" % (host_cmd))
|
||
sys.exit(4)
|
||
if build_fail:
|
||
print("固件编译失败")
|
||
sys.exit(5)
|
||
print("固件编译成功")
|
||
|
||
# -------------------------------拷贝编译文件-----------------------------------
|
||
print("开始拷贝文件")
|
||
port_trans = ssh.get_transport()
|
||
port_sftp = paramiko.SFTPClient.from_transport(port_trans)
|
||
# 目前仅拷贝xxx.bin文件
|
||
path_pwd = os.path.split(os.path.realpath(__file__))[0]
|
||
dir_dtest = os.path.join(path_pwd, "dtest_file")
|
||
if os.path.exists(dir_dtest):
|
||
shutil.rmtree(dir_dtest)
|
||
os.makedirs(dir_dtest)
|
||
timestamp = datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
|
||
for dtest in list_dtest_kl3:
|
||
path_remote = path_dtest_kl3 + dtest + path_dtest_build + dtest + ".bin"
|
||
dir_local = os.path.join(path_pwd, "history", timestamp, dtest)
|
||
if not os.path.exists(dir_local):
|
||
os.makedirs(dir_local)
|
||
path_local = os.path.join(dir_local, dtest) + ".bin"
|
||
print("remote path:", path_remote)
|
||
print("local path:", path_local)
|
||
try:
|
||
port_sftp.get(path_remote, path_local)
|
||
except Exception as e:
|
||
print("文件拷贝失败", e)
|
||
sys.exit(6)
|
||
shutil.copy(path_local, dir_dtest)
|
||
|
||
# -------------------------------远程执行命令,获取commit id----------------------
|
||
print("开始执行指令：", host_cmd_commit_id)
|
||
info_commit = ""
|
||
try:
|
||
build_fail = 0
|
||
# 执行命令,结果放到stdout中，如果有错误将放到stderr中
|
||
stdin, stdout, stderr = ssh.exec_command(host_cmd_commit_id, get_pty=False)
|
||
for line in iter(stdout.readline,""):
|
||
info_commit += line
|
||
print("获取到commit id：", info_commit)
|
||
except paramiko.SSHException as e_ssh:
|
||
print(e_ssh)
|
||
print("命令[%s]执行失败" % (host_cmd_commit_id))
|
||
except Exception as e:
|
||
print(e)
|
||
print("命令[%s]执行异常" % (host_cmd_commit_id))
|
||
if build_fail:
|
||
print("获取commit id失败")
|
||
print("获取commit id成功")
|
||
# 写入文件，用于附加到测试结果邮件中
|
||
path_commit_id = os.path.join(os.getcwd(), "dtest_file", "commit_info.log")
|
||
try:
|
||
fd_commit = open(path_commit_id, "w")
|
||
except Exception as e:
|
||
print("文件 %s 打开失败，原因：" % (path_commit_id), e)
|
||
fd_commit.write(info_commit)
|
||
fd_commit.close()
|
||
|
||
# -------------------------------关闭连接----------------------------------------
|
||
# 关闭连接
|
||
ssh.close()
|
||
print("脚本执行完成")
|
||
sys.exit(0)
|