forked from basti2342/retweet-bot
/
retweet.py
82 lines (64 loc) · 2.73 KB
/
retweet.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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os, tweepy, inspect, hashlib
path = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
# your hashtag or search query and tweet language (empty = all languages)
hashtag = os.environ["SEARCH_QUERY"]
tweetLanguage = os.environ["TWEET_LANGUAGE"]
# Number retweets per time
num = int(os.environ["NUMBER_OF_RT"])
# blacklisted users and words
userBlacklist = ["tipsder", "maximaformacion", "RosanaFerrero", "maximaDataSci_R"]
wordBlacklist = ["RT", u"♺"]
# build savepoint path + file
hashedHashtag = hashlib.md5(hashtag.encode('ascii')).hexdigest()
last_id_filename = "last_id_hashtag_%s" % hashedHashtag
rt_bot_path = os.path.dirname(os.path.abspath(__file__))
last_id_file = os.path.join(rt_bot_path, last_id_filename)
# create bot
auth = tweepy.OAuthHandler(os.environ["TWITTER_CONSUMER_KEY"], os.environ["TWITTER_CONSUMER_SECRET"])
auth.set_access_token(os.environ["TWITTER_ACCESS_TOKEN"], os.environ["TWITTER_ACCESS_TOKEN_SECRET"])
api = tweepy.API(auth)
# retrieve last savepoint if available
try:
with open(last_id_file, "r") as file:
savepoint = file.read()
except IOError:
savepoint = ""
print("No savepoint found. Bot is now searching for results")
# search query
timelineIterator = tweepy.Cursor(api.search, q=hashtag, since_id=savepoint, lang=tweetLanguage).items(num)
# put everything into a list to be able to sort/filter
timeline = []
for status in timelineIterator:
timeline.append(status)
try:
last_tweet_id = timeline[0].id
except IndexError:
last_tweet_id = savepoint
# filter @replies/blacklisted words & users out and reverse timeline
#timeline = filter(lambda status: status.text[0] = "@", timeline) - uncomment to remove all tweets with an @mention
timeline = filter(lambda status: not any(word in status.text.split() for word in wordBlacklist), timeline)
timeline = filter(lambda status: status.author.screen_name not in userBlacklist, timeline)
timeline = list(timeline)
timeline.reverse()
tw_counter = 0
err_counter = 0
# iterate the timeline and retweet
for status in timeline:
try:
print("(%(date)s) %(name)s: %(message)s\n" % \
{"date": status.created_at,
"name": status.author.screen_name.encode('utf-8'),
"message": status.text.encode('utf-8')})
api.retweet(status.id)
tw_counter += 1
except tweepy.error.TweepError as e:
# just in case tweet got deleted in the meantime or already retweeted
err_counter += 1
# print e
continue
print("Finished. %d Tweets retweeted, %d errors occured." % (tw_counter, err_counter))
# write last retweeted tweet id to file
with open(last_id_file, "w") as file:
file.write(str(last_tweet_id))