在特定时间运行 python 代码

Question

我有以下代码：

def gen_stream():
    stream  = twitter.STREAMING_API(key=1, payload={})
    for tweet in stream.run():
        yield tweet
def pipeline():
    for tweet in gen_stream():
        tweet_id=tweet.get('id')
        print(tweet_id)
if __name__ == '__main__':
    pipeline()

我怎样才能只运行 1 分钟的代码。 我试过使用

if __name__ == '__main__':
    endtime=time.time()+60*1
    while time.time()<endtime:
        pipeline()

但是这段代码在结束时间之后不断产生推文 ID。如何在指定时间内运行pipeline？

P.S gen_stream() 工作正常。

Answer 1

您必须在单独的线程或进程中生成管道并从外部终止它。很好的是使用 pub sub 模式并很好地断开与流的连接。

将参数传递给管道，但它不会很好地关闭连接。

Answer 2

问题是代码进入pipeline函数并在里面永远循环。

让pipeline 接受endtime 参数

def gen_stream():
    stream  = twitter.STREAMING_API(key=1, payload={})
    for tweet in stream.run():
        yield tweet

def pipeline(endtime):
    for tweet in gen_stream():
        if time.time() > endtime:
            break
        tweet_id=tweet.get('id')
        print(tweet_id)

if __name__ == '__main__':
    endtime=time.time()+60*1
    pipeline(endtime)