import requests
from requests.exceptions import RequestException
import re
import json
# from multiprocessing import Pool

# 測試了下 這裡需要自己新增頭部 否則得不到網頁
headers = {
    'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.102 Safari/537.36'
}
# 得到html程式碼
def
 
 get_one_page(url):
    try:
        response = requests.get(url, headers = headers)
        if response.status_code == 200:
            return response.text
        return None
    except RequestException:
        return None

# 解析html程式碼
def parse_one_page(html):
    pattern = re.compile('<dd>.*?board-index.*?>(\d+)</i>.*?data-src="(.*?)".*?name"><a.*?">(.*?)</a>.*?star">(.*?)</p>.*?releasetime">(.*?)</p>.*?integer">(.*?)</i>.*?fraction">(.*?)</i>.*?(/dd)
 
', re.S)
    items = re.findall(pattern, html)
    for item in items:
        # 將元組形式變為字典
        yield {
            '【排名】': item[0],
            '【圖片】': item[1],
            '【標題】': item[2],
            '【主演】': item[3].strip()[3:],
            '【上映時間】': item[4].strip()[5:],
            '【評分】': item[5] + item[6]
        }

 
# 寫入檔案
def write_to_file(content):
    # 注意encoding = 'utf-8'和ensure_ascii = False，不寫的話不能輸出漢字
    with open('電影.txt', 'a', encoding = 'utf-8') as f:
        f.write(json.dumps(content, ensure_ascii = False) + '\n')
        f.close()

# 主函式
def main(offset):
    url = 'http://maoyan.com/board/4?offset=' + str(offset)
    html = get_one_page(url)
    for item in parse_one_page(html):
        print(item)
        write_to_file(item)

if __name__ == '__main__':
    for i in range(10):
        main(i * 10)

# 多程序（測試有bug）
# if __name__ == '__main__':
#     pool = Pool()
#     pool.map(main, [i * 10 for i in range(10)])
#     pool.join()
#     pool.close()

程式碼來自崔慶才