day18 - 爬取租房信息作业

# 作业:爬贝壳租房信息
import requests
from re import findall, sub
# 定义函数
def rental_information(x):
    # 请求网页
    response = requests.get(x)
    # 按顺序获取小区等名字信息
    s1 = []
    for x in findall(r'[整合]租·.+\s.+\s[东南西/北卧]{1,3}', response.text):
        if x not in s1:
            s1.append(x)
    # 按顺序获取面积
    s2 = findall(r'\d+\.\d{2}㎡', response.text)
    # 按顺序获取价格
    s3 = findall(r'<em>\d+</em>', response.text)
    # 剔除价格中无用字符串
    for x in range(len(s3)):
        s3[x] = sub(r'\D','',s3[x])
    # 判断获取是否有效正确
    if len(s2) == len(s3) == len(s1):
        # 列表(字典)
        s5 = []
        for x in range(len(s1)):
            # 写入文件
            f = open('05.txt', 'at',encoding='utf-8')
            f.write(f'{s1[x]},{s2[x]},{s3[x]}\n')
            f.close()
            # 将字典添加至列表
            s5.append({'name': s1[x],'area': s2[x], 'price': s3[x],})
        # 有效正确返回列表
        return s5
    else:
        # 无效正确返回False
        return False

# 调用函数(不同页面地址不同)
print(rental_information('https://cd.zu.ke.com/zufang/pg20/#contentList'))