1. 程式人生 > >簡單Python3爬蟲程式(2)進階:偽裝瀏覽器、超時功能、儲存資料

簡單Python3爬蟲程式(2)進階:偽裝瀏覽器、超時功能、儲存資料

import urllib.request
import http.cookiejar

# head: dict of header
def makeMyOpener(head = {
    'Connection': 'Keep-Alive',
    'Accept': 'text/html, application/xhtml+xml, */*',
    'Accept-Language': 'en-US,en;q=0.8,zh-Hans-CN;q=0.5,zh-Hans;q=0.3',
    'User-Agent': 'Mozilla/5.0 (Windows NT 6.3; WOW64; Trident/7.0; rv:11.0) like Gecko'
}):
    cj = http.cookiejar.CookieJar()
    opener = urllib.request.build_opener(urllib.request.HTTPCookieProcessor(cj))
    header = []
    for key, value in head.items():
        elem = (key, value)
        header.append(elem)
    opener.addheaders = header
    return opener

def saveFile(data):
    save_path = 'D:\\output.out'
    f_obj = open(save_path,'wb')
    f_obj.write(data)
    f_obj.close()

oper = makeMyOpener()
uop = oper.open('http://www.baidu.com/', timeout = 1000)
data1 = uop.read()    #.decode('utf-8')
saveFile(data1)
print(data1)