python 爬取百度百科中《青春有你2》中参赛选手信息

 飞桨深度学习学院的《Python小白逆袭大神》课程

pip install BeautifulSoup4
pip install lxml

import json
import re
import requests
import datetime
from bs4 import BeautifulSoup
import os
import urllib.parse 

#获取当天的日期,并进行格式化,用于后面文件命名，格式:20200420
today = datetime.date.today().strftime('%Y%m%d')    
work_path = 'D:python/work/'

def crawl_wiki_data():
    """
    爬取百度百科中《青春有你2》中参赛选手信息，返回html
    """
    headers = { 
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36'
    }
    url='https://baike.baidu.com/item/青春有你第二季'      
    try:
        response = requests.get(url,headers=headers)

        
        #将一段文档传入BeautifulSoup的构造方法,就能得到一个文档的对象, 可以传入一段字符串
        soup = BeautifulSoup(response.text,'lxml')
        
        #返回的是class为table-view log-set-param的<table>所有标签
        # tables = soup.find_all('table',{'class':'table-view log-set-param'})
        tables = soup.find_all('table',{'log-set-param':'table_view'})
        
        crawl_table_title = "参赛学员"

        for table in  tables:           
            #对当前节点前面的标签和字符串进行查找
            table_titles = table.find_previous('div').find_all('h3')
            for title in table_titles:
                if(crawl_table_title in title):
                    return table       
    except Exception as e:
        print(e)


def parse_wiki_data(table_html):
    '''
    从百度百科返回的html中解析得到选手信息，以当前日期作为文件名，存JSON文件,保存到work目录下
    '''
    bs = BeautifulSoup(str(table_html),'lxml')

    all_trs = bs.find_all('tr')
    error_list = ['\'','\"']
    stars = []

    for tr in all_trs[1:]:
        all_tds = tr.find_all('td')
        
        star = {}

        #姓名
        star["name"]=all_tds[0].text
        #个人百度百科链接
        # star["link"]= 'https://baike.baidu.com' + all_tds[0].find('a').get('href')
        star["link"]= 'https://baike.baidu.com' + urllib.parse.unquote(all_tds[0].find('a').get('href'))
        #籍贯
        star["zone"]=all_tds[1].text
        #星座
        star["constellation"]=all_tds[2].text
        #身高
        # star["height"]=all_tds[3].text
        # #体重
        # star["weight"]= all_tds[4].text

        #花语,去除掉花语中的单引号或双引号
        flower_word = all_tds[3].text
        for c in flower_word:
            if  c in error_list:
                flower_word=flower_word.replace(c,'')
        star["flower_word"]=flower_word 

        #公司
        if not all_tds[4].find('a') is  None:
            star["company"]= all_tds[4].find('a').text
        else:
            star["company"]= all_tds[4].text  

        stars.append(star)

    json_data = json.loads(str(stars).replace("\'","\""))   

    if not os.path.exists(work_path):
        os.makedirs(work_path)

    with open(work_path + today + '.json', 'w', encoding='UTF-8') as f:
        json.dump(json_data, f, ensure_ascii=False)


def crawl_pic_urls():
    '''
    爬取每个选手的百度百科图片，并保存
    ''' 
    with open(work_path + today + '.json', 'r', encoding='UTF-8') as file:
        json_array = json.loads(file.read())

        headers = { 
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36' 
        }

        for star in json_array:

            name = star['name']
            link = star['link']

            response = requests.get(link,headers=headers)

            bs = BeautifulSoup(response.text,'lxml')

            # pic_list_url = bs.select('.summary-pic a')[0].get('href')

            a_list = bs.select('.lemma-album')
            if len(a_list) == 0:
                print(name + '下载失败跳过')
                continue

            pic_list_url = bs.select('.lemma-album')[0].get('href')
            
            # print(pic_list_url)
            pic_list_url = 'https://baike.baidu.com' + pic_list_url

            pic_list_response = requests.get(pic_list_url,headers=headers)
            bs = BeautifulSoup(pic_list_response.text,'lxml')
            pic_list_html = bs.select('.pic-list img')

            pic_urls = []
            for pic_html in pic_list_html:
                pic_url = pic_html.get('src')
                pic_urls.append(pic_url)
                #根据图片链接列表pic_urls, 下载所有图片，保存在以name命名的文件夹中
            down_pic(name,pic_urls)

def down_pic(name,pic_urls):
    '''
    根据图片链接列表pic_urls, 下载所有图片，保存在以name命名的文件夹中,
    '''
    path = 'D:python/work/pics/'+name+'/'

    if not os.path.exists(path):
        os.makedirs(path)

    for i, pic_url in enumerate(pic_urls):
        try:
            pic = requests.get(pic_url, timeout=15)
            string = str(i + 1) + '.jpg'
            with open(path+string, 'wb') as f:
                f.write(pic.content)
                print(name + '成功下载第%s张图片: %s' % (str(i + 1), str(pic_url)))
        except Exception as e:
            print(name + '下载第%s张图片时失败: %s' % (str(i + 1), str(pic_url)))
            print(e)
            continue


def show_pic_path(path):
    '''
    遍历所爬取的每张图片，并打印所有图片的绝对路径
    '''
    pic_num = 0
    for (dirpath,dirnames,filenames) in os.walk(path):
        for filename in filenames:
           pic_num += 1
           print("第%d张照片：%s" % (pic_num,os.path.join(dirpath,filename)))           
    print("共爬取《青春有你2》选手的%d照片" % pic_num)

if __name__ == '__main__':

    #爬取百度百科中《青春有你2》中参赛选手信息，返回html
    html = crawl_wiki_data()

    #  #解析html,得到选手信息，保存为json文件
    parse_wiki_data(html)

    #  #从每个选手的百度百科页面上爬取图片,并保存
    crawl_pic_urls()

    #  #打印所爬取的选手图片路径
    show_pic_path('D:python/work/pics/')

    print("所有信息爬取完成！")

参考教程 https://blog.csdn.net/m0_45085566/article/details/105743689