基于python编写的服务器之间流量传输netflow_exporter_python流量发送

def change_unit(unit):
if “Mb” in unit:
flow = float(unit.strip(“Mb”)) * 1024
return flow
elif “Kb” in unit:
flow = float(unit.strip(“Kb”))
return flow
elif “b” in unit:
flow = float(unit.strip(“b”)) / 1024
return flow

def get_flow():
#iftop参数：-t 使用不带ncurses的文本界面,-P显示主机以及端口信息,-N只显示连接端口号，不显示端口对应的服务名称,-n 将输出的主机信息都通过IP显示，不进行DNS解析,-s num num秒后打印一次文本输出然后退出
#1.服务器上运行
result = os.popen(“iftop -t -N -n -s 2 2>/dev/null |grep -A 1 -E ‘^ [0-9]’”).read()
#2.本地测试数据
#result = open(“basedatanoport.txt”).read()
#以换行符进行分割
iftop_list = result.split(“\n”)
#print(iftop_list)
count = int(len(iftop_list))
#定义字典 存放主机信息和进出流量
flow_dict = {}

for i in range(int(count/2)):
    flow_msg = ""
    #获取发送的ip地址(本地ip地址)，数据偶数位为本地发送流量信息
    location_li_s = iftop_list[i\*2]
    send_flow_lists = location_li_s.split(" ")

    #去空元素
    while '' in send_flow_lists:
        send_flow_lists.remove('')
    localhostip = send_flow_lists[1]
    send_flow = send_flow_lists[3]
    send_flow_float = change_unit(send_flow)

    #获取接收的流量
    location_li_r = iftop_list[i\*2+1]
    rec_flow_lists = location_li_r.split(" ")

    while '' in rec_flow_lists:
        rec_flow_lists.remove('')
    remote_host_ip = rec_flow_lists[0]
    rec_flow = rec_flow_lists[3]
    rec_flow_float = change_unit(rec_flow)
    local_remote_host=localhostip+str(' <==> ')+remote_host_ip
    flow_msg = str(float('%2.f' % send_flow_float)) + "|" + str(float('%.2f' % rec_flow_float))
    flow_dict[local_remote_host] = flow_msg

send_rows = []
rec_rows = []
for key in flow_dict:
    send_row_tmp_dict = {}
    rec_row_tmp_dict = {}
    flow_li = flow_dict[key].split("|")
    #flow\_li[0]为发送流量，flow\_li[1]为接收流量，单位是Kb
    #print(key + "|" + flow\_li[0] + "|" + flow\_li[1])
    send_row_tmp_dict['remoteip'] = key.replace('<','>')
    send_row_tmp_dict['value'] = flow_li[0]
    rec_row_tmp_dict['remoteip'] = key.replace('>','<')
    rec_row_tmp_dict['value'] = flow_li[1]
    send_rows.append(send_row_tmp_dict)
    rec_rows.append(rec_row_tmp_dict)
return send_rows,rec_rows

app = Flask(name)

REGISTRY = CollectorRegistry(auto_describe=False)
count = Counter(
‘count’,
‘count’,
registry=REGISTRY
)
networksSend = Gauge(
name=“send_flow”,
documentation=“Send_Flow_Kb”,
namespace=“netflow”,
labelnames=[“remoteip”],
registry=REGISTRY
)
networkReceive = Gauge(
name=“receive_flow”,
documentation=“Receive_Flow_Kb”,
namespace=“netflow”,
labelnames=[“remoteip”],
registry=REGISTRY
)

c = Gauge(‘my_requests_total’, ‘HTTP Failures’, [‘method’, ‘endpoint’],registry=REGISTRY)

@app.route(‘/metrics’)
def r_value():
#获取流量信息
send_rows,rec_rows = get_flow()

for row_s in send_rows:
    networksSend.labels(row_s['remoteip']).set(row_s['value'])

for row_r in rec_rows:
    networkReceive.labels(row_r['remoteip']).set(row_r['value'])

c.labels('test', '1').inc()
c.labels('post', '/submit').inc()
return Response(prometheus_client.generate_latest(REGISTRY),
                mimetype="text/plain")

@app.route(‘/’)
def index():
return “”
“”
“”
“

NetWorkTraffic Exporter

”
“

二、Python必备开发工具

工具都帮大家整理好了，安装就可直接上手！

三、最新Python学习笔记

当我学到一定基础，有自己的理解能力的时候，会去阅读一些前辈整理的书籍或者手写的笔记资料，这些笔记详细记载了他们对一些技术点的理解，这些理解是比较独到，可以学到不一样的思路。

四、Python视频合集

观看全面零基础学习视频，看视频学习是最快捷也是最有效果的方式，跟着视频中老师的思路，从基础到深入，还是很容易入门的。

五、实战案例

纸上得来终觉浅，要学会跟着视频一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

六、面试宝典

简历模板

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化学习资料的朋友，可以戳这里无偿获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！