'''
歡迎關(guān)注微信公眾號(hào)：'diandijishu'
 此平臺(tái)是網(wǎng)路工程師個(gè)人日常技術(shù)、項(xiàng)目案例經(jīng)驗(yàn)分享，
 為鞏固及提升技術(shù)能力乃至共享所學(xué)所知技術(shù)，
 也歡迎各位工程師一起分享、一起成長(zhǎng)。
'''

#!/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函數(shù)，用于登陸設(shè)備'
 def login_host(self , a_device):
  try:
   self.ssh = ConnectHandler(**a_device)
   self.ssh.enable()
   reply = self.ssh.find_prompt()
   print('>' * 10 + '成功登陸結(jié)果如下:' + '>' * 10 + '\n' + reply)
   return True
  except ValueError:
   logging.warning(a_device['host'] + ' Secret 密碼錯(cuò)誤')
  except NetMikoTimeoutException:
   logging.warning(a_device['host'] + ' 連接不上設(shè)備,請(qǐng)檢查網(wǎng)絡(luò)是否正常通信')
  except NetMikoAuthenticationException:
   logging.warning(a_device['host'] + ' 登陸失敗，用戶名或密碼錯(cuò)誤')

 '定義do_cmd函數(shù),用于執(zhí)行命令'
 def do_cmd(self,cmds):
  '讀取文件，for語(yǔ)句循環(huán)執(zhí)行命令'
  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() + ' 命令執(zhí)行結(jié)果如下:' + '>' * 10 + '\n' + reply)
 '定義logout_host函數(shù)，關(guān)閉程序'
 def logout_host(self):
  self.ssh.disconnect()

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