2016-09-20 84 views
0

我的sendShell对象在列表commandfactory []中运行后,我需要帮助终止我的SSH会话。Python - PARAMIKO SSH close session

我有一个python脚本,我使用paramiko通过ssh连接到思科实验室路由器;在commandfactory []中执行命令;并将结果输出到标准输出。一切似乎都起作用,除非在我的所有命令运行后,我似乎无法让SSH会话关闭。会话保持打开状态,直到我终止我的脚本。

import threading, paramiko, re, os 

class ssh: 
    shell = None 
    client = None 
    transport = None 


    def __init__(self, address, username, password): 
     print("Connecting to server on ip", str(address) + ".") 
     self.client = paramiko.client.SSHClient() 
     self.client.set_missing_host_key_policy(paramiko.client.AutoAddPolicy()) 
     self.client.connect(address, username=username, password=password, look_for_keys=False) 
     self.transport = paramiko.Transport((address, 22)) 
     self.transport.connect(username=username, password=password) 

     thread = threading.Thread(target=self.process) 
     thread.daemon = True 
     thread.start() 

    def closeConnection(self): 
     if(self.client != None): 
      self.client.close() 
      self.transport.close() 

    def openShell(self): 
     self.shell = self.client.invoke_shell() 

    def sendShell(self): 
     self.commandfactory = [] 
     print("\nWelcome to Command Factory. Enter Commands you want to execute.\nType \"done\" when you are finished:") 
     while not re.search(r"done.*", str(self.commandfactory)): 
      self.commandfactory.append(input(":")) 
      if self.commandfactory[-1] == "done": 
       del self.commandfactory[-1] 
       break 

     print ("Here are the commands you're going to execute:\n" + str(self.commandfactory)) 
     if(self.shell): 
      self.shell.send("enable" + "\n") 
      self.shell.send("ilovebeer" + "\n") 
      self.shell.send("term len 0" + "\n") 
      for cmdcnt in range(0,len(self.commandfactory)): 
       self.shell.send(self.commandfactory[cmdcnt] + "\n") 
      self.shell.send("exit" + "\n") 
      self.shell.send("\n") 

     else: 
      print("Shell not opened.") 

    def process(self): 
     global connection 
     while True: 
      # Print data when available 
      if self.shell != None and self.shell.recv_ready(): 
       alldata = self.shell.recv(1024) 
       while self.shell.recv_ready(): 
        alldata += self.shell.recv(1024) 
       strdata = str(alldata, "utf8") 
       strdata.strip() 
       print(strdata, end = "") 



sshUsername = "adrian" 
sshPassword = "ilovebeer" 
sshServer = "10.10.254.129" 

connection = ssh(sshServer, sshUsername, sshPassword) 
connection.openShell() 

while True: 
    connection.sendShell() 

我想一直跑了(下面的代码)SSH会话终止一旦所有的命令,在我的“的CommandFactory”名单。

def sendShell(self): 
    self.commandfactory = [] 
    print("\nWelcome to Command Factory. Enter Commands you want to execute.\nType \"done\" when you are finished:") 
    while not re.search(r"done.*", str(self.commandfactory)): 
     self.commandfactory.append(input(":")) 
     if self.commandfactory[-1] == "done": 
      del self.commandfactory[-1] 
      break 

    print ("Here are the commands you're going to execute:\n" + str(self.commandfactory)) 
    if(self.shell): 
     self.shell.send("enable" + "\n") 
     self.shell.send("ilovebeer" + "\n") 
     self.shell.send("term len 0" + "\n") 
     for cmdcnt in range(0,len(self.commandfactory)): 
      self.shell.send(self.commandfactory[cmdcnt] + "\n") 
     self.shell.send("exit" + "\n") 
     self.shell.send("\n") 

我的代码主要取自https://daanlenaerts.com/blog/2016/07/01/python-and-ssh-paramiko-shell/。非常感谢Daan Lenaerts建立一个好的博客。我做了自己的修改以适应我的需求。

回答

1

结束与self.transport.close的sendShell()函数,见http://docs.paramiko.org/en/2.0/api/transport.html

+0

VB,OMG,它一直在我面前。我知道要使用self.transport.close(),但我不知道在哪里,直到我看到您的回复;出于某种奇怪的原因,这正是我弄明白的原因。为了得到我想要的,我将transport.close()放在我的commandfactory迭代器之后。 现在,如果我可能会问,如果我关闭会话,那么我会如何打破我的循环并结束我的python脚本? – adrian

0

之所以能够通过我的迭代器后加入self.shell.transport.close()来解决。

def sendShell(self): 
    self.commandfactory = [] 
    print("\nWelcome to Command Factory. Enter Commands you want to execute.\nType \"done\" when you are finished:") 
    while not re.search(r"done.*", str(self.commandfactory)): 
     self.commandfactory.append(input(":")) 
     if self.commandfactory[-1] == "done": 
      del self.commandfactory[-1] 
      break 

    print ("Here are the commands you're going to execute:\n" + str(self.commandfactory)) 
    if(self.shell): 
     self.shell.send("enable" + "\n") 
     self.shell.send("ilovebeer" + "\n") 
     self.shell.send("term len 0" + "\n") 
     for cmdcnt in range(0,len(self.commandfactory)): 
      self.shell.send(self.commandfactory[cmdcnt] + "\n") 
     self.shell.send("exit" + "\n") 
     self.shell.transport.close() 
相关问题