如何在Python項目中使用SSH連接到網絡設備

本篇文章給大家分享的是有關如何在Python項目中使用SSH連接到網絡設備，小編覺得挺實用的，因此分享給大家學習，希望大家閱讀完這篇文章后可以有所收獲，話不多說，跟著小編一起來看看吧。

1. 測試環境及關鍵代碼解釋

1.1 簡單測試環境

• 使用python3環境

• 使用第三方Netmiko模塊(基于Paramiko庫進行改版)

• 簡單的實驗環境

1.2 關鍵代碼

• import xx：導入模塊

• class xx：定義類

• def xx: 定義函數

• try-except ：處理可能引發的異常

• ssh.enable()：進入enable模式

• ssh.find_prompt()：返回當前提示符

• ssh.send_command()：發送查詢命令并返回結果

• ssh.send_config_set()：發送配置命令到目標設備

• ssh.disconnect()：關閉連接

Tips: import ConnectHandler和import Netmiko是一樣的效果的哈。

2. 完整代碼

'''
歡迎關注微信公眾號：'diandijishu'
 此平臺是網路工程師個人日常技術、項目案例經驗分享，
 為鞏固及提升技術能力乃至共享所學所知技術，
 也歡迎各位工程師一起分享、一起成長。
'''

#!/usr/bin/env python
#coding:utf-8

from netmiko import ConnectHandler
from netmiko.ssh_exception import NetMikoTimeoutException
from netmiko.ssh_exception import NetMikoAuthenticationException
from datetime import datetime
import time
import logging
from my_devices import device_list as devices

'定義類'
class SSH_Client():
 '定義login_host函數，用于登陸設備'
 def login_host(self , a_device):
  try:
   self.ssh = ConnectHandler(**a_device)
   self.ssh.enable()
   reply = self.ssh.find_prompt()
   print('>' * 10 + '成功登陸結果如下:' + '>' * 10 + '\n' + reply)
   return True
  except ValueError:
   logging.warning(a_device['host'] + ' Secret 密碼錯誤')
  except NetMikoTimeoutException:
   logging.warning(a_device['host'] + ' 連接不上設備,請檢查網絡是否正常通信')
  except NetMikoAuthenticationException:
   logging.warning(a_device['host'] + ' 登陸失敗，用戶名或密碼錯誤')

 '定義do_cmd函數,用于執行命令'
 def do_cmd(self,cmds):
  '讀取文件，for語句循環執行命令'
  with open(cmds) as cmd_obj:
   for cmd in cmd_obj:
    reply = self.ssh.send_command(cmd)
    time.sleep(2)
    logging.warning('>' * 10 + cmd.rstrip() + ' 命令執行結果如下:' + '>' * 10 + '\n' + reply)
 '定義logout_host函數，關閉程序'
 def logout_host(self):
  self.ssh.disconnect()

if __name__ == '__main__':
 cmds = 'cmd.txt' # 存放執行命令文件，相對路徑
 ssh_client = SSH_Client()
 start_time = datetime.now()
 for a_device in devices:
  '如果登錄結果為True，則執行命令，然后退出'
  if ssh_client.login_host(a_device):
   ssh_client.do_cmd(cmds)
   ssh_client.logout_host()
   time.sleep(2)
 stop_time = datetime.now()
 print('總花費時長：{0}\n'.format(stop_time - start_time))

3. 運行效果

4. 報錯效果

4.1 遠程連接不上

4.2 用戶名和密碼錯誤

以上就是如何在Python項目中使用SSH連接到網絡設備，小編相信有部分知識點可能是我們日常工作會見到或用到的。希望你能通過這篇文章學到更多知識。更多詳情敬請關注億速云行業資訊頻道。