from fake_useragent import UserAgent  # 身份信息生成庫(kù)

from bs4 import BeautifulSoup  # 網(wǎng)頁(yè)元素解析庫(kù)
import numpy as np  # 科學(xué)計(jì)算庫(kù)
import requests  # 網(wǎng)頁(yè)下載庫(kù)
from requests.exceptions import RequestException  # 網(wǎng)絡(luò)請(qǐng)求異常庫(kù)
import pandas as pd  # 數(shù)據(jù)處理庫(kù)

user_agent = UserAgent()

def get_html_txt(url, page_index):
    '''
    獲取網(wǎng)頁(yè)html文本信息
    :param url: 爬取地址
    :param page_index:當(dāng)前頁(yè)數(shù)
    :return:
    '''
    try:
        headers = {
            'user-agent': user_agent.random
        }
        response = requests.request("GET", url, headers=headers, timeout=10)
        html_txt = response.text
        return html_txt
    except RequestException as e:
        print('獲取第{0}頁(yè)網(wǎng)頁(yè)元素失敗！'.format(page_index))
        return ''

def catch_html_data(url, page_index):
    '''
    處理網(wǎng)頁(yè)元素?cái)?shù)據(jù)
    :param url: 爬蟲(chóng)地址
    :param page_index:
    :return:
    '''

    # 下載網(wǎng)頁(yè)元素
    html_txt = str(get_html_txt(url, page_index))

    if html_txt.strip() != '':

        # 初始化網(wǎng)頁(yè)元素對(duì)象
        beautifulSoup = BeautifulSoup(html_txt, 'lxml')

        # 解析房源列表
        h_list = beautifulSoup.select('.resblock-list-wrapper li')

        # 遍歷當(dāng)前房源的詳細(xì)信息
        for n in range(len(h_list)):
            h_detail = h_list[n]

            # 提取房源名稱
            h_detail_name = h_detail.select('.resblock-name a.name')
            h_detail_name = [m.get_text() for m in h_detail_name]
            h_detail_name = ' '.join(map(str, h_detail_name))

            # 提取房源類型
            h_detail_type = h_detail.select('.resblock-name span.resblock-type')
            h_detail_type = [m.get_text() for m in h_detail_type]
            h_detail_type = ' '.join(map(str, h_detail_type))

            # 提取房源銷售狀態(tài)
            h_detail_status = h_detail.select('.resblock-name span.sale-status')
            h_detail_status = [m.get_text() for m in h_detail_status]
            h_detail_status = ' '.join(map(str, h_detail_status))

            # 提取房源單價(jià)信息
            h_detail_price = h_detail.select('.resblock-price .main-price .number')
            h_detail_price = [m.get_text() for m in h_detail_price]
            h_detail_price = ' '.join(map(str, h_detail_price))

            # 提取房源總價(jià)信息
            h_detail_total_price = h_detail.select('.resblock-price .second')
            h_detail_total_price = [m.get_text() for m in h_detail_total_price]
            h_detail_total_price = ' '.join(map(str, h_detail_total_price))

            h_info = [h_detail_name, h_detail_type, h_detail_status, h_detail_price, h_detail_total_price]
            h_info = np.array(h_info)
            h_info = h_info.reshape(-1, 5)
            h_info = pd.DataFrame(h_info, columns=['房源名稱', '房源類型', '房源狀態(tài)', '房源均價(jià)', '房源總價(jià)'])
            h_info.to_csv('北京房源信息.csv', mode='a+', index=False, header=False)

        print('第{0}頁(yè)房源信息數(shù)據(jù)存儲(chǔ)成功！'.format(page_index))
    else:
        print('網(wǎng)頁(yè)元素解析失敗！')

import threading  # 導(dǎo)入線程處理模塊


def thread_catch():
    '''
    線程處理函數(shù)
    :return:
    '''
    for num in range(1, 50, 3):
        url_pre = "https://bj.fang.lianjia.com/loupan/pg{0}/".format(str(num))
        url_cur = "https://bj.fang.lianjia.com/loupan/pg{0}/".format(str(num + 1))
        url_aft = "https://bj.fang.lianjia.com/loupan/pg{0}/".format(str(num + 2))

        thread_pre = threading.Thread(target=catch_html_data, args=(url_pre, num))
        thread_cur = threading.Thread(target=catch_html_data, args=(url_cur, num + 1))
        thread_aft = threading.Thread(target=catch_html_data, args=(url_aft, num + 2))
        thread_pre.start()
        thread_cur.start()
        thread_aft.start()


thread_catch()