Python数据爬虫,爬链家的二手房信息

# -*- coding:utf-8 -*-
from bs4 import BeautifulSoup

import re
import urllib
import urllib2
import math
import codecs

domain='http://bj.lianjia.com/'
url = domain + '/ershoufang/'

def get_all_xiaoqu_info():
    html_doc = urllib.urlopen(url).read()

    soup = BeautifulSoup(html_doc, 'html.parser')

    def has_href_but_title(tag):
        return tag.has_attr('href') and not tag.has_attr('title')
    #得到所有区域
    all_xiao_list=[]
    for item in soup.find(attrs={"data-role": "ershoufang"}).find_all('a'):
        #print "%s|%s" % (item.get('href') , item.string)
        quyu_url=item.get('href')
        quyu_html_doc = urllib.urlopen(domain + quyu_url).read()
        quyu_soup = BeautifulSoup(quyu_html_doc, 'html.parser')

        for item02 in quyu_soup.find(attrs={"data-role": "ershoufang"}).find_all(has_href_but_title):
            xiaoqu_info=list()
            xiqu_url=domain + item02.get('href')
            #print '%s|%s|%s' % (item.get('href'), item.string,  item02.string)
            xiaoqu_info.append(domain + item02.get('href'))
            xiaoqu_info.append(item.string)
            xiaoqu_info.append(item02.string)
            all_xiao_list.append(xiaoqu_info)
    # for item in all_xiao_list:
    #     print '|'.join(item)
    return all_xiao_list


headers = {
        'Accept':r'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
        'Accept-Encoding':r'gzip, deflate',
        'Accept-Language':r'zh-CN,zh;q=0.8,en-US;q=0.5,en;q=0.3',
        'Connection':r'keep-alive',
        'Host':r'bj.lianjia.com',
        'Referer':r'http://bj.lianjia.com/ershoufang/beiqijia',
        'User-Agent':r'Mozilla/5.0 (Windows NT 10.0; WOW64; rv:47.0) Gecko/20100101 Firefox/47.0',
        #'Cookie':r'lianjia_uuid=18acab8e-de62-43a9-b3d4-962b28968ec7; _smt_uid=57219c21.42bc10ea; CNZZDATA1253477573=294732555-1461819878-http%253A%252F%252Fwww.baidu.com%252F%7C1467557395; _jzqa=1.598087999714383000.1461820449.1467148977.1467525301.10; _jzqy=1.1461820449.1465381613.3.jzqsr=baidu.jzqsr=baidu|jzqct=%E9%93%BE%E5%AE%B6; CNZZDATA1254525948=1894244370-1461815948-http%253A%252F%252Fwww.baidu.com%252F%7C1467557174; CNZZDATA1255633284=1405522112-1461816002-http%253A%252F%252Fwww.baidu.com%252F%7C1467556338; CNZZDATA1255604082=136337279-1461816253-http%253A%252F%252Fwww.baidu.com%252F%7C1467557177; _ga=GA1.2.433541691.1461820452; miyue_hide=%20index%20%20index%20%20index%20%20index%20%20index%20; _jzqx=1.1461823997.1462357481.2.jzqsr=bj%2Elianjia%2Ecom|jzqct=/ershoufang/rs%e5%8d%97%e5%b2%b8.jzqsr=bj%2Elianjia%2Ecom|jzqct=/ershoufang/yanjiao/; select_city=110000; all-lj=b3d70adee01740a6a0beb306d0f6f412; _jzqb=1.31.10.1467525301.1; _jzqc=1; _jzqckmp=1; lianjia_ssid=dedfe462-961f-4b4e-a935-1e170a4cf252; _qzja=1.1927253355.1461820449108.1467148976556.1467525301331.1467528966161.1467528974420.0.0.0.136.10; _qzjb=1.1467525301331.31.0.0.0; _qzjc=1; _qzjto=31.1.0; _gat=1; _gat_past=1; _gat_global=1; _gat_new_global=1; _gat_dianpu_agent=1',
        'DN':'1',
        'Cache-Control': 'max-age=0'
}

def getHouseInfoForOneURL(quyuInfo):
    house_list=[]
    #xiaoqu_url=r'http://bj.lianjia.com/ershoufang/changying/p1/'
    xiaoqu_url=quyuInfo[0]
    xiao_html_doc= urllib.urlopen(xiaoqu_url).read()
    xiaoqu_soup = BeautifulSoup(xiao_html_doc, 'html.parser')
    total_house=int(xiaoqu_soup.find("h2",class_="total fl").find('span').string)

    if total_house == 0:
        print 'no hourse fouund'
        return house_list
        
    if total_house > 3000:
        print '3000+ hourse fouund, need split it into small pages'
        return house_list
        
    total_page = math.ceil(total_house / 30.0)
    for page_index in range(1, int(total_page)+1):
        page_url= '%s%d' % ('http://bj.lianjia.com/ershoufang/changying/p',page_index)
        page_html_doc= urllib.urlopen(page_url).read()
        page_soup = BeautifulSoup(page_html_doc, 'html.parser')
        #print page_url
        house_info=dict()
        for house_info in  page_soup.find('ul',class_='listContent').find_all('div', class_="info"):
            house_info['href']= house_info.find('a').get('href')
            house_info['title']= house_info.find('a').string
            house_info['address_xiaoqu']=house_info.find('div',class_="houseInfo").a.string
            house_info['address_info']= house_info.find('div',class_="houseInfo").a.next_sibling
            house_info['flood']= house_info.find('div',class_="flood").find('div',class_="positionInfo").span.next_sibling
            house_info['area']= house_info.find('div',class_="flood").find('div',class_="positionInfo").a.string
            house_info['totalPrice']= house_info.find('div',class_="totalPrice").span.string + house_info.find('div',class_="totalPrice").span.next_sibling
            house_info['unitPice']= house_info.find('div',class_="unitPrice").span.string
            house_info['area']= quyuInfo[1]
            house_info['area02']= quyuInfo[2]
            house_list.append(house_info)

    return house_list

file = codecs.open('lianjia.txt','wb','utf-8')

quyu_list = get_all_xiaoqu_info()
for quyu in quyu_list:
    print '%s %s :(%s)' % (quyu[0], quyu[1], quyu[2])
    house_list = getHouseInfoForOneURL(quyu)
    for house in house_list:
        house_str= '%s|%s|%s|%s|%s|%s|%s|%s|%s
' % (
            house['area'], house['area02'],house['title'], house['address_xiaoqu'], house['address_info'],house['flood'], house['area'],house['totalPrice'],house['unitPice']
        )
        file.write(house_str)
file.close()