Felix-sec/IDCtools
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
63f164f6e273e2c981ec1005b61f1fc3d7c6f1da
IDCtools/idctools.py

1431 lines
72 KiB
Python
Raw Normal View History

Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
# -*- 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'):
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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'):
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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(",")
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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:
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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')
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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'):
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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)
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
res = out.decode(encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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)
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
res = out.decode(encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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)
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
res = out.decode(encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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)
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
res = out.decode(encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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)
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
res = out.decode(encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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'):
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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)
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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)
2021-8-6修复若干bug 测试完成
2021-08-06 15:28:42 +08:00
print(Currsql_res_List)
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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():
2021-8-6修复若干bug 测试完成
2021-08-06 15:28:42 +08:00
print(ColorLogDecorator().green('【INFO】封堵完成！请自行Ping测是否有效', 'strong'))
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
# 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():
2021-8-6修复若干bug 测试完成
2021-08-06 15:28:42 +08:00
print(ColorLogDecorator().green('【INFO】封堵完成！请自行Ping测是否有效', 'strong'))
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
# 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():
2021-8-6修复若干bug 测试完成
2021-08-06 15:28:42 +08:00
print(ColorLogDecorator().green('【INFO】封堵完成！请自行Ping测是否有效', 'strong'))
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
# 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():
2021-8-6修复若干bug 测试完成
2021-08-06 15:28:42 +08:00
print(ColorLogDecorator().green('【INFO】封堵完成！请自行Ping测是否有效', 'strong'))
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
#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):
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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())
2021-8-6修复若干bug 测试完成
2021-08-06 15:28:42 +08:00
ssh.connect(hostname=self.CurrBlockDev_Info[i][0][2], port=22, username=config['account']['username'], password=config['account']['password'])
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
# 执行命令
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")
2021-8-6修复若干bug 测试完成
2021-08-06 15:28:42 +08:00
time.sleep(3)
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
# command.recv(65535) #这里可接收结果写入日志
ssh.close()
2021-8-6修复若干bug 测试完成
2021-08-06 15:28:42 +08:00
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
return True
def setRuijie(self):
pass
def setH3C(self):
2021-8-6修复若干bug
2021-08-06 17:17:54 +08:00
config = ConfigObj('blockconfig.ini', encoding='UTF-8')
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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())
2021-8-6修复若干bug 测试完成
2021-08-06 15:28:42 +08:00
ssh.connect(hostname=self.CurrBlockDev_Info[i][0][2], port=22, username=config['account']['username'],password=config['account']['password'])
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
# 执行命令
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)):
2021-8-6修复若干bug 测试完成
2021-08-06 15:28:42 +08:00
print(' {0}. {1}'.format(i+1, self.jifang_list[i]))
Add files via upload 2021-8-5 v1.0.85
2021-08-05 16:16:46 +08:00
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
Reference in New Issue Copy Permalink