import os
import urllib.request
import random
import time
class BeikeSpider:
    def __init__(self, save_path="./beike"):
        """
        貝殼爬蟲構(gòu)造函數(shù)
        :param save_path: 網(wǎng)頁(yè)保存目錄
        """

# 網(wǎng)址模式
        self.url_mode = "http://{}.fang.ke.com/loupan/pg{}/"
        # 需爬取的城市
        self.cities = ["cd", "sh", "bj"]
        # 每個(gè)城市爬取的頁(yè)數(shù)
        self.total_pages = 20
        # 讓爬蟲程序隨機(jī)休眠5-10秒
        self.sleep = (5, 10)
        # 網(wǎng)頁(yè)下載保存根目錄
        self.save_path = save_path
        # 設(shè)置用戶代理，是爬蟲程序偽裝成瀏覽器
        self.headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/89.0.4389.114 Safari/537.36"}
        # 代理IP的信息
        self.proxies = [
            {"https": "123.163.67.50:8118"},
            {"https": "58.56.149.198:53281"},
            {"https": "14.115.186.161:8118"}
        ]

        # 創(chuàng)建保存目錄
        if not os.path.exists(self.save_path):
            os.makedirs(self.save_path)
   def crawl(self):
        """
        執(zhí)行爬取任務(wù)
        :return: None
        """

 for city in self.cities:
            print("正在爬取的城市:", city)
            # 每個(gè)城市的網(wǎng)頁(yè)用單獨(dú)的目錄存放
            path = os.path.join(self.save_path, city)
            if not os.path.exists(path):
                os.makedirs(path)

            for page in range(1, self.total_pages+1):
                # 構(gòu)建完整的url
                url = self.url_mode.format(city, page)
                # 構(gòu)建Request對(duì)象, 將url和請(qǐng)求頭放入對(duì)象中
                request = urllib.request.Request(url, headers=self.headers)

                # 隨機(jī)選擇一個(gè)代理IP
                proxy = random.choice(self.proxies)
                # 構(gòu)建代理服務(wù)器處理器
                proxy_handler = urllib.request.ProxyHandler(proxy)
                # 構(gòu)建opener
                opener = urllib.request.build_opener(proxy_handler)
                # 使用構(gòu)建的opener打開網(wǎng)頁(yè)
                response = opener.open(request)
                html = response.read().decode("utf-8")
                # 網(wǎng)頁(yè)保存文件名(包含路徑)
                filename = os.path.join(path, str(page)+".html")

                # 保存網(wǎng)頁(yè)
                self.save(html, filename)
                print("第%d頁(yè)保存成功！" % page)

                # 隨機(jī)休眠
                sleep_time = random.randint(self.sleep[0], self.sleep[1])
                time.sleep(sleep_time)

def save(self, html, filename):
        """
        保存下載的網(wǎng)頁(yè)
        :param html: 網(wǎng)頁(yè)內(nèi)容
        :param filename: 保存的文件名
        :return:
        """

        f = open(filename, 'w', encoding="utf-8")
        f.write(html)
        f.close()

    def parse(self):
        """
        解析網(wǎng)頁(yè)數(shù)據(jù)
        :return:
        """
        pass

if __name__ == "__main__":
    spider = BeikeSpider()
    spider.crawl()