发帖
翼度科技»论坛 › 编程开发 › python › 查看内容

交换机相关最终

狗蛋村书记

5

主题

5

帖子

15

积分

新手上路

Rank: 1

积分
15
显示全部楼层

  • 用判断语句实现所有可能场景
    `
    import telnetlib
    import time
import pandas as pd
import re
def telnet_login(host_ip,username,password):
tn=telnetlib.Telnet()
try:
tn.open(host_ip,port=23,timeout=5)
tn.read_until(b'Username:',timeout=5)
tn.write(username.encode('ascii')+b'\n')
tn.read_until(b'Password:',timeout=5)
tn.write(password.encode('ascii')+b'\n')
time.sleep(1)
  1.     command_result=tn.read_until(b'#',timeout=5)
  2.     if b'#' in command_result:
  3.         print(f'{host_ip} telnet success')
  4.     else:
  5.         print(f'{host_ip} telnet failed')
  6.     return tn
  7. except:
  8.     print(f'{host_ip} telnet failed')
复制代码
def get_intface_state(tn,host_ip):
command="show ip interface brief"
cmd=bytes(command,encoding='ascii')#注意不能写成encodings='ascii'
tn.write(cmd+b'\n')
time.sleep(1)
resultlist=[]
while True:
command_result=tn.read_very_eager().decode('ascii')
#print(command_result)
#注意不能写成tn.read_very_eager('ascii')
resultlist.append(command_result)#不能写成resultlist=resultlist.append(command_result)
if re.findall(r'#',command_result.strip()):
break
elif re.findall(r'--More--',command_result.strip()):
tn.write(b" ")
else:
time.sleep(0.05)
continue
resultstr="\n".join(resultlist)
liststr=resultstr.split('\n')
pd_output=pd.DataFrame()
listtemp=[]
for j in liststr:
if len(re.findall(r'Interface\s+.+',j))>0:
new_columns=re.split(r'\s+',j.strip())
if len(re.findall(r'.+\d+.\d+.\d+.\d+\s+.+',j))>0:
list_find_str=re.split(r'\s+',j.strip())
listtemp.append(list_find_str)
#print(listtemp)
pd_output=pd.DataFrame(listtemp)
pd_output.columns=new_columns
print(pd_output)
pd_output.to_csv(f'{host_ip}.csv')
return pd_output
def get_ospf_config(tn):
try:
tn.write(b"show running-config ospfv2"+b"\n")
time.sleep(1)
result_list = []
command_result = tn.read_very_eager().decode('ascii')
result_list.append(command_result.strip())
result_str="\n".join(result_list)
result_list = result_str.split('\n')
#print(result_str)
pd_output=pd.DataFrame()
import_list = []
data = []
df_import = pd.DataFrame()
row = None
for i in result_list:
  1.         match_area = re.match(r'^\s*area\s+(\d\.\d\.\d\.\d)$', i.strip())
  2.         match_interface = re.match(r'^\s*interface\s+(\w+.+)$', i.strip())
  3.         match_network = re.match(r'^\s*network\s+(\w+.+)$', i.strip())
  4.         match_redistribute = re.match(r'^\s*redistribute\s+(\w+)$', i.strip())
  6.         if match_area:
  7.             data.append({'area': match_area.group(1)})
  8.         elif match_interface:
  9.              data[-1]['interface'] = match_interface.group(1)
  10.         elif match_network:
  11.              data[-1]['network'] = match_network.group(1)
  13.         if match_redistribute:
  14.             findstr = match_redistribute.group(1)
  15.             import_list.append(findstr)
  16.     if len(import_list) > 0:
  17.         df_import = pd.DataFrame(import_list)
  18.     else:
  19.         df_import.loc[0, 0] = ''
  20.     df_import.columns = ['import']
  21.     #print(data)
  22.     df_area = pd.DataFrame(data)
  23.     # if 'area' not in df_area.columns:
  24.     #     df_area['area']=''
  25.     # if 'interface' not in df_area.columns:
  26.     #     df_area['interface']=''
  27.     # if 'network' not in df_area.columns:
  28.     #     df_area['network']=''
  29.     df = pd.concat([df_area, df_import], axis=1)
  30.     #df['host_ip'] = host_ip
  31.     print(df)
  32.     return df
  33. except:
  34.     print('error')
复制代码
if name=='main':
hosts_ip=['192.168.8.1','192.168.2.2']
interfaces=['xgei-0/1/1/49','loopback1']
planed_ip=['10.0.0.2','2.2.2.2','10.0.0.1','3.3.3.3']
  1. username='zte'
  2. password='zte'
  3. df=pd.DataFrame()#注意不能写成df=DataFrame()
  4. df1=pd.DataFrame()
  5. df2=pd.DataFrame()
  6. #tn=telnet_login(host_ip=hosts_ip[0],username=username,password=password)
  7. df_ospf_two = pd.DataFrame()
  8. a=0
  9. for host_ip in hosts_ip:
  10.     tn=telnet_login(host_ip,username,password)
  11.     if tn:
  12.          df=get_intface_state(tn,host_ip)
  13.          for interface in interfaces:
  14.              if df[df['Interface']==interface]['Admin'].values[0]=='down':
  15.                 print(f'{interface} is admin down, please no shutdown it')
  16.              if df[df['Interface']==interface]['IP-Address'].values[0]!=planed_ip[a]:
  17.                  print(f'{interface} ip address is not {planed_ip[a]}')
  18.              a=a+1
  19.          tn.close()
  20.     else:
  21.          print('telnet failed')
  22. tn1 = telnet_login(hosts_ip[0], username, password)
  23. tn2 = telnet_login(hosts_ip[1], username, password)
  25. df1 = get_ospf_config(tn1)
  26. df2 = get_ospf_config(tn2)
  28. if 'area' not in df1.columns:
  29.     if 'interface' in df2.columns:
  30.         if interfaces[0] in df2['interface'].values:
  31.             area_config= df2[df2['interface'] == interfaces[0]]['area'].values[0]
  32.             if 'network' in df2.columns:
  33.                 network_config=df2[df2['interface']==interfaces[0]]['network'].values[0]
  34.                 print(f'router1 configuredarea:{area_config},interface:{interfaces[0]},network:{network_config}')
  35.             else:
  36.                 print(f'router 1 configured:{area_config},two router network:network p2p')
  37.         else:
  38.             area_config='0.0.0.0'
  39.             network_config='network p2p'
  40.             print(f'configured two router {area_config},{interfaces[0]},{network_config}')
  41. else:
  42.     if 'interface' in df1.columns:
  43.         if interfaces[0] in df1['interface'].values:
  44.             if 'interface' in df2.columns:
  45.                 if interfaces[0] in df2['interface'].values:
  47.                     if df1[df1['interface'] == interfaces[0]]['area'].values[0] == \
  48.                             df2[df2['interface'] == interfaces[0]]['area'].values[0]:
  49.                         if 'network' not in df1.columns:
  50.                             if 'network' in df2.columns:
  51.                                 network_config = df2[df2['interface'] == interfaces[0]]['network'].values[0]
  52.                                 print(f'router1 configured:{network_config}')
  53.                             else:
  54.                                 print('two router configured:network p2p')
  55.                         else:
  56.                             if 'network' in df2.columns:
  57.                                 if df1[df1['interface'] == interfaces[0]]['network'].values[0] != \
  58.                                         df2[df2['interface'] == interfaces[0]]['network'].values[0]:
  59.                                     network_config = df1[df1['interface'] == interfaces[0]]['network'].values[0]
  60.                                     print(f'router2 configured:{network_config}')
  61.                             else:
  62.                                 network_config = df1[df1['interface'] == interfaces[0]]['network'].values[0]
  63.                                 print(f'router2 configured:{network_config}')
  64.                     else:
  65.                         if 'network' in df1.columns:
  66.                             area_config = df1[df1['interface'] == interfaces[0]]['area'].values[0]
  67.                             network_config = df1[df1['interface'] == interfaces[0]]['network'].values[0]
  68.                             print(
  69.                                 f'router2 delete the interface adn configured_area:{area_config},{interfaces[0]},configured_network:{network_config}')
  70.                         else:
  71.                             area_config = df1[df1['interface'] == interfaces[0]]['area'].values[0]
  72.                             print(f'router1 configured:network p2p')
  73.                             print(
  74.                                 f'router2 configured_area:{area_config},{interfaces[0]},configured_network:network p2p')
  75.                 else:
  76.                     if 'network' in df1.columns:
  77.                         configured_area = df1[df1['interface'] == interfaces[0]]['area'].values[0]
  78.                         configured_network = df1[df1['interface'] == interfaces[0]]['network'].values[0]
  79.                         print(
  80.                             f'router2 configured_area:{configured_area},{interfaces[0]},configured_network:network p2p')
  81.                     else:
  82.                         configured_area = df1[df1['interface'] == interfaces[0]]['area'].values[0]
  83.                         print(
  84.                             f'router2 configured_area:{configured_area},{interfaces[0]},configured_network:network p2p')
  85.                         print(f'router1 configured:network p2p')
  86.             else:
  87.                 if 'network' in df1.columns:
  88.                     area_config = df1[df1['interface'] == interfaces[0]]['area'].values[0]
  89.                     network_config = df1[df1['interface'] == interfaces[0]]['network'].values[0]
  90.                     print(
  91.                         f'router2 configured_area:{area_config},{interfaces[0]},configured_network:{network_config}')
  92.                 else:
  93.                     area_config = df1[df1['interface'] == interfaces[0]]['area'].values[0]
  94.                     print(f'router2 configured_area:{area_config},{interfaces[0]},configured_network:network p2p')
  95.                     print(f'router1 configured:network p2p')
  96.         else:
  97.             if 'interface' in df2.columns:
  98.                 if interfaces[0] in df2['interface'].values:
  99.                     area_config = df2[df2['interface'] == interfaces[0]]['area'].values[0]
  100.                     if 'network' in df2.columns:
  101.                         network_config = df2[df2['interface'] == interfaces[0]]['network'].values[0]
  102.                         print(
  103.                             f'router1 configured_area:{area_config}, interfaces[0], configured_network:{network_config}')
  104.                     else:
  105.                         network_config = 'network p2p'
  106.                         print(
  107.                             f'router1 configured_area:{area_config}, interfaces[0], configured_network:{network_config}')
  108.                         print(f'router2  configured_network:network p2p')
  109.                 else:
  110.                     area_config = '0.0.0.0'
  111.                     network_config = 'network p2p'
  112.                     print(f'configured two router {area_config},{interfaces[0]},{network_config}')
  114.     else:
  115.         if 'interface' in df2.columns:
  116.             if interfaces[0] in df2['interface'].values:
  117.                 area_config = df2[df2['interface'] == interfaces[0]]['area'].values[0]
  118.                 if 'network' in df2.columns:
  119.                     network_config = df2[df2['interface'] == interfaces[0]]['network'].values[0]
  120.                     print(f'router1 configured_area:{area_config},{interfaces[0]},network:{network_config}')
  121.                 else:
  122.                     print(f'router 1 configured_area:{area_config},two router network:network p2p')
  123.             else:
  124.                 area_config = '0.0.0.0'
  125.                 network_config = 'network p2p'
  126.                 print(f'configured two router {area_config},{interfaces[0]},{network_config}')
复制代码
2.信心采集状态
import paramiko
import os
import time
import random
import datetime
import pandas as pd
import re
import numpy as np
from sqlalchemy import text, create_engine
import psycopg2
from psycopg2 import sql
from sqlalchemy.orm import sessionmaker
from sqlalchemy.dialects.postgresql import insert as pg_insert
from sqlalchemy import create_engine, MetaData, Table, inspect
import schedule
import telnetlib
import threading
import time
def get_info_telnet(host_ip, username, password):
  1. host_ip = '129.60.161.169'
  2. username = 'zte'
  3. password = 'zte'
  5. tn = telnetlib.Telnet()
  6. try:
  8.     tn.open(host_ip, port=23, timeout=5)
  9.     print('%s connected ssuccess !' % host_ip)
  11.     tn.read_until(b'Username:', timeout=5)
  12.     tn.write(username.encode('ascii') + b'\n')
  14.     tn.read_until(b'Password:', timeout=5)
  15.     tn.write(password.encode('ascii') + b'\n')
  16.     time.sleep(1)
  18.     command_result = tn.read_until(b'#', timeout=5)
  19.     if b'#' not in command_result:
  20.         print('%s登录失败' % host_ip)
  21.     else:
  22.         print('%s登录成功' % host_ip)
  24. except:
  26.     print('%s网络连接失败' % host_ip)
  28. command = "show clock"
  29. command = bytes(command, encoding='utf-8')
  30. tn.write(command + b'\r\n')
  31. run_time = tn.read_until(b'#')
  32. run_time = re.findall(r"\d+:\d+:\d+\s+\w+\s+\w+\s+\w+\s+\d+\s+2024", run_time.decode('GB18030'))[0]
  34. command = "show interface brief"
  35. command = bytes(command, encoding='utf-8')
  36. tn.write(command + b'\n')
  37. time.sleep(1)
  39. result_list = []
  40. while (True):
  41.     command_result = tn.read_very_eager().decode('ascii')
  42.     # print(command_result)
  43.     result_list.append(command_result)
  44.     if re.findall(r"--More--", command_result.strip()):
  45.         tn.write(b" ")
  47.     elif re.findall(r"#", command_result.strip()):
  48.         break
  49.     else:
  50.         time.sleep(0.05)
  51.         continue
  53. result_str = "\n".join(result_list)
  54. list_str = result_str.split('\n')
  56. pd_result = pd.DataFrame()
  57. list_temperature_vec = []
  58. for j in list_str:
  59.     regex = re.compile(r'\w+gei.+\s+.+\s+.+\s+.+\s+.+\s+.+\s+.+', re.S)
  60.     # print(regex.findall(j))
  61.     # print(len(regex.findall(j)))
  62.     if len(re.findall(r"Interface", j)) > 0:
  63.         new_columns = list_find_str = re.split(r'\s+', j)
  64.         new_columns = new_columns[0:8]
  66.     if len(regex.findall(j)) > 0:
  67.         list_find_str = regex.findall(j)[0]
  68.         list_find_str = re.split(r'\s+', list_find_str)
  69.         list_temperature_vec.append(list_find_str)
  70. pd_result = pd.DataFrame(list_temperature_vec)
  71. pd_result.columns = new_columns
  74. tn.close()
  75. return pd_result
复制代码
if name == 'main':
  1. host_ip = '129.60.161.169'
  2. username = 'zte'
  3. password = 'zte'
  4. pd_output = get_info_telnet(host_ip, username, password)
复制代码
`
3.故障数据分析
`
import seaborn as sns
import matplotlib.pyplot as plt
import matplotlib as mpl
import os
import numpy as np
import math
import random
import threading
from datetime import datetime, timedelta
import pandas as pd
from sqlalchemy import create_engine, text
from sqlalchemy.orm import sessionmaker
from datetime import datetime
import json
import re
import time
import itertools
import warnings
warnings.filterwarnings("ignore")
if name == 'main':
try:
engine = create_engine('postgresql://5ga-cmcc:5ga-cmcc@localhost:5432/postgres')
Session = sessionmaker(bind=engine)
query = """
SELECT * FROM traffic_table where port_name = 'xgei-0/2/0/24' ORDER BY time desc LIMIT 200;
"""
with Session() as session:
result = session.execute(text(query))
data = pd.DataFrame(result.fetchall(), columns=result.keys())
data['time'] = pd.to_datetime(data['time'], format='%Y-%m-%d %H%M%S')
data.sort_values('time', inplace=True)
data.reset_index(level=None, drop=True, inplace=True)
session.close()
# 转换结果为列表
  1.     data['rx_bytes'] = data['rx_bytes'].diff(1)
  2.     data['tx_bytes'] = data['tx_bytes'].diff(1)
  4.     # data.index = pd.DatetimeIndex(data['time'])
  5.     # data = data.resample('10s',closed='right').max()
  7.     data.dropna(axis=0, how='any', inplace=True)
  9. except Exception as e:
  10.     print(e)
  12. # 边缘图
  13. import scipy.stats as stats
  14. df = data.loc[:, ['time', 'rx_bytes']]
  15. df.columns = ['time', 'data']
  17. fig = plt.figure(figsize=(8, 6), dpi=80)
  18. grid = plt.GridSpec(4, 4, hspace=0.5, wspace=0.2)
  20. ax_main = fig.add_subplot(grid[:-1, :-1], xticklabels=[])
  21. ax_right = fig.add_subplot(grid[:-1, -1], xticklabels=[], yticklabels=[])
  23. ax_main.scatter('time', 'data',  alpha=.9, data=df)
  24. sns.boxplot(df.data, ax=ax_right, orient="v")
  26. print(df.describe())
  28. # 正态分布
  29. for i in ['rx_bytes', 'tx_bytes']:
  30.     print(i)
  31.     X = data[i].astype(float)
  32.     plt.figure(figsize=(18, 8), dpi=200)
  33.     plt.title(i+'直方图', color='dimgray')
  34.     plt.hist(X, bins=50)
  35.     plt.show()
  37.     plt.figure(figsize=(18, 8), dpi=200)
  38.     plt.title(i+'概率密度图', color='dimgray')
  39.     sns.kdeplot(X, kernel='gau', color="g", alpha=.7)
  40.     plt.show()
  42.     print(stats.skew(X))  # 计算偏度
  43.     print(stats.kurtosis(X))  # 计算峰度
  45. df = data.loc[:, ['tx_bytes', 'rx_bytes']]
  48. # 相关性
  49. ts = data.loc[:, ['rx_bytes']]
  50. ts.plot(kind="line")  # 默认绘制折线图
  52. ts = data.loc[:, ['tx_bytes']]
  53. ts.plot(kind="line")  # 默认绘制折线图
  55. ts = data.loc[:, ['tx_bytes', 'rx_bytes']]
  56. ts.plot(kind="line")  # 默认绘制折线图
  58. # 协方差：绝对值越大，线性关系越强
  59. data['tx_bytes'].cov(data['rx_bytes'])
  60. # 相关系数：相关系数在-1到1之间，接近1为正相关，接近-1为负相关，0为不相关。
  61. data['tx_bytes'].corr(data['rx_bytes'])
复制代码
4.创建数据库
import os
import re
import glob
import datetime
import pandas as pd
from sqlalchemy import text, create_engine
import psycopg2
from psycopg2 import sql
import psycopg2
from psycopg2 import sql
def create_DB(schema_name, table_name, columns_info, db_config):
try:
# 创建到数据库的连接
conn = psycopg2.connect(**db_config)
cur = conn.cursor()
  1.     # 如果不存在，则创建架构
  2.     cur.execute(sql.SQL("CREATE SCHEMA IF NOT EXISTS {}").format(sql.Identifier(schema_name)))
  3.     conn.commit()
  5.     # 构造 CREATE TABLE 查询
  6.     column_defs = [
  7.         sql.SQL('{} {}').format(
  8.             sql.Identifier(col_name),
  9.             sql.SQL(columns_info[col_name]['type']) if 'type' in columns_info[col_name] else sql.SQL('TEXT')
  10.         ) + (sql.SQL(' PRIMARY KEY') if columns_info[col_name].get('primary_key', False) else sql.SQL(''))
  11.         for col_name in columns_info
  12.     ]
  14.     create_table_query = sql.SQL("""
  15.         CREATE TABLE IF NOT EXISTS {}.{} (
  16.             {}
  17.         )
  18.     """).format(
  19.         sql.Identifier(schema_name),
  20.         sql.Identifier(table_name),
  21.         sql.SQL(', ').join(column_defs)
  22.     )
  24.     # 创建表
  25.     cur.execute(create_table_query)
  26.     conn.commit()
  28.     # 删除表中的所有数据（如果有的话）
  29.     delete_query = sql.SQL("DELETE FROM {}.{}").format(
  30.         sql.Identifier(schema_name),
  31.         sql.Identifier(table_name)
  32.     )
  33.     cur.execute(delete_query)
  34.     conn.commit()
  36.     print(f"成功在架构 {schema_name} 中创建表 {table_name}")
  38. except psycopg2.Error as e:
  39.     print(f"创建表时出错: {e}")
  40.     if conn:
  41.         conn.rollback()  # 回滚事务以撤销当前的更改
  43. finally:
  44.     if cur:
  45.         cur.close()
  46.     if conn:
  47.         conn.close()
复制代码
创建数据表

if name == 'main':
db_config = {
'dbname': 'postgres',
'user': '5ga-cmcc',
'password': '5ga-cmcc',
'host': 'localhost',
'port': '5432'
}
schema_name = 'public'
  1. columns_info = {
  2.     'time': {'type': 'timestamp without time zone'},
  3.     'port_name': {'type': 'CHARACTER VARYING'},
  4.     'rx_bytes': {'type': 'integer'},
  5.     'tx_bytes': {'type': 'integer'},
  6.     'id': {'type': 'text'}
  7. }
  8. table_name = 'traffic_table'
  10. create_DB(schema_name, table_name, columns_info, db_config)
复制代码
5.信息采集
import os
import time
import datetime
import pandas as pd
import schedule
import telnetlib
import paramiko
from sqlalchemy import create_engine, MetaData, Table, inspect
from sqlalchemy.orm import sessionmaker
from sqlalchemy.dialects.postgresql import insert as pg_insert
获取环境变量中的凭据

TELNET_HOST = os.getenv('TELNET_HOST', '129.60.161.169')
TELNET_USERNAME = os.getenv('TELNET_USERNAME', 'zte')
TELNET_PASSWORD = os.getenv('TELNET_PASSWORD', 'zte')
SSH_HOST = os.getenv('SSH_HOST', '10.89.164.70')
SSH_USERNAME = os.getenv('SSH_USERNAME', 'root')
SSH_PASSWORD = os.getenv('SSH_PASSWORD', 'PON@jkfx')
DATABASE_URI = os.getenv('DATABASE_URI', 'postgresql://5ga-cmcc:5ga-cmcc@127.0.0.1:5432/postgres')
TABLE_NAME = 'traffic_table'
def get_info_telnet(host_ip, username, password):
tn = telnetlib.Telnet()
try:
tn.open(host_ip, port=23, timeout=5)
print(f'{host_ip} connected successfully!')
  1.     tn.read_until(b'Username:', timeout=5)
  2.     tn.write(username.encode('ascii') + b'\n')
  4.     tn.read_until(b'Password:', timeout=5)
  5.     tn.write(password.encode('ascii') + b'\n')
  6.     time.sleep(1)
  8.     command_result = tn.read_until(b'#', timeout=5)
  9.     if b'#' not in command_result:
  10.         print(f'{host_ip} login failed')
  11.     else:
  12.         print(f'{host_ip} login success')
  14.     command = "show clock"
  15.     command = bytes(command, encoding='utf-8')
  16.     tn.write(command + b'\r\n')
  17.     run_time = tn.read_until(b'#')
  18.     run_time = re.findall(r"\d+:\d+:\d+\s+\w+\s+\w+\s+\w+\s+\d+\s+2024", run_time.decode('GB18030'))[0]
  20.     command = "show ip interface brief"
  21.     command = bytes(command, encoding='utf-8')
  22.     tn.write(command + b'\n')
  23.     time.sleep(1)
  25.     result_list = []
  26.     while True:
  27.         command_result = tn.read_very_eager().decode('ascii')
  28.         result_list.append(command_result)
  29.         if re.findall(r"--More--", command_result.strip()):
  30.             tn.write(b" ")
  31.         elif re.findall(r"#", command_result.strip()):
  32.             break
  33.         else:
  34.             time.sleep(0.05)
  36.     result_str = "\n".join(result_list)
  37.     list_str = result_str.split('\n')
  39.     pd_result = pd.DataFrame()
  40.     list_temperature_vec = []
  41.     for line in list_str:
  42.         if "Interface" in line:
  43.             new_columns = re.split(r'\s+', line.strip())
  44.             new_columns = new_columns[:7]
  45.         elif re.match(r'\w+gei.+\s+.+\s+.+\s+.+\s+.+\s+.+\s+.+', line.strip()):
  46.             list_find_str = re.split(r'\s+', line.strip())
  47.             list_temperature_vec.append(list_find_str)
  49.     pd_result = pd.DataFrame(list_temperature_vec, columns=new_columns)
  50.     pd_result = pd_result[pd_result['Phy'] == 'up']
  52.     pd_output = pd.DataFrame()
  53.     for check_port in pd_result['Interface']:
  54.         dict_output = {
  55.             "port_name": check_port,
  56.             "time": run_time
  57.         }
  59.         command = f"show interface {check_port}"
  60.         command = bytes(command, encoding='utf-8')
  61.         tn.write(command + b'\n')
  62.         time.sleep(1)
  64.         result_list = []
  65.         while True:
  66.             command_result = tn.read_very_eager().decode('ascii')
  67.             result_list.append(command_result)
  68.             if re.findall(r"--More--", command_result.strip()):
  69.                 tn.write(b" ")
  70.             elif re.findall(r"#", command_result.strip()):
  71.                 break
  72.             else:
  73.                 time.sleep(0.05)
  75.         result_str = "\n".join(result_list)
  76.         rx_bytes = int(re.search(r'In_Bytes\s+(\d+)', result_str).group(1))
  77.         tx_bytes = int(re.search(r'E_Bytes\s+(\d+)', result_str).group(1))
  78.         dict_output["rx_bytes"] = rx_bytes
  79.         dict_output["tx_bytes"] = tx_bytes
  80.         pd_output = pd.concat([pd_output, pd.DataFrame.from_dict([dict_output])], ignore_index=True)
  82.     pd_output['time'] = pd.to_datetime(pd_output['time']).dt.strftime('%Y-%m-%d %H:%M:%S')
  83.     pd_output["id"] = pd_output["port_name"] + '_' + pd_output["time"]
  85.     tn.close()
  86.     return pd_output
  88. except Exception as e:
  89.     print(f"Error: {e}")
  90.     return None
复制代码
def database_write(pd_data):
engine = create_engine(DATABASE_URI)
metadata = MetaData()
table = Table(TABLE_NAME, metadata, autoload_with=engine)
Session = sessionmaker(bind=engine, autocommit=False)
session = Session()
  1. try:
  2.     pd_same_period = pd_data.fillna(value=0)
  3.     dict_rows = pd_same_period.to_dict('records')
  5.     stmt = pg_insert(table)
  6.     primary_keys = [key.name for key in inspect(table).primary_key]
  7.     stmt = stmt.on_conflict_do_update(
  8.         index_elements=primary_keys,
  9.         set_={key: value for key, value in stmt.excluded.items()}
  10.     )
  11.     session.execute(stmt, dict_rows)
  12.     session.commit()
  13. except Exception as e:
  14.     session.rollback()
  15.     print(f"Error saving data to database: {e}")
  16. finally:
  17.     session.close()
复制代码
def monitor_task():
pd_output = get_info_telnet(TELNET_HOST, TELNET_USERNAME, TELNET_PASSWORD)
if pd_output is not None:
database_write(pd_output)
if name == 'main':
schedule.every(10).seconds.do(monitor_task)
while True:
schedule.run_pending()
time.sleep(1)
6.信息采集——多远
import os
import time
import pandas as pd
import re
import glob
import numpy as np
import telnetlib
import threading
def do_telnet(host_ip, path, username, password):
  1. # path = r'D:\Users\10079494.A24808550\Desktop\信通院python培训\\'
  2. # host_ip = r'129.60.161.169'
  3. # username = 'zte'
  4. # password = 'zte'
  6. tn = telnetlib.Telnet()
  7. try:
  9.     tn.open(host_ip, port=23, timeout=5)
  10.     print('%s connected ssuccess !' % host_ip)
  12.     tn.read_until(b'Username:', timeout=5)
  13.     tn.write(username.encode('ascii') + b'\n')
  15.     tn.read_until(b'Password:', timeout=5)
  16.     tn.write(password.encode('ascii') + b'\n')
  17.     time.sleep(1)
  19.     command_result = tn.read_until(b'#', timeout=5)
  20.     if b'#' not in command_result:
  21.         print('%s登录失败' % host_ip)
  22.     else:
  23.         print('%s登录成功' % host_ip)
  25. except:
  27.     print('%s网络连接失败' % host_ip)
  29. command = "show clock"
  30. command = bytes(command, encoding='utf-8')
  31. tn.write(command + b'\r\n')
  32. run_time = tn.read_until(b'#')
  33. run_time = re.findall(r"\d+:\d+:\d+\s+\w+\s+\w+\s+\w+\s+\d+\s+2024", run_time.decode('GB18030'))[0]
  35. command = "show ip ospf neighbor detail"
  36. command = bytes(command, encoding='utf-8')
  37. tn.write(command + b'\n')
  38. time.sleep(1)
  40. result_list = []
  41. while (True):
  42.     command_result = tn.read_very_eager().decode('ascii')
  43.     # print(command_result)
  44.     result_list.append(command_result)
  45.     if re.findall(r"--More--", command_result.strip()):
  46.         tn.write(b" ")
  48.     elif re.findall(r"#", command_result.strip()):
  49.         break
  50.     else:
  51.         time.sleep(0.05)
  52.         continue
  54. result_str = "\n".join(result_list)
  56. dict_ouput = {}
  58. dict_ouput["host_ip"] = host_ip
  59. dict_ouput["time"] = run_time
  61. startpattern = re.compile(r'OSPF Router with ID (.+)')
  62. strtext = re.search(startpattern, str(result_str)).group(1)
  63. dict_ouput["OSPF Router with ID"] = strtext
  65. startpattern = re.compile(r'Neighbor\s+(\d+.\d+.\d+.\d+)')
  66. strtext = re.search(startpattern, str(result_str)).group(1)
  67. dict_ouput["Neighbor"] = strtext
  69. startpattern = re.compile(r'In the area\s+(.+)')
  70. strtext = re.search(startpattern, str(result_str)).group(1)
  71. dict_ouput["area"] = strtext
  73. startpattern = re.compile(r'State\s+(\w+), ')
  74. strtext = re.search(startpattern, str(result_str)).group(1)
  75. dict_ouput["State"] = strtext
  77. pd_output = pd.DataFrame.from_dict([dict_ouput])
  79. pd_output['time'] = pd.to_datetime(pd_output['time'])
  80. pd_output['time'] = pd_output['time'].apply(lambda x: x.strftime('%Y-%m-%d %H:%M:%S'))
  82. pd_output.to_csv(path+os.sep+r'ospf'+'-'+str(host_ip)+'.csv', index=None, encoding='gb18030')
  84. tn.close()
复制代码
def job_get_info(path):
  1. list_ip = ['129.60.161.169', '129.60.161.170']
  2. username = 'zte'
  3. password = 'zte'
  5. print("Begin......%s" % time.ctime())
  6. st = time.time()
  8. for i in range(len(list_ip)):
  9.     t = threading.Thread(target=do_telnet, args=(list_ip[i], path, username, password,))
  10.     t.start()
  12. print('time cost：', time.time()-st)
复制代码
if name == 'main':
path = r'D:\Users\10079494.A24808550\Desktop\信通院python培训\'
job_get_info(path)
  1. file_list = glob.glob(os.path.join(os.path.abspath(path), r'ospf*.csv'))
  2. pd_data_o = pd.DataFrame()
  3. for file in file_list:
  4.     try:
  5.         pd_sheet = pd.read_csv(file, encoding='gb18030', doublequote=False,
  6.                                converters={u'code': str}, engine="python")
  7.     except:
  8.         print('读取异常')
  9.     pd_data_o = pd.concat([pd_data_o, pd_sheet], axis=0)
复制代码
`

来源:https://www.cnblogs.com/tjrk/p/18453786/show_ip_interface_brief
免责声明：由于采集信息均来自互联网，如果侵犯了您的权益，请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容，谢谢合作！

上一篇: 一个好用的免费IP离线地理信息数据库

下一篇: 以pytorch的forward hook为例探究hook机制

发表于 2024-10-9 14:43:24

举报 回复 使用道具

返回列表 发新帖

Archiver|手机版|小黑屋|翼度科技

Copyright © 2001-2020, Tencent Cloud.　Powered by Discuz! X3.4