1431 lines
72 KiB
Python
1431 lines
72 KiB
Python
# -*- coding:utf-8 -*-
|
||
|
||
import paramiko
|
||
import time
|
||
import datetime
|
||
import os
|
||
import sqlite3
|
||
import re
|
||
from configobj import ConfigObj
|
||
import pypinyin
|
||
import openpyxl
|
||
from openpyxl.styles import Font, Alignment
|
||
|
||
#菜单
|
||
def handleMenu():
|
||
userInput = input("\n请输入序列号或指令('quit'或'q'可返回菜单)：")
|
||
return userInput
|
||
|
||
#管理和查看本地数据库
|
||
def manageLocalDB():
|
||
ColorLogDecorator().active()
|
||
if os.path.exists("alldata.db") == False:
|
||
print(ColorLogDecorator().yellow("【WARNING】未发现本地数据库，正在初始化中...","strong"))
|
||
time.sleep(1)
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
c.execute('''CREATE TABLE DEVICES
|
||
(ID INTEGER PRIMARY KEY autoincrement,
|
||
NAME varchar(255) NOT NULL,
|
||
IP varchar(255) NOT NULL,
|
||
TYPE varchar(255) NOT NULL,
|
||
IDC varchar(255) NOT NULL);
|
||
''')
|
||
c.execute("INSERT INTO DEVICES (NAME,IP,TYPE,IDC) \
|
||
VALUES ('IDC-example-NE40-RT01','1.1.1.1','华为','测试')")
|
||
conn.commit()
|
||
conn.close()
|
||
print(ColorLogDecorator().green("【INFO】数据库初始化完成","strong"))
|
||
else:
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
print(ColorLogDecorator().green("【INFO】数据库连接成功！","strong"))
|
||
selResult = c.execute("SELECT * FROM devices")
|
||
print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
|
||
print(" 【ID】 ---- 【设备名称】 ---- 【ip】 ---- 【厂商】 ---- 【IDC名称】 \n")
|
||
for row in selResult:
|
||
print(' ',row[0],'----',row[1],'----',row[2],'----',row[3],'----',row[4])
|
||
|
||
print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
|
||
conn.commit()
|
||
conn.close()
|
||
print("\n\n")
|
||
while True:
|
||
|
||
manaInput = input("【输入序列号 1.添加设备信息(单个录入) 2.批量添加设备信息 3.修改设备信息 4.删除设备 5.管理封堵设备】\n返回上一级请输入指令'quit'或'q'\n:")
|
||
if manaInput == "1":
|
||
# ColorLogDecorator().active()
|
||
devInfo = []
|
||
os.system('cls')
|
||
print("======================================")
|
||
print(" 单个设备录入")
|
||
print("当前进度：")
|
||
print(ColorLogDecorator().blue("> *设备名称", "strong"))
|
||
print("*设备IP\n*厂商名称\n*IDC名称")
|
||
print("======================================")
|
||
addInput = input("(取消保存并返回上一级菜单输入'q'或'quit')\n请输入信息：")
|
||
if addInput == "":
|
||
print(ColorLogDecorator().red("【WARNING】不允许输入空值！正在跳转到上一级目录...", "strong"))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == "q" or addInput == "quit":
|
||
os.system('cls')
|
||
continue
|
||
else:
|
||
devInfo.append(addInput)
|
||
|
||
os.system('cls')
|
||
print("======================================")
|
||
print(" 单个设备录入")
|
||
print("当前进度：")
|
||
print("*设备名称")
|
||
print(ColorLogDecorator().blue("> *设备IP", "strong"))
|
||
print("*厂商名称\n*IDC名称")
|
||
print("======================================")
|
||
addInput = input("(取消保存并返回上一级菜单输入'q'或'quit')\n(重新输入上一步请输入'c')\n请输入信息：")
|
||
if addInput == "":
|
||
print(ColorLogDecorator().red("【WARNING】不允许输入空值！正在跳转到上一级目录..."))
|
||
time.sleep(1.5)
|
||
continue
|
||
elif addInput == "q" or addInput == "quit":
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == 'c':
|
||
os.system('cls')
|
||
print("======================================")
|
||
print(" 单个设备录入")
|
||
print("当前进度：")
|
||
print(ColorLogDecorator().blue("> *设备名称", "strong"))
|
||
print("*设备IP\n*厂商名称\n*IDC名称")
|
||
print("======================================")
|
||
addInput = input("(取消保存并返回上一级菜单输入'q'或'quit')\n请输入信息：")
|
||
if addInput == "":
|
||
print(ColorLogDecorator().red("【WARNING】不允许输入空值！正在跳转到上一级目录..."))
|
||
time.sleep(1.5)
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == "q" or addInput == "quit":
|
||
os.system('cls')
|
||
continue
|
||
else:
|
||
devInfo[0] = addInput
|
||
os.system('cls')
|
||
print("======================================")
|
||
print(" 单个设备录入")
|
||
print("当前进度：")
|
||
print("*设备名称")
|
||
print(ColorLogDecorator().blue("> *设备IP", "strong"))
|
||
print("*厂商名称\n*IDC名称")
|
||
print("======================================")
|
||
addInput = input("(取消保存并返回上一级菜单输入'q'或'quit')\n请输入信息：")
|
||
if addInput == "":
|
||
print(ColorLogDecorator().yellow("【WARNING】不允许输入空值！正在跳转到上一级目录..."))
|
||
time.sleep(1.5)
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == "q" or addInput == "quit":
|
||
os.system('cls')
|
||
continue
|
||
else:
|
||
devInfo.append(addInput)
|
||
|
||
else:
|
||
devInfo.append(addInput)
|
||
|
||
os.system('cls')
|
||
print("======================================")
|
||
print(" 单个设备录入")
|
||
print("当前进度：")
|
||
print("*设备名称")
|
||
print("*设备IP")
|
||
print(ColorLogDecorator().blue("> *厂商名称", "strong"))
|
||
print("*IDC名称")
|
||
print("======================================")
|
||
addInput = input("(取消保存并返回上一级菜单输入'q'或'quit')\n(重新输入上一步请输入'c')\n请输入信息：")
|
||
if addInput == "":
|
||
print(ColorLogDecorator().red("【WARNING】不允许输入空值！", "strong"))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == "q" or addInput == "quit":
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == 'c':
|
||
os.system('cls')
|
||
print("======================================")
|
||
print(" 单个设备录入")
|
||
print("当前进度：")
|
||
print("*设备名称")
|
||
print(ColorLogDecorator().blue("> *设备IP", "strong"))
|
||
print("*厂商名称\n*IDC名称")
|
||
print("======================================")
|
||
addInput = input("(取消保存并返回上一级菜单输入'q'或'quit')\n请输入信息：")
|
||
if addInput == "":
|
||
print(ColorLogDecorator().yellow("【WARNING】不允许输入空值！正在跳转到上一级目录..."))
|
||
time.sleep(1.5)
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == "q" or addInput == "quit":
|
||
os.system('cls')
|
||
continue
|
||
else:
|
||
devInfo[1] = addInput
|
||
os.system('cls')
|
||
print("======================================")
|
||
print(" 单个设备录入")
|
||
print("当前进度：")
|
||
print("*设备名称")
|
||
print("*设备IP")
|
||
print(ColorLogDecorator().blue("> *厂商名称", "strong"))
|
||
print("*IDC名称")
|
||
print("======================================")
|
||
addInput = input("(取消保存并返回上一级菜单输入'q'或'quit')\n请输入信息：")
|
||
if addInput == "":
|
||
print(ColorLogDecorator().yellow("【WARNING】不允许输入空值！正在跳转到上一级目录..."))
|
||
time.sleep(1.5)
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == "q" or addInput == "quit":
|
||
os.system('cls')
|
||
continue
|
||
else:
|
||
devInfo.append(addInput)
|
||
else:
|
||
devInfo.append(addInput)
|
||
|
||
os.system('cls')
|
||
print("======================================")
|
||
print(" 单个设备录入")
|
||
print("当前进度：")
|
||
print("*设备名称")
|
||
print("*设备IP")
|
||
print("*厂商名称")
|
||
print(ColorLogDecorator().blue("> *IDC名称", "strong"))
|
||
print("======================================")
|
||
addInput = input("(取消保存并返回上一级菜单输入'q'或'quit')\n(重新输入上一步请输入'c')\n请输入信息：")
|
||
if addInput == "":
|
||
print(ColorLogDecorator().red("【WARNING】不允许输入空值！正在跳转到上一级目录..."))
|
||
time.sleep(1.5)
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == "q" or addInput == "quit":
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == 'c':
|
||
os.system('cls')
|
||
print("======================================")
|
||
print(" 单个设备录入")
|
||
print("当前进度：")
|
||
print("*设备名称")
|
||
print("*设备IP")
|
||
print(ColorLogDecorator().blue("> *厂商名称", "strong"))
|
||
print("*IDC名称")
|
||
print("======================================")
|
||
addInput = input("(取消保存并返回上一级菜单输入'q'或'quit')\n请输入信息：")
|
||
if addInput == "":
|
||
print(ColorLogDecorator().red("【WARNING】不允许输入空值！正在跳转到上一级目录..."))
|
||
time.sleep(1.5)
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == "q" or addInput == "quit" or addInput == 'c':
|
||
os.system('cls')
|
||
continue
|
||
else:
|
||
devInfo[2] = addInput
|
||
os.system('cls')
|
||
print("======================================")
|
||
print(" 单个设备录入")
|
||
print("当前进度：")
|
||
print("*设备名称")
|
||
print("*设备IP")
|
||
print("*厂商名称")
|
||
print(ColorLogDecorator().blue("> *IDC名称", "strong"))
|
||
print("======================================")
|
||
addInput = input("(取消保存并返回上一级菜单输入'q'或'quit')\n请输入信息：")
|
||
if addInput == "":
|
||
print(ColorLogDecorator().red("【WARNING】不允许输入空值！正在跳转到上一级目录..."))
|
||
time.sleep(1.5)
|
||
os.system('cls')
|
||
continue
|
||
elif addInput == "q" or addInput == "quit":
|
||
os.system('cls')
|
||
continue
|
||
else:
|
||
devInfo.append(addInput)
|
||
else:
|
||
devInfo.append(addInput)
|
||
|
||
print(devInfo)
|
||
isWantSave = input(ColorLogDecorator().yellow("！！！是否要保存？ [Y/N/C]"))
|
||
if isWantSave.upper() == "N" or isWantSave.upper() == "C":
|
||
print(ColorLogDecorator().red("【INFO】已取消保存)"))
|
||
time.sleep(1.5)
|
||
continue
|
||
elif isWantSave.upper() == "Y":
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
sql = '''INSERT INTO DEVICES (NAME,IP,TYPE,IDC) VALUES ('{0}','{1}','{2}','{3}');'''.format(devInfo[0],devInfo[1],devInfo[2],devInfo[3])
|
||
print(sql)
|
||
c.execute(sql)
|
||
conn.commit()
|
||
conn.close()
|
||
print(ColorLogDecorator().green("【INFO】保存成功！)"))
|
||
time.sleep(1.5)
|
||
os.system('cls')
|
||
continue
|
||
|
||
elif manaInput == "2":
|
||
if os.path.isfile('设备表.xlsx') == False:
|
||
wb = openpyxl.Workbook()
|
||
wb.create_sheet(index=0, title='Sheet1')
|
||
sheet = wb.worksheets[0]
|
||
font = Font(name='黑体', size=10, bold=True)
|
||
alignment = Alignment(horizontal='left',vertical='center')
|
||
sheet['A1'].value = '设备名称'
|
||
sheet['A1'].font = font
|
||
sheet['A1'].alignment = alignment
|
||
sheet['B1'].value = 'IP'
|
||
sheet['B1'].font = font
|
||
sheet['B1'].alignment = alignment
|
||
sheet['C1'].value = '设备厂商'
|
||
sheet['C1'].font = font
|
||
sheet['C1'].alignment = alignment
|
||
sheet['D1'].value = '机房'
|
||
sheet['D1'].font = font
|
||
sheet['D1'].alignment = alignment
|
||
wb.save('设备表.xlsx')
|
||
print(ColorLogDecorator().green('【INFO】检测到模板文件不存在，已自动生成！\n', 'strong'))
|
||
isY_input= input(' 请将设备信息填入模板后，在此输入Y即可批量导入：')
|
||
if isY_input.upper() == 'Y':
|
||
wb = openpyxl.load_workbook('设备表.xlsx')
|
||
sheet = wb.worksheets[0]
|
||
rowsList = []
|
||
cellsList = []
|
||
for row in sheet.iter_rows():
|
||
for cell in row:
|
||
cellsList.append(cell.value)
|
||
rowsList.append(cellsList)
|
||
cellsList = []
|
||
if len(rowsList) < 2:
|
||
print(ColorLogDecorator().red('【ERROR】无法检测到数据，请先填写数据。\n', 'strong'))
|
||
continue
|
||
del rowsList[0]
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
for devs in rowsList:
|
||
PreInDBvalue = c.execute("SELECT * FROM DEVICES WHERE IP = '{0}'".format(devs[1]))
|
||
if len(list(PreInDBvalue)) == 0:
|
||
c.execute("INSERT INTO DEVICES (NAME,IP,TYPE,IDC) VALUES ('{0}','{1}','{2}','{3}')".format(devs[0],devs[1],devs[2],devs[3]))
|
||
print(ColorLogDecorator().green("【INFO】已添加设备 {0}".format(devs[0]), "strong"))
|
||
else:
|
||
print(ColorLogDecorator().red("【ERROR】设备 {0} 已存在数据库！请勿重复添加！".format(devs[0]), "strong"))
|
||
conn.commit()
|
||
conn.close()
|
||
print(ColorLogDecorator().green('【INFO】批量导入完成 ！\n', 'strong'))
|
||
else:
|
||
print(ColorLogDecorator().yellow('【INFO】操作已取消！\n', 'strong'))
|
||
continue
|
||
|
||
elif manaInput == "3":
|
||
print('\n-----------------------------')
|
||
IDnumInput = input("【选择要修改的设备ID号】：")
|
||
#判断输入的值是否只由纯数字组成
|
||
if IDnumInput.isdigit() == True:
|
||
sql = '''SELECT * FROM devices WHERE ID='{0}';'''.format(IDnumInput)
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
selResult = c.execute(sql)
|
||
selResult1 = list(selResult)
|
||
conn.commit()
|
||
conn.close()
|
||
if len(selResult1) == 0:
|
||
print(ColorLogDecorator().red('【结果】不存在该设备\n', 'strong'))
|
||
continue
|
||
else:
|
||
TheDevsInfo = []
|
||
titleList = [' ID：',' 设备名称：',' IP：',' 厂商：',' IDC名称：']
|
||
#os.system('cls')
|
||
print('\n++++++++++++++++当前数据+++++++++++++++++\n')
|
||
print('{0}{1}\n'.format(titleList[0],str(selResult1[0][0])))
|
||
TheDevsInfo.append(str(selResult1[0][0]))
|
||
print('{0}{1}\n'.format(titleList[1],selResult1[0][1]))
|
||
TheDevsInfo.append(selResult1[0][1])
|
||
print('{0}{1}\n'.format(titleList[2],selResult1[0][2]))
|
||
TheDevsInfo.append(selResult1[0][2])
|
||
print('{0}{1}\n'.format(titleList[3],selResult1[0][3]))
|
||
TheDevsInfo.append(selResult1[0][3])
|
||
print('{0}{1}\n'.format(titleList[4],selResult1[0][4]))
|
||
TheDevsInfo.append(selResult1[0][4])
|
||
print('\n++++++++++++++++++++++++++++++++++++++++++++')
|
||
updateInput = input('修改 设备名称【不修改则留空】：')
|
||
if updateInput != "":
|
||
TheDevsInfo[1] = updateInput
|
||
updateInput = input('修改 IP【不修改则留空】：')
|
||
if updateInput != "":
|
||
TheDevsInfo[2] = updateInput
|
||
updateInput = input('修改 厂商【不修改则留空】：')
|
||
if updateInput != "":
|
||
TheDevsInfo[3] = updateInput
|
||
updateInput = input('修改 IDC名称【不修改则留空】：')
|
||
if updateInput != "":
|
||
TheDevsInfo[4] = updateInput
|
||
|
||
os.system('cls')
|
||
print('++++++++++++++++更新为以下数据+++++++++++++++++\n')
|
||
print('{0}{1}\n'.format(titleList[0], str(TheDevsInfo[0])))
|
||
print('{0}{1}\n'.format(titleList[1], TheDevsInfo[1]))
|
||
print('{0}{1}\n'.format(titleList[2], TheDevsInfo[2]))
|
||
print('{0}{1}\n'.format(titleList[3], TheDevsInfo[3]))
|
||
print('{0}{1}\n'.format(titleList[4], TheDevsInfo[4]))
|
||
print('\n++++++++++++++++++++++++++++++++++++++++++++')
|
||
updateResultInput = input(ColorLogDecorator().yellow(' 是否要保存？【Y/N】 ：','bg-strong'))
|
||
while True:
|
||
if updateResultInput.upper() == 'Y':
|
||
sql = "UPDATE devices SET NAME='{0}',IP='{1}',TYPE='{2}',IDC='{3}' WHERE ID='{4}';".format(TheDevsInfo[1],TheDevsInfo[2],TheDevsInfo[3],TheDevsInfo[4],TheDevsInfo[0])
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
c.execute(sql)
|
||
conn.commit()
|
||
conn.close()
|
||
print(ColorLogDecorator().green('\n 【INFO】保存成功！将返回主菜单...','strong'))
|
||
time.sleep(2)
|
||
break
|
||
elif updateResultInput.upper() == 'N':
|
||
break
|
||
else:
|
||
updateResultInput = input(ColorLogDecorator().yellow(' 是否要保存？【Y/N】：', 'bg-strong'))
|
||
os.system('cls')
|
||
break
|
||
else:
|
||
print(ColorLogDecorator().red('【WARNING】请输入正确的ID号！\n','strong'))
|
||
|
||
elif manaInput == "4":
|
||
print('\n------------------------------------------------------')
|
||
delNumInput = input(' 请输入要删除的设备ID号【退出请输入c或q】：')
|
||
if delNumInput == 'c' or delNumInput == 'q' or delNumInput == 'C' or delNumInput == 'Q':
|
||
continue
|
||
elif delNumInput.isdigit() == True:
|
||
sql = '''SELECT * FROM devices WHERE ID='{0}';'''.format(delNumInput)
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
devsList = c.execute(sql)
|
||
if len(list(devsList)) != 0:
|
||
conn.commit()
|
||
conn.close()
|
||
while True:
|
||
wantDelInput = input(ColorLogDecorator().yellow(' 确定要删除?【Y/N】：','strong'))
|
||
if wantDelInput.upper() == 'Y':
|
||
### 删除设备时，如果是关联的封堵设备，也会一同从封堵设备列表中删除
|
||
if os.path.isfile('blockconfig.ini'):
|
||
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
|
||
cfgDict = config['devList']
|
||
for key, values in cfgDict.items():
|
||
if delNumInput + ',' in values:
|
||
repAfter = values.replace(delNumInput + ',', "")
|
||
config['devList'][key] = repAfter
|
||
config.write()
|
||
elif delNumInput == values:
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
Currsql_res = c.execute(
|
||
"SELECT * FROM devices WHERE ID='{0}'".format(delNumInput))
|
||
Currsql_res_List = list(Currsql_res)
|
||
conn.commit()
|
||
conn.close()
|
||
saveVal = config['list']['jifang'].replace(Currsql_res_List[0][4] + "|", "")
|
||
config['list']['jifang'] = saveVal
|
||
config.write()
|
||
saveVal = config['list']['jifang'].replace("|" + Currsql_res_List[0][4], "")
|
||
config['list']['jifang'] = saveVal
|
||
config.write()
|
||
saveVal = config['list']['jifang'].replace(Currsql_res_List[0][4], "")
|
||
config['list']['jifang'] = saveVal
|
||
config.write()
|
||
del config['devList'][key]
|
||
config.write()
|
||
elif delNumInput in values:
|
||
repAfter = values.replace(',' + delNumInput, "")
|
||
config['devList'][key] = repAfter
|
||
config.write()
|
||
### END
|
||
|
||
sql = '''DELETE FROM devices WHERE ID='{0}';'''.format(delNumInput)
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
c.execute(sql)
|
||
conn.commit()
|
||
conn.close()
|
||
print(ColorLogDecorator().green('\n【INFO】删除设备成功！\n', 'strong'))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
break
|
||
elif wantDelInput.upper() == 'N':
|
||
os.system('cls')
|
||
break
|
||
else:
|
||
continue
|
||
continue
|
||
|
||
else:
|
||
conn.commit()
|
||
conn.close()
|
||
print(ColorLogDecorator().yellow('【WARNING】不存在该设备！\n','strong'))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
|
||
else:
|
||
print(ColorLogDecorator().red("【ERROR】ID号为纯数字，请正确填写！\n", "strong"))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
|
||
elif manaInput == "5":
|
||
os.system('cls')
|
||
if os.path.isfile('blockconfig.ini') == False:
|
||
config = ConfigObj()
|
||
config.filename = './blockconfig.ini'
|
||
config['list'] = {}
|
||
config['list']['jifang'] = ''
|
||
config['devList'] = {}
|
||
config['account'] = {}
|
||
config['account']['username'] = ''
|
||
config['account']['password'] = ''
|
||
config.write()
|
||
print(ColorLogDecorator().green('【INFO】封堵信息库已初始化完毕，请添加封堵设备！','strong'))
|
||
while True:
|
||
manaBlockInput = input("\n 【管理封堵设备】(返回上一级请输入指令'q'或'quit')\n 1.查看封堵设备列表 2.添加封堵设备 3.删除封堵设备 4.添加/更改通用设备登陆账号\n(输入)：")
|
||
if manaBlockInput == "1":
|
||
os.system('cls')
|
||
if os.path.isfile('blockconfig.ini'):
|
||
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
|
||
if config['list']['jifang'] == '':
|
||
print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
|
||
print(" 【封堵设备列表】")
|
||
print(" 【ID】 ---- 【设备名称】 ---- 【ip】 ---- 【厂商】 ---- 【机房名称】 ")
|
||
print(
|
||
"+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
else:
|
||
print(
|
||
"+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
|
||
print(" 【封堵设备列表】")
|
||
print(" 【ID】 ---- 【设备名称】 ---- 【ip】 ---- 【厂商】 ---- 【机房名称】 ")
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
# Currsql_res = c.execute("SELECT * FROM devices WHERE ID='{0}'".format(ipblock_Devices))
|
||
for key,values in config['devList'].items():
|
||
if ',' in values:
|
||
AfterID_Lst = values.split(',')
|
||
for textAfterID in AfterID_Lst:
|
||
returnsql = c.execute(
|
||
"SELECT * FROM devices WHERE ID='{0}'".format(textAfterID))
|
||
ressql = list(returnsql)
|
||
print(
|
||
" {0}----{1}----{2}----{3}----{4}\n".format(ressql[0][0], ressql[0][1],
|
||
ressql[0][2], ressql[0][3],
|
||
ressql[0][4]))
|
||
else:
|
||
returnsql = c.execute("SELECT * FROM devices WHERE ID='{0}'".format(values))
|
||
ressql = list(returnsql)
|
||
print(" {0}----{1}----{2}----{3}----{4}\n".format(ressql[0][0], ressql[0][1], ressql[0][2], ressql[0][3], ressql[0][4]))
|
||
print(
|
||
"+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
|
||
conn.commit()
|
||
conn.close()
|
||
os.system('pause')
|
||
continue
|
||
|
||
elif manaBlockInput == "2":
|
||
os.system('cls')
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
print(ColorLogDecorator().green("【INFO】数据库连接成功！", "strong"))
|
||
selResult = c.execute("SELECT * FROM devices")
|
||
print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
|
||
print(" 【ID】 ---- 【设备名称】 ---- 【ip】 ---- 【厂商】 ---- 【IDC名称】 \n")
|
||
for row in selResult:
|
||
print(row[0], '----', row[1], '----', row[2], '----', row[3], '----', row[4])
|
||
|
||
print("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++")
|
||
conn.commit()
|
||
conn.close()
|
||
print("\n\n")
|
||
addValue = input('从设备信息总库导入到封堵设备列表 ，请输入要添加的设备ID(批量添加用英文逗号分隔)：')
|
||
if addValue == '':
|
||
print(ColorLogDecorator().red("【ERROR】请正确输入值！","strong"))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
# 批量添加
|
||
if ',' in addValue:
|
||
# 清除用户可能首尾多输入的逗号
|
||
addValue = addValue.strip(',')
|
||
# 分割判断一遍是否全为纯数字和输入是否合法，不合法则全部都跳过
|
||
ad_v = addValue.split(',')
|
||
for i in ad_v:
|
||
if i.isdigit() == False:
|
||
print(ColorLogDecorator().red("【ERROR】请正确输入值！", "strong"))
|
||
os.system('pause')
|
||
continue
|
||
# 查询设备ID的归属机房，以写入配置文件
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
ini_path = "./blockconfig.ini"
|
||
for d_id in ad_v:
|
||
ressql = c.execute("SELECT * FROM devices WHERE ID='{0}'".format(d_id))
|
||
devs_info = list(ressql)
|
||
if len(devs_info) > 0:
|
||
config = ConfigObj(ini_path, encoding='UTF-8')
|
||
beforeLst = config['list']['jifang']
|
||
s = ''
|
||
for single_word in pypinyin.pinyin(devs_info[0][4], style=pypinyin.NORMAL):
|
||
s += ''.join(single_word)
|
||
if beforeLst != '' and devs_info[0][4] not in beforeLst:
|
||
config['list']['jifang'] = beforeLst + '|' + devs_info[0][4]
|
||
config['devList'][s] = d_id
|
||
config.write()
|
||
elif beforeLst != '' and devs_info[0][4] in beforeLst:
|
||
currIDC_devs = config['devList'][s] + ',' + d_id
|
||
config['devList'][s] = currIDC_devs
|
||
config.write()
|
||
elif beforeLst == '':
|
||
config['list']['jifang'] = devs_info[0][4]
|
||
config['devList'][s] = d_id
|
||
config.write()
|
||
print(ColorLogDecorator().green("【INFO】添加完成！","strong"))
|
||
else:
|
||
print(ColorLogDecorator().red("【ERROR】不存在该设备！无法添加","strong"))
|
||
conn.commit()
|
||
conn.close()
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
|
||
# 单个添加
|
||
elif addValue.isdigit() == True:
|
||
# 查询设备ID的归属机房，以写入配置文件
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
ini_path = "./blockconfig.ini"
|
||
ressql = c.execute("SELECT * FROM devices WHERE ID='{0}'".format(addValue))
|
||
devs_info = list(ressql)
|
||
if len(devs_info) > 0:
|
||
config = ConfigObj(ini_path, encoding='UTF-8')
|
||
beforeLst = config['list']['jifang']
|
||
s = ''
|
||
for single_word in pypinyin.pinyin(devs_info[0][4], style=pypinyin.NORMAL):
|
||
s += ''.join(single_word)
|
||
if beforeLst != '' and devs_info[0][4] not in beforeLst:
|
||
config['list']['jifang'] = beforeLst + '|' + devs_info[0][4]
|
||
config['devList'][s] = addValue
|
||
config.write()
|
||
elif beforeLst != '' and devs_info[0][4] in beforeLst:
|
||
currIDC_devs = config['devList'][s] + ',' + addValue
|
||
config['devList'][s] = currIDC_devs
|
||
config.write()
|
||
elif beforeLst == '':
|
||
config['list']['jifang'] = devs_info[0][4]
|
||
config['devList'][s] = addValue
|
||
config.write()
|
||
print(ColorLogDecorator().green("【INFO】添加完成！", "strong"))
|
||
else:
|
||
print(ColorLogDecorator().red("【ERROR】不存在该设备！无法添加","strong"))
|
||
conn.commit()
|
||
conn.close()
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
|
||
else:
|
||
print(ColorLogDecorator().red("【ERROR】请正确输入值！", "strong"))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
|
||
elif manaBlockInput == "3":
|
||
deldevs = input(" 请输入要删除的设备id(多个设备用英文逗号分隔;留空或输入q则直接退出)：")
|
||
if "," in deldevs and deldevs != "":
|
||
afterCutLst = deldevs.split(",")
|
||
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
|
||
cfgDict = config['devList']
|
||
for afterCutValue in afterCutLst:
|
||
for key,values in cfgDict.items():
|
||
if afterCutValue + ',' in values:
|
||
repAfter = values.replace(afterCutValue + ',', "")
|
||
config['devList'][key] = repAfter
|
||
config.write()
|
||
elif afterCutValue == values:
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
Currsql_res = c.execute("SELECT * FROM devices WHERE ID='{0}'".format(afterCutValue))
|
||
Currsql_res_List = list(Currsql_res)
|
||
conn.commit()
|
||
conn.close()
|
||
saveVal = config['list']['jifang'].replace(Currsql_res_List[0][4]+"|", "")
|
||
config['list']['jifang'] = saveVal
|
||
config.write()
|
||
config['list']['jifang'].replace("|"+Currsql_res_List[0][4], "")
|
||
config['list']['jifang'] = saveVal
|
||
config.write()
|
||
config['list']['jifang'].replace(Currsql_res_List[0][4],"")
|
||
config['list']['jifang'] = saveVal
|
||
config.write()
|
||
del config['devList'][key]
|
||
config.write()
|
||
elif afterCutValue in values:
|
||
repAfter = values.replace(',' + afterCutValue, "")
|
||
config['devList'][key] = repAfter
|
||
config.write()
|
||
print('【INFO】已完成删除！\n')
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
|
||
elif deldevs.isdigit() == True:
|
||
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
|
||
cfgDict = config['devList']
|
||
for key, values in cfgDict.items():
|
||
if deldevs + ',' in values:
|
||
repAfter = values.replace(deldevs + ',', "")
|
||
config['devList'][key] = repAfter
|
||
config.write()
|
||
elif deldevs == values:
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
Currsql_res = c.execute(
|
||
"SELECT * FROM devices WHERE ID='{0}'".format(deldevs))
|
||
Currsql_res_List = list(Currsql_res)
|
||
conn.commit()
|
||
conn.close()
|
||
saveVal = config['list']['jifang'].replace(Currsql_res_List[0][4] + "|", "")
|
||
config['list']['jifang'] = saveVal
|
||
config.write()
|
||
config['list']['jifang'].replace("|" + Currsql_res_List[0][4], "")
|
||
config['list']['jifang'] = saveVal
|
||
config.write()
|
||
config['list']['jifang'].replace(Currsql_res_List[0][4], "")
|
||
config['list']['jifang'] = saveVal
|
||
config.write()
|
||
del config['devList'][key]
|
||
config.write()
|
||
elif deldevs in values:
|
||
repAfter = values.replace(',' + deldevs, "")
|
||
config['devList'][key] = repAfter
|
||
config.write()
|
||
print(ColorLogDecorator().green('【INFO】已完成删除！\n', 'strong'))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
|
||
elif deldevs == "" or deldevs.upper() == "Q":
|
||
os.system('cls')
|
||
continue
|
||
|
||
elif manaBlockInput == "4":
|
||
os.system('cls')
|
||
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
|
||
print("============================")
|
||
print(" 【通用设备登录账号管理】")
|
||
print(" 账号：{0}".format(config['account']['username']))
|
||
print(" 密码：{0}".format(config['account']['password']))
|
||
print("============================\n")
|
||
modifyUser = input(" 修改账号为(留空则不修改)：")
|
||
modifyPass = input(" 修改密码为(留空则不修改)：")
|
||
if modifyUser != "":
|
||
config['account']['username'] = modifyUser
|
||
if modifyPass != "":
|
||
config['account']['password'] = modifyPass
|
||
config.write()
|
||
print(ColorLogDecorator().green("【INFO】修改完成！\n", "strong"))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
|
||
elif manaBlockInput.upper() == "Q" or manaBlockInput.upper() == "QUIT":
|
||
os.system('cls')
|
||
break
|
||
else:
|
||
os.system('cls')
|
||
continue
|
||
|
||
elif manaInput == "quit" or manaInput == "q":
|
||
os.system('cls')
|
||
break
|
||
else:
|
||
print(ColorLogDecorator.red("指令错误！\n", "strong"))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
|
||
|
||
# 导出所有设备配置信息
|
||
class devConfig_export:
|
||
def __init__(self):
|
||
self.out_filepath = ''
|
||
|
||
def main(self):
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
sqlRes = c.execute("SELECT * FROM devices")
|
||
if os.path.isfile('blockconfig.ini'):
|
||
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
|
||
conf_un = config['account']['username']
|
||
conf_pw = config['account']['password']
|
||
if conf_un == '' or conf_pw == '':
|
||
print('【ERROR】设备通用登录账号未设置,请设置后再操作！\n')
|
||
return
|
||
else:
|
||
print('【ERROR】设备通用登录账号未设置,请设置后再操作！\n')
|
||
return
|
||
|
||
self.out_filepath = "配置文件" + (datetime.datetime.today()).strftime('%Y-%m-%d')
|
||
if not os.path.exists(self.out_filepath):
|
||
os.mkdir(self.out_filepath)
|
||
|
||
for i in sqlRes:
|
||
if '华为' in i[3]:
|
||
self.setHuawei(i[1],i[2],conf_un,conf_pw)
|
||
elif 'H3C' in i[3]:
|
||
self.setH3C(i[1],i[2],conf_un,conf_pw)
|
||
elif '中兴' in i[3]:
|
||
self.setZhongxing(i[1],i[2],conf_un,conf_pw)
|
||
elif '锐捷' in i[3]:
|
||
self.setRuijie(i[1],i[2],conf_un,conf_pw)
|
||
elif '烽火' in i[3]:
|
||
self.setFenghuo(i[1],i[2],conf_un,conf_pw)
|
||
else:
|
||
print('【ERROR】设备【{0}】未导出配置，原因：程序暂不支持【{1}】厂商\n'.format(i[1], i[3]))
|
||
conn.commit()
|
||
conn.close()
|
||
|
||
def setHuawei(self, devName, conf_ip, conf_uname, conf_pass):
|
||
try:
|
||
# 创建ssh对象
|
||
ssh = paramiko.SSHClient()
|
||
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
ssh.connect(hostname=conf_ip, port=22, username=conf_uname, password=conf_pass)
|
||
# 执行命令
|
||
command = ssh.invoke_shell() # 激活Terminal
|
||
time.sleep(3)
|
||
command.send("N\n")
|
||
command.send("screen-length 0 temporary\n")
|
||
command.send("dis cu\n")
|
||
data = []
|
||
while True:
|
||
out = command.recv(65535)
|
||
res = out.decode(encoding='UTF-8')
|
||
data.append(res)
|
||
if 'return' in res:
|
||
break
|
||
|
||
r_time = '.\\' + self.out_filepath + '\\' +devName + '.txt'
|
||
res = ''.join(data)
|
||
f1 = open(r_time, 'w')
|
||
f1.write(res)
|
||
f1.close()
|
||
with open(r_time, 'r') as f2:
|
||
final_data = f2.read().replace('\n\n', '\n')
|
||
f2.close()
|
||
f3 = open(r_time, 'w')
|
||
f3.write(final_data)
|
||
f3.close()
|
||
ssh.close()
|
||
print('【{0}】执行完毕！'.format(devName))
|
||
except:
|
||
print('【ERROR】连接 [{0}] 超时！'.format(devName))
|
||
|
||
def setH3C(self, devName, conf_ip, conf_uname, conf_pass):
|
||
try:
|
||
# 创建ssh对象
|
||
ssh = paramiko.SSHClient()
|
||
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
ssh.connect(hostname=conf_ip, port=22, username=conf_uname, password=conf_pass)
|
||
# 执行命令
|
||
command = ssh.invoke_shell() # 激活Terminal
|
||
time.sleep(3)
|
||
command.send("screen-length disable\n")
|
||
command.send("dis cu\n")
|
||
data = []
|
||
while True:
|
||
out = command.recv(65535)
|
||
res = out.decode(encoding='UTF-8')
|
||
data.append(res)
|
||
if 'return' in res:
|
||
break
|
||
|
||
r_time = '.\\' + self.out_filepath + '\\' + devName + '.txt'
|
||
res = ''.join(data)
|
||
f1 = open(r_time, 'w')
|
||
f1.write(res)
|
||
f1.close()
|
||
with open(r_time, 'r') as f2:
|
||
final_data = f2.read().replace('\n\n\n', '\n')
|
||
f2.close()
|
||
f3 = open(r_time, 'w')
|
||
f3.write(final_data)
|
||
f3.close()
|
||
ssh.close()
|
||
print('【{0}】执行完毕！'.format(devName))
|
||
except:
|
||
print('【ERROR】连接 [{0}] 超时！'.format(devName))
|
||
|
||
def setZhongxing(self, devName, conf_ip, conf_uname, conf_pass):
|
||
try:
|
||
# 创建ssh对象
|
||
ssh = paramiko.SSHClient()
|
||
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
ssh.connect(hostname=conf_ip, port=22, username=conf_uname, password=conf_pass)
|
||
# 执行命令
|
||
command = ssh.invoke_shell() # 激活Terminal
|
||
time.sleep(3)
|
||
command.send("enable\n")
|
||
command.send("terminal length 0\n")
|
||
command.send("show running-config\n")
|
||
data = []
|
||
while True:
|
||
out = command.recv(65535)
|
||
res = out.decode(encoding='UTF-8')
|
||
data.append(res)
|
||
if '!</static>' in res:
|
||
break
|
||
|
||
r_time = '.\\' + self.out_filepath + '\\' + devName + '.txt'
|
||
res = ''.join(data)
|
||
f1 = open(r_time, 'w')
|
||
f1.write(res)
|
||
f1.close()
|
||
with open(r_time, 'r') as f2:
|
||
final_data = f2.read().replace('\n\n', '\n')
|
||
f2.close()
|
||
f3 = open(r_time, 'w')
|
||
f3.write(final_data)
|
||
f3.close()
|
||
ssh.close()
|
||
print('【{0}】执行完毕！'.format(devName))
|
||
except:
|
||
print('【ERROR】连接 [{0}] 超时！'.format(devName))
|
||
|
||
def setRuijie(self, devName, conf_ip, conf_uname, conf_pass):
|
||
try:
|
||
# 创建ssh对象
|
||
ssh = paramiko.SSHClient()
|
||
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
ssh.connect(hostname=conf_ip, port=22, username=conf_uname, password=conf_pass)
|
||
# 执行命令
|
||
command = ssh.invoke_shell() # 激活Terminal
|
||
time.sleep(3)
|
||
command.send("terminal length 0\n")
|
||
command.send("show running-config\n")
|
||
command.send("show hosts\n")
|
||
data = []
|
||
while True:
|
||
out = command.recv(65535)
|
||
res = out.decode(encoding='UTF-8')
|
||
data.append(res)
|
||
if "!end" in res or "Host" in res:
|
||
break
|
||
|
||
r_time = '.\\' + self.out_filepath + '\\' + devName + '.txt'
|
||
res = ''.join(data)
|
||
f1 = open(r_time, 'w')
|
||
f1.write(res)
|
||
f1.close()
|
||
with open(r_time, 'r') as f2:
|
||
final_data = f2.read().replace('\n\n\n', '\n')
|
||
f2.close()
|
||
f3 = open(r_time, 'w')
|
||
f3.write(final_data)
|
||
f3.close()
|
||
ssh.close()
|
||
print('【{0}】执行完毕！'.format(devName))
|
||
except:
|
||
print('【ERROR】连接 [{0}] 超时！'.format(devName))
|
||
|
||
|
||
def setFenghuo(self, devName, conf_ip, conf_uname, conf_pass):
|
||
try:
|
||
# 创建ssh对象
|
||
ssh = paramiko.SSHClient()
|
||
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
ssh.connect(hostname=conf_ip, port=22, username=conf_uname, password=conf_pass)
|
||
# 执行命令
|
||
command = ssh.invoke_shell() # 激活Terminal
|
||
time.sleep(3)
|
||
command.send("terminal length 0\n")
|
||
command.send("show running-config\n")
|
||
data = []
|
||
while True:
|
||
out = command.recv(65535)
|
||
res = out.decode(encoding='UTF-8')
|
||
data.append(res)
|
||
if "!end" in res or "Host" in res:
|
||
break
|
||
|
||
r_time = '.\\' + self.out_filepath + '\\' + devName + '.txt'
|
||
res = ''.join(data)
|
||
f1 = open(r_time, 'w')
|
||
f1.write(res)
|
||
f1.close()
|
||
with open(r_time, 'r') as f2:
|
||
final_data = f2.read().replace('\n\n', '\n')
|
||
f2.close()
|
||
f3 = open(r_time, 'w')
|
||
f3.write(final_data)
|
||
f3.close()
|
||
ssh.close()
|
||
print('【{0}】执行完毕！'.format(devName))
|
||
except:
|
||
print('【ERROR】连接 [{0}] 超时！'.format(devName))
|
||
|
||
|
||
# ip封堵 类，只需调用main函数
|
||
class ip_block:
|
||
#初始化一些变量属性
|
||
def __init__(self):
|
||
self.jifang_list = []
|
||
self.jifang_select = ''
|
||
self.block_ip = ''
|
||
self.CurrBlockDev_Info = []
|
||
ColorLogDecorator.active()
|
||
|
||
# 读取和初始化封堵设备信息
|
||
def Init_jifang_list(self):
|
||
if os.path.isfile('blockconfig.ini'):
|
||
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
|
||
if config['list']['jifang'] == '':
|
||
return False
|
||
else:
|
||
if '|' in config['list']['jifang']:
|
||
self.jifang_list = config['list']['jifang'].split('|')
|
||
else:
|
||
self.jifang_list.append(config['list']['jifang'])
|
||
return True
|
||
else:
|
||
return False
|
||
|
||
|
||
#执行封堵
|
||
def startFengdu(self):
|
||
s = ''
|
||
for single_word in pypinyin.pinyin(self.jifang_select, style=pypinyin.NORMAL):
|
||
s += ''.join(single_word)
|
||
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
|
||
ipblock_Devices = config['devList'][s]
|
||
if ',' in ipblock_Devices:
|
||
Curr_devList = ipblock_Devices.split(',')
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
self.CurrBlockDev_Info = []
|
||
for a in Curr_devList:
|
||
Currsql_res = c.execute("SELECT * FROM devices WHERE ID='{0}'".format(a))
|
||
Currsql_res_List = list(Currsql_res)
|
||
print(Currsql_res_List)
|
||
self.CurrBlockDev_Info.append(Currsql_res_List)
|
||
if 'HUAWEI' in Currsql_res_List[0][3] or 'huawei' in Currsql_res_List[0][3] or '华为' in Currsql_res_List[0][3]:
|
||
if self.setHuawei():
|
||
print(ColorLogDecorator().green('【INFO】封堵完成！请自行Ping测是否有效', 'strong'))
|
||
# self.exportToFile()
|
||
# print(ColorLogDecorator.green('【INFO】已导出配置记录表和封堵脚本至程序运行目录下', 'strong'))
|
||
elif 'H3C' in Currsql_res_List[0][3] or '华三' in Currsql_res_List[0][3]:
|
||
if self.setH3C():
|
||
print(ColorLogDecorator().green('【INFO】封堵完成！请自行Ping测是否有效', 'strong'))
|
||
# self.exportToFile()
|
||
elif 'RUIJIE'in Currsql_res_List[0][3] or '锐捷' in Currsql_res_List[0][3]:
|
||
if self.setRuijie():
|
||
pass
|
||
#self.exportToFile()
|
||
elif 'ZHONGXING'in Currsql_res_List[0][3] or 'ZTE' in Currsql_res_List[0][3] or '中兴'in Currsql_res_List[0][3]:
|
||
if self.setZhongxing():
|
||
pass
|
||
# self.exportToFile()
|
||
else:
|
||
print(ColorLogDecorator().red('【ERROR】抱歉，本程序不支持该厂商设备！', 'strong'))
|
||
|
||
conn.commit()
|
||
conn.close()
|
||
os.system('pause')
|
||
elif ipblock_Devices.isdigit() == True:
|
||
conn = sqlite3.connect('alldata.db')
|
||
c = conn.cursor()
|
||
Currsql_res = c.execute("SELECT * FROM devices WHERE ID='{0}'".format(ipblock_Devices))
|
||
Currsql_res_List = list(Currsql_res)
|
||
if 'HUAWEI' in Currsql_res_List[0][3] or 'huawei' in Currsql_res_List[0][3] or '华为' in Currsql_res_List[0][3]:
|
||
if self.setHuawei():
|
||
print(ColorLogDecorator().green('【INFO】封堵完成！请自行Ping测是否有效', 'strong'))
|
||
# self.exportToFile()
|
||
# print(ColorLogDecorator.green('【INFO】已导出配置记录表和封堵脚本至程序运行目录下', 'strong'))
|
||
elif 'H3C' in Currsql_res_List[0][3] or '华三' in Currsql_res_List[0][3]:
|
||
if self.setH3C():
|
||
print(ColorLogDecorator().green('【INFO】封堵完成！请自行Ping测是否有效', 'strong'))
|
||
#self.exportToFile()
|
||
elif 'RUIJIE' in Currsql_res_List[0][3] or '锐捷' in Currsql_res_List[0][3]:
|
||
if self.setRuijie():
|
||
pass
|
||
#self.exportToFile()
|
||
elif 'ZHONGXING' in Currsql_res_List[0][3] or 'ZTE' in Currsql_res_List[0][3] or '中兴' in Currsql_res_List[0][3]:
|
||
if self.setZhongxing():
|
||
pass
|
||
#self.exportToFile()
|
||
else:
|
||
print(ColorLogDecorator.red('【ERROR】抱歉，本程序不支持该厂商设备！', 'strong'))
|
||
os.system('pause')
|
||
else:
|
||
print(ColorLogDecorator.red('【ERROR】抱歉，发生未知错误！', 'strong'))
|
||
os.system('pause')
|
||
|
||
#导出封堵脚本和配置记录表
|
||
def exportToFile(self):
|
||
pass
|
||
|
||
def setHuawei(self):
|
||
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
|
||
if config['account']['username'] == '' or config['account']['password'] == '':
|
||
print(ColorLogDecorator().red('【ERROR】设备通用SSH账号密码未设置，请设置后再使用本功能！', 'strong'))
|
||
return False
|
||
for i in range(len(self.CurrBlockDev_Info)):
|
||
# 创建ssh对象
|
||
ssh = paramiko.SSHClient()
|
||
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
ssh.connect(hostname=self.CurrBlockDev_Info[i][0][2], port=22, username=config['account']['username'], password=config['account']['password'])
|
||
# 执行命令
|
||
command = ssh.invoke_shell() # 激活Terminal
|
||
time.sleep(3)
|
||
command.send("N\n")
|
||
command.send("sys\n")
|
||
command.send("ip route-static {0} 255.255.255.255 NULL0\n".format(self.block_ip))
|
||
# VRP8版本的高端设备需加commit
|
||
command.send("commit\n")
|
||
command.send("return\n")
|
||
command.send("save\n")
|
||
command.send("Y\n")
|
||
time.sleep(3)
|
||
# command.recv(65535) #这里可接收结果写入日志
|
||
ssh.close()
|
||
|
||
return True
|
||
|
||
|
||
def setRuijie(self):
|
||
pass
|
||
|
||
def setH3C(self):
|
||
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
|
||
if config['account']['username'] == '' or config['account']['password'] == '':
|
||
print(ColorLogDecorator().red('【ERROR】设备通用SSH账号密码未设置，请设置后再使用本功能！', 'strong'))
|
||
return False
|
||
for i in range(len(self.CurrBlockDev_Info)):
|
||
# 创建ssh对象
|
||
ssh = paramiko.SSHClient()
|
||
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
ssh.connect(hostname=self.CurrBlockDev_Info[i][0][2], port=22, username=config['account']['username'],password=config['account']['password'])
|
||
# 执行命令
|
||
command = ssh.invoke_shell() # 激活Terminal
|
||
time.sleep(3)
|
||
command.send("sys\n")
|
||
command.send("ip route-static {0} 255.255.255.255 NULL0\n".format(self.block_ip))
|
||
command.send("return\n")
|
||
command.send("save\n")
|
||
command.send("Y\n")
|
||
command.send("\n")
|
||
command.send("Y\n")
|
||
time.sleep(1)
|
||
# command.recv(65535) #这里可接收结果写入日志
|
||
ssh.close()
|
||
print(ColorLogDecorator().green('【INFO】封堵完成！请自行Ping测是否有效', 'strong'))
|
||
return True
|
||
|
||
def setZhongxing(self):
|
||
pass
|
||
|
||
def setCisco(self):
|
||
pass
|
||
|
||
|
||
def main(self):
|
||
if self.Init_jifang_list():
|
||
while True:
|
||
os.system('cls')
|
||
|
||
print('''
|
||
===============================================
|
||
[选择] >>要封堵的机房<<
|
||
''')
|
||
for i in range(len(self.jifang_list)):
|
||
print(' {0}. {1}'.format(i+1, self.jifang_list[i]))
|
||
|
||
print('\n (tips: 返回上一级菜单请输入q或quit)')
|
||
print('===============================================\n')
|
||
sel_jifang = input('请输入：')
|
||
if sel_jifang == 'q' or sel_jifang == 'quit' or sel_jifang == 'Q':
|
||
break
|
||
elif sel_jifang.isdigit() == True and int(sel_jifang) <= len(self.jifang_list) and int(sel_jifang) != 0:
|
||
self.jifang_select = self.jifang_list[int(sel_jifang)-1]
|
||
|
||
print('\n\n')
|
||
print('*********************************************')
|
||
inBlockIP = input(' 请输入要封堵的IP：')
|
||
check_ip = re.compile('^(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|[1-9])\.(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|\d)\.(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|\d)\.(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|\d)$')
|
||
if check_ip.match(inBlockIP):
|
||
self.block_ip = inBlockIP
|
||
self.startFengdu()
|
||
os.system('cls')
|
||
continue
|
||
else:
|
||
print(ColorLogDecorator().red('【ERROR】IP不合法！请重新操作！', 'strong'))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
else:
|
||
print(ColorLogDecorator().red('【ERROR】ip封堵设备信息库为空，请添加相关封堵设备！', 'strong'))
|
||
os.system('pause')
|
||
|
||
### 自定义颜色
|
||
class ColorLogDecorator:
|
||
"""
|
||
class: ColorLogDecorator
|
||
Desc: ues for decorate the string with ANSI escape code (color function)
|
||
|
||
class_var:
|
||
__IS_ACTIVE: whether the decorate func active
|
||
__DEFAULT_STYLE: the default style for a color selected
|
||
__END_CODE: the end escape code
|
||
"""
|
||
|
||
__IS_ACTIVE = False
|
||
__DEFAULT_STYLE = "normal"
|
||
__COLOR_CODE = {
|
||
"red": {
|
||
"normal": "\033[31m",
|
||
"strong": "\033[1;31m",
|
||
"bg": "\033[0;37;41m",
|
||
"bg-strong": "\033[1;37;41m",
|
||
},
|
||
"green": {
|
||
"normal": "\033[32m",
|
||
"strong": "\033[1;32m",
|
||
"bg": "\033[0;37;42m",
|
||
"bg-strong": "\033[1;37;42m",
|
||
},
|
||
"yellow": {
|
||
"normal": "\033[33m",
|
||
"strong": "\033[1;33m",
|
||
"bg": "\033[0;30;43m",
|
||
"bg-strong": "\033[1;30;43m",
|
||
},
|
||
"blue": {
|
||
"normal": "\033[34m",
|
||
"strong": "\033[1;34m",
|
||
"bg": "\033[0;37;44m",
|
||
"bg-strong": "\033[1;37;44m",
|
||
},
|
||
"black": {
|
||
"normal": "\033[30m",
|
||
"strong": "\033[1;30m",
|
||
"bg": "\033[0;37;40m",
|
||
"bg-strong": "\033[1;37;40m",
|
||
},
|
||
"white": {
|
||
"normal": "\033[37m",
|
||
"strong": "\033[1;37;0m",
|
||
"bg": "\033[0;30;47m",
|
||
"bg-strong": "\033[1;30;47m",
|
||
}
|
||
}
|
||
__END_CODE = "\033[0m"
|
||
|
||
@classmethod
|
||
def active(cls) -> None:
|
||
"""
|
||
active the color decorate function
|
||
it will use a special menthol for windows os
|
||
:return: None
|
||
"""
|
||
cls.__IS_ACTIVE = True
|
||
if os.name == "nt":
|
||
os.system("")
|
||
|
||
@classmethod
|
||
def deactivate(cls) -> None:
|
||
"""
|
||
deactivate the color decorate function
|
||
:return: None
|
||
"""
|
||
cls.__IS_ACTIVE = False
|
||
|
||
@classmethod
|
||
def __msg_decorator(cls, msg: str, color: str, style: str) -> str:
|
||
"""
|
||
use to decorate the msg str with special style color escape code
|
||
:param msg: the msg str
|
||
:param color: the color str to select
|
||
:param style: the style str to select
|
||
:return: decorated str
|
||
"""
|
||
if not cls.__IS_ACTIVE:
|
||
return msg
|
||
|
||
style_selected = cls.__DEFAULT_STYLE if style.lower() not in cls.__COLOR_CODE[color].keys() \
|
||
else style.lower()
|
||
|
||
return cls.__COLOR_CODE[color][style_selected] + msg + cls.__END_CODE
|
||
|
||
@classmethod
|
||
def red(cls, msg: str, style: str = "normal") -> str:
|
||
"""
|
||
red log str
|
||
:param msg: the msg str
|
||
:param style: the style to select
|
||
:return: decorated str
|
||
"""
|
||
return cls.__msg_decorator(msg, "red", style)
|
||
|
||
@classmethod
|
||
def green(cls, msg: str, style: str = "normal") -> str:
|
||
"""
|
||
green log str
|
||
:param msg: the msg str
|
||
:param style: the style to select
|
||
:return: decorated str
|
||
"""
|
||
return cls.__msg_decorator(msg, "green", style)
|
||
|
||
@classmethod
|
||
def yellow(cls, msg: str, style: str = "normal") -> str:
|
||
"""
|
||
yellow log str
|
||
:param msg: the msg str
|
||
:param style: the style to select
|
||
:return: decorated str
|
||
"""
|
||
return cls.__msg_decorator(msg, "yellow", style)
|
||
|
||
@classmethod
|
||
def blue(cls, msg: str, style: str = "normal") -> str:
|
||
"""
|
||
blue log str
|
||
:param msg: the msg str
|
||
:param style: the style to select
|
||
:return: decorated str
|
||
"""
|
||
return cls.__msg_decorator(msg, "blue", style)
|
||
|
||
@classmethod
|
||
def black(cls, msg: str, style: str = "normal") -> str:
|
||
"""
|
||
black log str
|
||
:param msg: the msg str
|
||
:param style: the style to select
|
||
:return: decorated str
|
||
"""
|
||
return cls.__msg_decorator(msg, "black", style)
|
||
|
||
@classmethod
|
||
def white(cls, msg: str, style: str = "normal") -> str:
|
||
"""
|
||
white log str
|
||
:param msg: the msg str
|
||
:param style: the style to select
|
||
:return: decorated str
|
||
"""
|
||
return cls.__msg_decorator(msg, "white", style)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
menu = """
|
||
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||
+ 简易运维自动化工具-测试版 +
|
||
+ +
|
||
+ 1. 导出所有网络设备配置信息 +
|
||
+ +
|
||
+ 2. Ping测ip地址（暂未开发） +
|
||
+ +
|
||
+ 3. 封堵ip地址 +
|
||
+ +
|
||
+ 4. 管理和查看设备信息数据库 +
|
||
+ +
|
||
+ ----------------------------- +
|
||
+ 本工具使用帮助 请输入 help +
|
||
+ 作者信息 请输入 author +
|
||
+ +
|
||
+ Powered By HuiJi +
|
||
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||
"""
|
||
|
||
while True:
|
||
print(menu)
|
||
currInput = handleMenu()
|
||
if currInput == "quit" or currInput == "q":
|
||
os.system('cls')
|
||
continue
|
||
elif currInput == "1":
|
||
devOut = devConfig_export()
|
||
devOut.main()
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
elif currInput == "2":
|
||
os.system('cls')
|
||
ColorLogDecorator().active()
|
||
print(ColorLogDecorator().blue('\n【INFO】功能待开发！', 'strong'))
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
elif currInput == "3":
|
||
use_test = ip_block()
|
||
use_test.main()
|
||
#os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
elif currInput == "4":
|
||
os.system('cls')
|
||
manageLocalDB()
|
||
os.system('cls')
|
||
continue
|
||
elif currInput == "help":
|
||
os.system('cls')
|
||
print('''
|
||
----------------------------------------------------------------------------
|
||
本工具可用于自动化运维，代替一些重复的事情，后续有时间会学习PyQT或
|
||
其它任何可行方式，使工具改成UI界面更便于使用
|
||
|
||
***重点*** -----本工具完美适配Windows系统，其它操作系统下会报错
|
||
|
||
使用说明：
|
||
*输入对应序列号或特定指令即可操作
|
||
*首次使用，请先进入[4]添加设备信息进数据库
|
||
*封堵功能首次使用准备步骤：添加设备信息进数据库->添加封堵设备->IP封堵
|
||
由于本人工作中只用到华为和H3C的设备，其它设备的封堵命令没有写进去。
|
||
*导出所有设备配置信息即：导出所有配置记录，目前只支持 华为、H3C、
|
||
锐捷、中兴、烽火 设备。
|
||
----------------------------------------------------------------------------
|
||
''')
|
||
print("\n\n")
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
elif currInput == "author":
|
||
os.system('cls')
|
||
print('''
|
||
------------------------------------------------
|
||
本人是一名网工，在工作中会有经常性的
|
||
重复工作，写这个工具目的除了能使工作
|
||
简要化之外，还能提升编程能力。
|
||
*作者网名：灰机
|
||
*具备技能：
|
||
Python、Linux、网络管理、路由交换、
|
||
PHP 等
|
||
*个人博客：blog.huiji888.cn
|
||
|
||
平时喜欢写代码、搭建和维护网站、搞搞网络，
|
||
希望各路大佬前来交流，互相促进。
|
||
-------------------------------------------------
|
||
''')
|
||
print("\n\n")
|
||
os.system('pause')
|
||
os.system('cls')
|
||
continue
|
||
|