-
Notifications
You must be signed in to change notification settings - Fork 35
/
bot.py
91 lines (73 loc) · 3.23 KB
/
bot.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import sys
import markovify
import ConfigParser
import json
import sys
from expiringdict import ExpiringDict
from tweepy.streaming import StreamListener
from tweepy import OAuthHandler
from tweepy import Stream
from tweepy import API
# read configuration file (.twitter)
config = ConfigParser.ConfigParser()
config.read('.twitter')
reload(sys)
sys.setdefaultencoding("utf-8")
# store Twitter specific credentials
consumer_key = config.get('apikey', 'key')
consumer_secret = config.get('apikey', 'secret')
access_token = config.get('token', 'token')
access_token_secret = config.get('token', 'secret')
stream_rule = config.get('app', 'rule')
account_screen_name = config.get('app', 'account_screen_name').lower()
account_user_id = config.get('app', 'account_user_id')
auth = OAuthHandler(consumer_key, consumer_secret)
auth.set_access_token(access_token, access_token_secret)
twitterApi = API(auth, wait_on_rate_limit=True, wait_on_rate_limit_notify=True)
class ReplyToTweet(StreamListener):
def __init__(self):
self.__max_per_user = 2 # max replies per user within an hour
self.__rate_limits_per_user = ExpiringDict(max_len=1000, max_age_seconds=(1 * 60))
self.initialize_model()
def initialize_model(self):
with open("./tweets.txt") as f:
text = f.read()
self.__text_model = markovify.Text(text)
def response(self, text):
return str(self.__text_model.make_short_sentence(80, tries=10).lower() or '')
def on_data(self, data):
if not data:
return
tweet = json.loads(str(data).strip())
retweeted = tweet.get('retweeted')
from_self = tweet.get('user', {}).get('id_str','') == account_user_id
# if retweeted is not None and not retweeted and not from_self:
if retweeted is not None and not retweeted:
tweetId = tweet.get('id_str')
screenName = tweet.get('user', {}).get('screen_name')
tweetText = tweet.get('text')
# check if we already replied to this user and when
# rate limit replies per user
# update rate for this user
existing_rate = int(0 if self.__rate_limits_per_user.get(screenName) is None else self.__rate_limits_per_user.get(screenName))
self.__rate_limits_per_user[screenName] = existing_rate + 1
# check rate limit, and reply accordingly
if (existing_rate == self.__max_per_user):
replyText = '@' + screenName + ' lo siento, pero solo 2 respuestas por hora 🤖👮'
twitterApi.update_status(status=replyText, in_reply_to_status_id=tweetId)
print("replied to this user, saying he/she should wait for an hour.")
return
elif (existing_rate > self.__max_per_user):
print("*already* replied to this user, he/she should wait for an hour, ignoring.")
return
replyText = '@' + screenName +" "+ self.response("")
twitterApi.update_status(status=replyText, in_reply_to_status_id=tweetId)
return
def on_error(self, status):
print(status)
if __name__ == '__main__':
streamListener = ReplyToTweet()
twitterStream = Stream(auth, streamListener)
twitterStream.filter(track=['@'+account_screen_name])