前言

本文主要給大家介紹了關(guān)于利用python 3.5檢測路由器流量并寫入txt的相關(guān)內(nèi)容，分享出來供大家參考學(xué)習，下面話不多說了，來一起看看詳細的介紹。

環(huán)境交代：win10+python3.6

代碼非常簡單，

• 模擬登陸，沒有
• 網(wǎng)頁標簽過濾，沒有
• 多線程，也沒有
• 文本處理，只有涉及到字符串截取
• 本地文本寫入，有

這么低級的代碼是因為這個路由器頁面非常垃圾，用不到~~~，不過這樣也適合初學(xué)者觀看，當然了，后續(xù)會嘗試添加更多功能

首先我們對自己的需求要進行分析，新手嘛，先把復(fù)雜的東西簡單化，模塊化，整理出思路，再一步步的去實現(xiàn)，最后整合。

1、獲得數(shù)據(jù)

• 網(wǎng)頁編碼，編碼沒有處理好會報錯，涉及到一些函數(shù)
• 編碼轉(zhuǎn)換，read()方法獲取到的非字符串類型，要預(yù)先進行處理

2、處理數(shù)據(jù)

• 方法有很多，正則，字符串截取，等等不一一介紹，適合的才是最好的，我覺得正則是很強大的，但是也是相當反人類的

3、保存數(shù)據(jù)

• 注意win下路徑和linux下路徑寫法不同
• 寫入的編碼類型需要進行處理

重點講一講我遇到的坑

一般來講右鍵查看頁面編碼，如圖所示，因為在國外是非常奇怪的編碼

當時我就蒙蔽了，這是什么鬼~

這個時候我們需要用到chardet庫來判斷編碼類型，拿百度舉例，自行百度python第三方庫如何安裝，這里不做闡述

import chardet
import urllib.request
html = urllib.request.urlopen('http://www.baidu.com/').read()
print (chardet.detect(html))

得到的結(jié)果如下：

C:\python\python.exe D:/python/test/2.py
{'encoding': 'utf-8', 'confidence': 0.99, 'language': ''}

Process finished with exit code 0

1.取得字符串

import urllib.request
import os
from bs4 import BeautifulSoup
import time
def getHtml(url):
 html = urllib.request.urlopen(url).read()
 return html
#獲取字符串，因為我在國外，路由器low到爆，編碼格式也蛋疼了我很久
html = getHtml("http://192.168.0.254/pub/fbx_info.txt")
#將read()獲取的是bytes編碼轉(zhuǎn)化成str
html = html.decode("ISO-8859-1")

因為是歐洲網(wǎng)站，獲取的卻是一個 【ISO-8859-1】的網(wǎng)頁編碼。

由于這里我們用read()方法讀取，獲取到的是bytes類型，此時需要轉(zhuǎn)換成str類型的，才可以進行下一步的處理，如果不處理就會有下列錯誤

TypeError: a bytes-like object is required, not 'str'

這里可以使用decode()方法處理一下

html = html.decode("ISO-8859-1")

使用type()方法檢測下編碼

print(type(html))

反饋

<class 'str'>

2.處理字符串

#操作字符串
html = html.split('WAN')[1].split('Ethernet')[0]
time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
#因為使用記事本，在記事本中換行只認\r\n
str = time+html+"\r\n"

這里我使用的是split()方法，使用方法呢，推薦大家學(xué)會查手冊，這里也不詳細再解釋了。

我通過split()方法截取到了自己需要的字符串，然后用time()函數(shù)獲取當前本地時間，然后用+連接符把當前本地時間和處理后的字符串連接起來，再在末尾接上換行符rn，因為懶到只用記事本，換行符這里只算一個小坑

3.寫入txt

# 保存文件為txt，win環(huán)境路徑
outfile = open("C:\\Users\\sw\\Desktop\\route.txt","a+",encoding="utf-8",newline='\n')
outfile.write(str)
outfile.close()
print("文件已保存到本地")

關(guān)于open函數(shù)，每一個參數(shù)，此篇博文講的非常清楚非常詳細，感謝博主的整理，大家可以看看，建議收藏

http://www.dbjr.com.cn/article/58002.htm

要強調(diào)的有2個參數(shù)

encoding和newline，因為用的比較少，然后很多文章并不會提到

有人可能會說，哇，真的懶

沒錯，我就是懶，而且再說了，寫了還沒人家寫的好，那我何必誤人子弟。

下面貼上全部代碼

import urllib.request
import os
from bs4 import BeautifulSoup
import time
def getHtml(url):
 html = urllib.request.urlopen(url).read()
 return html

#獲取字符串，因為我在國外，路由器low到爆，編碼格式也蛋疼了我很久
html = getHtml("http://192.168.0.254/pub/fbx_info.txt")
#將read()獲取的是bytes編碼轉(zhuǎn)化成str
html = html.decode("ISO-8859-1")
#再次檢測編碼
#print(type(html))
#操作字符串
html = html.split('WAN')[1].split('Ethernet')[0]
time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
#因為使用記事本，在記事本中換行只認\r\n
str = time+html+"\r\n"
# 保存文件為txt，win環(huán)境路徑
#print(type(str))
outfile = open("C:\\Users\\sw\\Desktop\\route.txt","a+",encoding="utf-8",newline='\n')
outfile.write(str)
outfile.close()
print("文件已保存到本地")

本來當初的想法是每秒獲取一次流量，寫入txt，再讀txt，通過數(shù)據(jù)來繪圖

但是有事暫時擱下了，只是win計劃任務(wù)定時運行，勉強湊活用下日后再學(xué)習補充

總結(jié)

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學(xué)習或者工作具有一定的參考學(xué)習價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。

最新評論