langbot-app--langbot
RockChinQ
58 行
1.7 KiB
Python
58 行
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()
|