mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 11:42:44 +08:00
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
import os
|
||
import sys
|
||
import paramiko
|
||
import time
|
||
import select
|
||
|
||
|
||
class sshClient:
|
||
#创建一个ssh客户端，和服务器连接上，准备发消息
|
||
def __init__(self,host,port,user,password):
|
||
self.trans = paramiko.Transport((host, port))
|
||
self.trans.start_client()
|
||
self.trans.auth_password(username=user, password=password)
|
||
self.channel = self.trans.open_session()
|
||
self.channel.get_pty()
|
||
self.channel.invoke_shell()
|
||
|
||
#给服务器发送一个命令
|
||
def sendCmd(self,cmd):
|
||
self.channel.sendall(cmd)
|
||
|
||
#接收的时候，有时候服务器处理的比较慢，需要设置一个延时等待一下。
|
||
def recvResponse(self,timeout):
|
||
data=b''
|
||
while True:
|
||
try:
|
||
#使用select，不断的读取数据，直到没有多余的数据了，超时返回。
|
||
readable,w,e= select.select([self.channel],[],[],timeout)
|
||
if self.channel in readable:
|
||
data = self.channel.recv(1024)
|
||
else:
|
||
sys.stdout.write(data.decode())
|
||
sys.stdout.flush()
|
||
return data.decode()
|
||
except TimeoutError:
|
||
sys.stdout.write(data.decode())
|
||
sys.stdout.flush()
|
||
return data.decode
|
||
#关闭客户端
|
||
def close(self):
|
||
self.channel.close()
|
||
self.trans.close()
|
||
|
||
host='host'
|
||
port=22#your port
|
||
user='root'
|
||
pwd='pass'
|
||
|
||
ssh = sshClient(host,port,user,pwd)
|
||
response = ssh.recvResponse(1)
|
||
response = ssh.sendCmd("ls\n")
|
||
ssh.sendCmd("cd /home\n")
|
||
response = ssh.recvResponse(1)
|
||
ssh.sendCmd("ls\n")
|
||
response = ssh.recvResponse(1)
|
||
|
||
ssh.close()
|