python爬取網頁源代碼

編輯器:vscode

import urllib.request


def grab(url):
    # 打開傳入的網址
    resp = urllib.request.urlopen(url)
    # 讀取網頁源碼内容
    data = resp.read()
    # 輸入存儲檔案名
    name = input("請輸入檔案名")
    # 打開檔案
    file_name = open(name, "wb")
    # 将代碼寫入檔案
    file_name.write(data)
    # 關閉檔案
    file_name.close()

    print("源碼爬取完成")


if __name__ == '__main__':
    # 按照格式輸入網址
    web_addr = input("請輸入要抓取的網址:")
    try:
        grab(web_addr)
    except:
        print("網址有誤")