我希望在MongoDB中存储了x#条推文后,让Tweepy Streaming API停止拉入推文。
我已经在类中尝试了 IF 和 WHILE 语句,使用计数器进行定义,但无法让它在某个 X 数量处停止。这对我来说是一个真正的头撞。我在这里找到了这个链接: https://groups.google.com/forum/#!topic/tweepy/5IGlu2Qiug4 但我复制它的努力失败了。它总是告诉我 init 需要一个额外的参数。我相信我们的 Tweepy 身份验证集不同,所以它不是苹果对苹果。
有什么想法吗?
from tweepy.streaming import StreamListener
from tweepy import OAuthHandler
from tweepy import Stream
import json, time, sys
import tweepy
auth = tweepy.OAuthHandler(CONSUMER_KEY, CONSUMER_SECRET)
auth.set_access_token(OAUTH_TOKEN, OAUTH_TOKEN_SECRET)
class StdOutListener(StreamListener):
def on_status(self, status):
text = status.text
created = status.created_at
record = {'Text': text, 'Created At': created}
print record #See Tweepy documentation to learn how to access other fields
collection.insert(record)
def on_error(self, status):
print 'Error on status', status
def on_limit(self, status):
print 'Limit threshold exceeded', status
def on_timeout(self, status):
print 'Stream disconnected; continuing...'
stream = Stream(auth, StdOutListener())
stream.filter(track=['tv'])
>您需要在类中添加一个计数器 __init__
,然后在 on_status
内递增它。然后,当计数器低于 20 时,它将在集合中插入一条记录。这可以如下所示:
def __init__(self, api=None):
super(StdOutListener, self).__init__()
self.num_tweets = 0
def on_status(self, status):
record = {'Text': status.text, 'Created At': status.created_at}
print record #See Tweepy documentation to learn how to access other fields
self.num_tweets += 1
if self.num_tweets < 20:
collection.insert(record)
return True
else:
return False