Python多进程：如何在不依赖Queue的情况下传递结果

news2024/9/23 17:24:29

随着数据的爆炸式增长，网络爬虫成为获取信息的强大工具。在爬取大量数据时，多进程技术可以显著提高效率。然而，如何在多进程中传递结果，而不依赖Queue，成为了一个值得探讨的问题。本文将以采集抖音短视频为案例，详尽讲解如何在Python中实现这一目标。

文章目录

简介
多进程与Queue的局限性
替代方案：使用管道、共享内存和临时文件
实战案例：采集抖音短视频
结论

1. 简介

在爬虫技术中，多进程可以显著提高数据采集效率。然而，传统的Queue在某些场景下存在局限性。本文将探讨如何在不依赖Queue的情况下，实现多进程间的数据传递。

2. 多进程与Queue的局限性

Queue是Python多进程模块提供的一种进程间通信机制，但它有以下局限性：

性能瓶颈：在大量数据传递时，Queue可能成为性能瓶颈。
复杂性：在复杂的多进程架构中，Queue的管理和维护较为复杂。

3. 替代方案

为了解决这些问题，我们可以使用以下替代方案：

管道（Pipe）：用于进程间的双向通信。
共享内存（Shared Memory）：通过共享变量实现数据传递。
临时文件：将数据写入临时文件，由主进程读取。

4. 实战案例：采集抖音短视频

环境配置

在开始之前，我们需要配置爬虫代理IP和设置useragent及cookie，以提高爬虫的成功率。本文使用的爬虫代理服务为亿牛云。

代码实现

import multiprocessing
import requests
import json
import time
from multiprocessing import Pipe, Process
from bs4 import BeautifulSoup

# 代理配置 亿牛云爬虫代理加强版
proxy_host = "www.16yun.cn" #代理域名
proxy_port = "代理端口"
proxy_user = "代理用户名"
proxy_pass = "代理密码"
proxy = {
    "http": f"http://{proxy_user}:{proxy_pass}@{proxy_host}:{proxy_port}",
    "https": f"http://{proxy_user}:{proxy_pass}@{proxy_host}:{proxy_port}"
}

# User-Agent 和 Cookie
headers = {
    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
    "Cookie": "your_cookie_here"
}

def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        elapsed_time = end_time - start_time
        return result, elapsed_time
    return wrapper

@timer
def fetch_video_data(video_url):
    response = requests.get(video_url, headers=headers, proxies=proxy)
    soup = BeautifulSoup(response.content, 'html.parser')
    video_data = soup.find('script', {'type': 'application/json'}).string
    return json.loads(video_data)

def worker(video_url, conn):
    result, elapsed_time = fetch_video_data(video_url)
    conn.send((result, elapsed_time))
    conn.close()

def main():
    video_urls = ["https://www.douyin.com/video/1", "https://www.douyin.com/video/2"]
    processes = []
    parent_connections = []

    for url in video_urls:
        parent_conn, child_conn = Pipe()
        p = Process(target=worker, args=(url, child_conn))
        processes.append(p)
        parent_connections.append(parent_conn)
        p.start()

    for p in processes:
        p.join()

    for parent_conn in parent_connections:
        result, elapsed_time = parent_conn.recv()
        print(f"Video Data: {result}")
        print(f"Elapsed Time: {elapsed_time}")

if __name__ == '__main__':
    main()