shitpost/replybot.py at master · yrsegal/shitpost · GitHub

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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
# coding=utf-8
from __future__ import print_function
from shitpostbot import *
import atexit
import string
import multiprocessing
path = os.path.dirname(sys.argv[0])

def getUniques(l):
	nl = l[:]
	cl = l[:]
	for i in l:
		intimes = 0
		while i in cl:
			intimes += 1
			cl.remove(i)
		if intimes > 1:
			for j in xrange(intimes-1):
				nl.remove(i)
	return nl


class cSL(tweepy.StreamListener):

	def on_data(self, data):
		jdata = json.loads(data.strip().encode('ascii',errors='ignore'))
		name = jdata.get('user', {}).get('screen_name', 'Name not found')
		displayname = jdata.get('user', {}).get('name', 'Name not found').encode('ascii',errors='ignore')
		selfname = api.me().screen_name
		color_print("Replying to tweet: ")
		color_print(format("{name} (@{handle})", name=displayname, handle=name))
		color_print(jdata.get('text'))

		retweeted = jdata.get('retweeted', False)
		from_self = jdata.get('user', {}).get('id',0) == api.me().id
		people = jdata.get('entities', {}).get('user_mentions', [])
		peoplenames = getUniques([i.get('screen_name') for i in people])
		if name in peoplenames:     peoplenames.remove(name)
		if selfname in peoplenames: peoplenames.remove(selfname)
		peoplenames.insert(0, name)
		names = " ".join([format("@{handle}", handle=i) for i in peoplenames if i])

		dot = "." if config.get('public') in jdata.get('text', '') else ""

		if isinstance(config.get('notags'), str):
			if config.get('notags') in jdata.get('text', ''):
				dot, names = "", ""
		elif isinstance(config.get('notags'), list):
			removedots = True
			for i in config.get('notags'):
				if i not in jdata.get('text', ''):
					removedots = False
			if removedots:
				dot, names = "", ""

		if jdata.get('text', '')[:2] != "RT" and not retweeted and not from_self:
			try:
				text = generate(debug=True)
				api.update_status(status=dot+names+" "+text, in_reply_to_status_id = jdata.get('id_str', ''))
				color_print(format("Sent tweet: {text}", text=dot+names+" "+text))
			except Exception, e:
				color_print(format_traceback(e, "Error in sending tweet:"), color=ansi_colors.RED)
		print()
		return True

	def on_error(self, status):
		color_print("Error: "+str(status), color=ansi_colors.DARKRED)
		time.sleep(5)
		return True

def tweetStream():
	l = cSL()
	stream = tweepy.Stream(api.auth, l)

	while True:
		try:
			targets = []
			if isinstance(config.get("searchfor"), str):
				targets.append(config.get("searchfor"))
			elif isinstance(config.get("searchfor"), list):
				targets = config.get("searchfor")
			else:
				targets = ["gimme a shitpost", "gimme a public shitpost"]
			while not connected_to_internet():
				time.sleep(1)
			stream.filter(track=targets)
		except Exception, e:
			if isinstance(e, KeyboardInterrupt):
				break
			color_print(format_traceback(e, "Error in stream filter:"), color=ansi_colors.RED)
			print()

def cleanup():
	global replyThread, mainThread
	replyThread.terminate()
	mainThread.terminate()

def centralmain():
	global replyThread, mainThread
	atexit.register(cleanup)
	if not os.path.exists(os.path.join(path, "shitpostconfig.json")):
		json.dump({
			"time": 360,
			"notifytime": 60,
			"searchfor": ["gimme a shitpost", "gimme a public shitpost"],
			"public": "public",
			"notags": ["just", "do it"]
			}, open(os.path.join(path, "shitpostconfig.json"), "w"), indent=2, sort_keys=True)
	replyThread = multiprocessing.Process(target=tweetStream)
	mainThread = multiprocessing.Process(target=main)
	replyThread.start()
	mainThread.start()
	try:
		replyThread.join()
	except KeyboardInterrupt: cleanup()

if __name__ == "__main__": centralmain()