import paramiko import pytest import psutil from common.abs_common import log @pytest.fixture(scope="module") def ssh_client(): print('-----ssh远程连接-------') ssh = paramiko.SSHClient() ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) ssh.connect('10.67.6.56', username='root', password='1') yield ssh ssh.close() print("获取PCIe设备的bus number,带宽和速度") def test_network_interface(ssh_client): stdin, stdout, stderr = ssh_client.exec_command("lspci|grep -i 'non' | cut -f1 -d ' '") interface = stdout.read().decode() stdin, stdout, stderr = ssh_client.exec_command(f"lspci -s {interface} -vvv |grep -i Lnkcap | cut -f2 -d ','") widthcap = stdout.read().decode().strip() outputwidthcap = stdout.read().decode("utf-8") stdin, stdout, stderr = ssh_client.exec_command(f"lspci -s {interface} -vvv |grep -i Lnksta | cut -f2 -d ','") widthsta = stdout.read().decode().strip() outputwidthsta = stdout.read().decode("utf-8") stdin, stdout, stderr = ssh_client.exec_command(f"lspci -s {interface} -vvv |grep -i Lnkcap | cut -f3 -d ','") speedcap = stdout.read().decode().strip() outputspeedcap = stdout.read().decode("utf-8") stdin, stdout, stderr = ssh_client.exec_command(f"lspci -s {interface} -vvv |grep -i Lnksta | cut -f3 -d ','") speedsta = stdout.read().decode().strip() outputspeedsta = stdout.read().decode("utf-8") print("比较lspci中PCIe设备的带宽和速度Lnkcap和Lnksta") assert widthcap == widthsta assert speedcap == speedsta print(outputwidthcap,outputwidthsta,outputspeedcap,outputspeedsta) if __name__ == '__main__': pytest.main(['-o log_cli=True --html=./test_get_nic_width.html -s'])
时间: 2024-04-03 18:32:59 浏览: 119
这段代码是用于测试获取PCIe设备的带宽和速度,其中使用了paramiko库建立远程SSH连接,执行了一些命令来获取相关信息。在test_network_interface函数中,首先使用lspci命令获取PCIe设备的bus number,然后通过grep和cut命令获取Lnkcap和Lnksta信息,分别表示PCIe设备的带宽和速度,最后通过assert语句比较这两个值是否相等。如果测试通过,会输出相关信息,并生成一个HTML格式的测试报告。
相关问题
from django.shortcuts import render import paramiko,time,os # import tkinter.messagebox def shutdown_remote_system(request): # 远程服务器的IP地址和SSH登录信息 remote_host = '192.168.0.106' remote_port = 22 remote_username = 'root' remote_password = '123' # 创建SSH连接 ssh = paramiko.SSHClient() ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) ssh.connect(remote_host, remote_port, remote_username, remote_password) # 执行关机命令 ssh.exec_command('sudo shutdown -h now') time.sleep(10) #判断系统是否已经关机 while True: response=os.system('ping -c 1'+remote_host) if response ==0: time.sleep(5) return render(request, 'index.html') else: break # 关闭SSH连接 ssh.close()
这段代码似乎是一个 Django 视图函数,当用户访问该视图时,会执行关机远程服务器的操作,并且在系统关机后返回一个 HTML 页面给用户。
在这段代码中,您使用了 Django 的 render 函数来渲染一个 HTML 模板,并返回给用户。但是,需要注意的是,在 while 循环中,只有在系统关机后才会返回 HTML 页面,因此用户可能需要等待较长时间,才能看到页面。另外,在这段代码中,您并没有处理用户可能在关机期间的其他请求,如果用户在系统关机期间发送了其他请求,这些请求可能会被阻塞,直到系统关机完成。
如果您想要实现一个更好的用户体验,可以考虑使用 Ajax 技术来实现异步请求和响应。具体实现方式可以参考下面的代码:
```
from django.http import JsonResponse
import paramiko,time,os
def shutdown_remote_system(request):
# 远程服务器的IP地址和SSH登录信息
remote_host = '192.168.0.106'
remote_port = 22
remote_username = 'root'
remote_password = '123'
# 创建SSH连接
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(remote_host, remote_port, remote_username, remote_password)
# 执行关机命令
ssh.exec_command('sudo shutdown -h now')
time.sleep(10)
#判断系统是否已经关机
while True:
response = os.system('ping -c 1 '+remote_host)
if response == 0:
time.sleep(5)
else:
return JsonResponse({'status': 'success'})
break
# 关闭SSH连接
ssh.close()
```
在这段代码中,我们使用了 Django 的 JsonResponse 函数来返回一个 JSON 格式的响应,而不是 HTML 页面。在前端页面中,可以使用 Ajax 技术定时向后端发送请求,检查系统是否已经关机,如果已经关机,则可以提示用户关机完成。通过这种方式,即使系统关机期间用户发送了其他请求,也不会被阻塞,从而提高了用户体验。
import paramiko def shutdown_remote_system(): # 远程服务器的IP地址和SSH登录信息 remote_host = '192.168.0.106' remote_port = 22 remote_username = 'root' remote_password = '123' # 创建SSH连接 ssh = paramiko.SSHClient() ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) ssh.connect(remote_host, remote_port, remote_username, remote_password) # 执行关机命令 ssh.exec_command('sudo shutdown -h now') # 关闭SSH连接 ssh.close() 缺少是否已关机判断
你的代码是远程执行关机命令的代码,但是确实缺少判断远程系统是否已经关机的部分。
你可以通过在远程服务器上执行一个命令,判断系统是否已经关机。例如,你可以通过 ping 命令测试远程服务器是否可达。
以下是一个示例代码,可以在远程服务器上执行指定的命令并判断系统是否已经关机:
```python
import paramiko
import time
def shutdown_remote_system():
# 远程服务器的IP地址和SSH登录信息
remote_host = '192.168.0.106'
remote_port = 22
remote_username = 'root'
remote_password = '123'
# 创建SSH连接
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(remote_host, remote_port, remote_username, remote_password)
# 执行关机命令,并等待一段时间
ssh.exec_command('sudo shutdown -h now')
time.sleep(10)
# 判断系统是否已经关机
while True:
response = os.system('ping -c 1 ' + remote_host)
if response == 0:
time.sleep(5)
else:
break
# 关闭SSH连接
ssh.close()
```
这段代码中,使用了 `os.system` 函数来执行 ping 命令,判断远程服务器是否可达。如果远程服务器不可达,则说明系统已经关机。循环检测远程服务器是否可达,直到远程服务器不可达时退出循环。
阅读全文