bot.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
# 
# Copyright (c) 2013 Malte Bublitz, https://malte-bublitz.de
# Copyright (c) 2013 Benjamin Leuckefeld, https://atroxlp.de
# Copyright (c) 2012 Juan C. Olivares <juancri@juancri.com>
# based on original code by Christian Palomares <palomares.c@gmail.com>
# 
# Distributed under the MIT/X11 license. Check LICENSE for details.
# 

import time
import os
import _twitter
import sys
import codecs

class RetweetBot(object):
	def run(self, me, hashtag, add_hashtags, blacklist, blacklistuser, sleep, count, native_retweet, con_key, con_sec, tok_key, tok_sec, gconfig, check_conds, logfilename):
		import log
		logger = log.Log(logfilename)
		if len(sys.argv)>1:
			if sys.argv[1] == "-v":
				logger.setConsoleLogLevel(log.INFO)
	
		logger.info("----------------------------------")
		logger.info("     Flying Sheep Retweet Bot")
		logger.info("         Bot is starting!")
		logger.info("----------------------------------")
		logger.info("")
		
		api = _twitter.Api (
			consumer_key = con_key,
			consumer_secret = con_sec,
			access_token_key = tok_key,
			access_token_secret = tok_sec
		)
		
		logger.important("Bot started!")
		logger.important("Searching tweets containing '" + hashtag + "'.")
		
		# loop
		lastid = None
		first = 1
		while 1:
			# get last tweets
			try:
				timeline = api.GetSearch(hashtag, since_id = lastid, per_page = count)
			except:
				logger.error("Could not get tweets!")
				continue
			
			# update last ID
			if len (timeline) > 0:
				lastid = timeline [0].id
				logger.info("Last ID updated:" + str(lastid))
			
			# skip the first time
			if first > 0:
				first = 0
				continue
			
			# check tweets
			for status in timeline:	
				# check if tweet contains hashtag
				if status.text.lower().find(hashtag.lower()) < 0:
					continue
			
				# check if tweet send by you
				if status.user.screen_name.lower() == me.lower():
					continue
					
				# check if tweet send by blacklistuser
				if len(blacklistuser) > 0:
					send = True
					for u in blacklistuser:
						if status.user.screen_name.lower() == u.lower():
							send = False
					if not send:
						continue
				
				# check if tweet contains blacklisted word
				if len(blacklist) > 0:
					send = True
					for bl in blacklist:
						if status.text.lower().find(bl.lower()) >= 0:
							send = False
					if not send:
						continue
				
				# check if tweet contains one additional hashtag
				if len(add_hashtags) > 0:
					send = False
					for ht in add_hashtags:
						if status.text.lower().find(ht.lower()) >= 0:
							send = True
					if not send:
						continue
						
				# check general configs
				for gc in gconfig:
					if gc.lower() == "blacklist":
						send = True
						file = codecs.open("blacklist.bc", "a+", "utf-8")
						for line in file: 
							if not line == "":
								if not line == " ":
									if status.text.lower().find(line.lower()) >= 0:
										send = False
									
						file.close()
						if not send:
							continue
							
					if gc.lower() == "blacklistusers":
						send = True
						file = codecs.open("blacklistusers.bc", "a+", "utf-8")
						for line in file: 
							if not line == "":
								if not line == " ":
									if status.user.screen_name.lower() == line.lower():
										send = False
									
						file.close()
						if not send:
							continue
						
					if gc.lower() == "additionalhashtags":
						send = False
						file = codecs.open("additionalHashtags.bc", "a+", "utf-8")
						for line in file: 
							if not line == "":
								if not line == " ":
									if status.text.lower().find(line.lower()) >= 0:
										send = True
									
						file.close()
						if not send:
							continue
				
				# check additional conditions
				if not check_conds(status):
					continue
				
				try:
					# let's retweet
					if native_retweet:
						api.PostRetweet (status.id)
						logger.important("Retweeting: " + status.user.screen_name + " " + status.text)
					else:
						retweet = 'RT @' + status.user.screen_name + ": " + status.text
						if len (retweet) > 140:
							retweet = retweet [:137] + "..."
						api.PostUpdate (retweet)
						logger.important("Tweeting: " + retweet)
				except:
					logger.error("Could not retweet!")

			# zZzZzZ
			time.sleep(sleep)
			
def main():
	bot = RetweetBot()
	bot.run()

if __name__=="__main__":
	try:
		main()
	except KeyboardInterrupt:
		pass