爬取弹幕内容

danmu.py

@@ -0,0 +1,144 @@
+import requests
+import re
+import xml.etree.ElementTree as ET
+from DrissionPage import ChromiumPage
+from bs4 import BeautifulSoup
+from collections import OrderedDict
+import pandas as pd
+
+
+headers = {
+    # cookie用户信息：用于检测用户是否登录账号
+    "cookie": "buvid4=B1349383-F2A6-E4E5-ED2C-A5B428CC0ED955473-022061918-9v4qmUz9VFkZrXyhSPgawQ%3D%3D; buvid_fp_plain=undefined; LIVE_BUVID=AUTO2216562263398747; is-2022-channel=1; CURRENT_FNVAL=4048; DedeUserID=406267828; DedeUserID__ckMd5=a7899a47ba8a07ab; enable_web_push=DISABLE; rpdid=|(u))kkYu||u0J'u~|JkJR)ul; FEED_LIVE_VERSION=V_WATCHLATER_PIP_WINDOW3; CURRENT_QUALITY=80; buvid3=3D0F285C-7560-9E88-C05A-1AFF4D9CAD7617643infoc; b_nut=1718944717; _uuid=BAAADDA5-4DDD-184B-DB310-EC4105D97D51E17974infoc; header_theme_version=CLOSE; fingerprint=5d019e2bddbce042baa44b97851bb2fd; buvid_fp=5d019e2bddbce042baa44b97851bb2fd; PVID=3; SESSDATA=e97fd178%2C1741840358%2C6c71f%2A91CjB7QCyKSGsn3CjY0C0HRTQF_8a0AYz9r9QLq2FL2YiXLfQFehx5MdXHV3EDcePi4mwSVlM3TVE3LWcyczhLbXoycnhOTWtwa2g5OEl4RVdxOVRQZjljWGJ5d2czbFZpTkVhd3hfRnFmYTE4TVRISTlkdXFpemd1SmhOejNnVm1mYlJ0UDZEcTNBIIEC; bili_jct=4b351ffc7724b0bb10c029efdfad7f75; sid=6jnrbe6l; bili_ticket=eyJhbGciOiJIUzI1NiIsImtpZCI6InMwMyIsInR5cCI6IkpXVCJ9.eyJleHAiOjE3MjY1NDc1NzEsImlhdCI6MTcyNjI4ODMxMSwicGx0IjotMX0.l-PQPoCUFvsMgRpDwqUKwUY9jLLjI-p-HZ1Qaq7AIjI; bili_ticket_expires=1726547511; b_lsid=1564BAD10_191F00CC884; bp_t_offset_406267828=976996173829111808; home_feed_column=5; browser_resolution=2048-1030",
+    # user-agent用户代理：表示浏览器设备的基本信息
+    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36 Edg/128.0.0.0"
+}
+
+def get_video_details(search_url):
+
+    response = requests.get(search_url, headers=headers)
+    # 确保请求成功
+    response.raise_for_status()
+
+    soup = BeautifulSoup(response.text, 'html.parser')
+
+    # 匹配 BV 号
+    bv_pattern = re.compile(r'BV[1-9A-Za-z]{10}')
+    bv_l=[]
+    for tag in soup.find_all('a', href=True):
+        if 'BV' in tag['href']:
+            bv = bv_pattern.search(tag['href'])
+            bv_l.append(bv.group())
+            bv_l = list(OrderedDict.fromkeys(bv_l))# 去重
+
+    return bv_l
+
+def get_video_info(bv_id):
+    """ 获取视频信息并解析CID """
+    url = f'https://api.bilibili.com/x/web-interface/view?bvid={bv_id}'
+    response = requests.get(url, headers=headers)
+    response.encoding = 'utf-8'
+    response.raise_for_status()
+    data = response.json()
+    cid = data['data']['cid']
+    return cid
+
+
+def get_danmaku_xml(cid):
+    """ 根据CID获取弹幕XML数据 """
+    url = f'https://comment.bilibili.com/{cid}.xml'
+    response = requests.get(url, headers=headers)
+    response.encoding = 'utf-8'
+    # 确保请求成功
+    response.raise_for_status()
+    return response.text
+
+
+def parse_danmaku_xml(xml_str):
+    """ 解析弹幕XML数据 """
+    root = ET.fromstring(xml_str)
+    danmaku_list = [d.text for d in root.findall('.//d') if d.text]
+    return danmaku_list
+
+
+def save_danmaku_to_excel(danmaku_content, file_name='danmaku_content.xlsx'):
+    """
+    将弹幕内容保存到 Excel 文件中
+    :param danmaku_content: 包含弹幕内容的列表
+    :param file_name: Excel 文件名
+    """
+    # 检查是否已经存在文件
+    try:
+        # 尝试读取现有文件
+        existing_df = pd.read_excel(file_name,engine='openpyxl')
+    except FileNotFoundError:
+        # 文件不存在，则创建新的 DataFrame
+        existing_df = pd.DataFrame(columns=['content'])
+
+    # 将新弹幕内容转换为 DataFrame，并自动生成 'id'
+    new_df = pd.DataFrame(danmaku_content, columns=['content'])
+    # new_df['id'] = range(len(existing_df), len(existing_df) + len(new_df))
+
+    # 合并现有内容和新内容
+    updated_df = pd.concat([existing_df, new_df], ignore_index=True)
+
+    # 保存到 Excel 文件
+    updated_df.to_excel(file_name, index=False)
+
+
+def savef(video_url):
+    """ 主函数，获取并展示弹幕数据 """
+
+    # 提取BV号
+    bv_id = re.search(r'BV\w+', video_url).group(0)
+
+    # 获取视频CID
+    cid = get_video_info(bv_id)
+    # print(f"提取的CID: {cid}")
+
+    # 获取弹幕数据
+    xml_data = get_danmaku_xml(cid)
+    danmakus = parse_danmaku_xml(xml_data)
+
+    save_danmaku_to_excel(danmakus, file_name='danmaku_content.xlsx')
+
+    # # 保存并打印弹幕
+    # for i, danmaku in enumerate(danmakus):
+    #     # 遍历爬取的弹幕内容并逐条保存
+    #
+    #     print(f"弹幕 {i + 1}: {danmaku}")
+    # file_name = 'danmaku_content.xlsx'
+    # print(f"弹幕内容已保存到 {file_name}")
+
+# 网址
+driver = ChromiumPage()
+url = "https://search.bilibili.com/all?keyword=2024%E5%B7%B4%E9%BB%8E%E5%A5%A5%E8%BF%90%E4%BC%9A&from_source=webtop_search&spm_id_from=333.1007&search_source=3&page=2&o=36"
+driver.get('https://search.bilibili.com/all?keyword=2024%E5%B7%B4%E9%BB%8E%E5%A5%A5%E8%BF%90%E4%BC%9A&from_source=webtop_search&spm_id_from=333.1007&search_source=3&page=2&o=36')
+j=0
+for page in range(8):
+    bv_list = get_video_details(url)
+    for bv in bv_list:
+        if j==300:
+            break
+        j+=1
+        # 函数入口，定义函数需要调用
+        video_url = f'https://www.bilibili.com/video/{bv}/?spm_id_from=333.337'
+        # 调用保存的函数
+        print(f"v{j}")
+        savef(video_url)
+    # 点击下一页
+    driver.ele('css:vui_button vui_pagenation--btn vui_pagenation--btn-side').click()
+
+
+
+
+
+
+
+
+
+
+
+
+
+