Question

我使用脚本来拼写一些网站并从那里获取新闻。此脚本中的每个函数都解析一个站点并返回文章列表，然后我想将它们组合在一个大列表中。如果我按站点分区，它需要很长时间，我决定使用多线程。我在底部发现了一个类似这样的样本，但对我来说它似乎不是pithonic。如果我再添加一个函数来解析另一个站点，我将需要每次添加相同的代码块：

qN = Queue()
Thread(target=wrapper, args=(last_news_from_bar, qN)).start()
news_from_N = qN.get()
for new in news_from_N:
    news.append(new)

还有另一种方法可以做这种事吗？

#!/usr/bin/python
# -*- coding: utf-8 -*-
from queue import Queue
from threading import Thread


def wrapper(func, queue):
    queue.put(func())


def last_news_from_bar():
    ...
    return list_of_articles #[['title1', 'http://someurl1', '2017-09-13'],['title2', 'http://someurl2', '2017-09-13']]


def last_news_from_foo():
    ...
    return list_of_articles


q1, q2 = Queue(), Queue()

Thread(target=wrapper, args=(last_news_from_bar, q1)).start()
Thread(target=wrapper, args=(last_news_from_foo, q2)).start()

news_from_bar = q1.get()
news_from_foo = q2.get()

all_news = []

for new in news_from_bar:
    news.append(new)
for new in news_from_foo:
    news.append(new)

print(all_news)

Answer 1

您应该做的就是使用单个队列并扩展结果数组：

q1 = Queue()

Thread(target=wrapper, args=(last_news_from_bar, q1)).start()
Thread(target=wrapper, args=(last_news_from_foo, q1)).start()

all_news = []

all_news.extend(q1.get())
all_news.extend(q1.get())

print(all_news)

Answer 2

没有Queue的解决方案：

NEWS = []
LOCK = Lock()

def gather_news(url):
    while True:
        news = news_from(url)
        if not news: break
        with LOCK:
            NEWS.append(news)

if __name__ == '__main__':
    T = []
     for url in ['url1', 'url2', 'url3']:
        t = Thread(target=gather_news, args=(url,))
        t.start()
        T.append(t)

# Wait until all Threads done
for t in T: 
    t.join()

print(NEWS)

用于不同函数的Python多线程，返回值存储在一个列表中

2 个答案: