class ZulipBot(object):
def __init__(self):
self.client = zulip.Client(site="https://technh.zulipchat.com/api/", api_key="vkEQgQYDPUgAGmXaTXdMPsMwlkkgMfM5", email="*****@*****.**")
self.subscribe_all()
self.hacknews = Hackernews()
self.trans = Translate()
self.movie= Movie()
self.lyrics = Lyrics()
self.holiday = Holiday()
self.currency = Currency()
self.cricket = Cricket()
self.github = GitHub()
self.chatbot = ChatBot(name="technehru")
print("done init")
self.subkeys = ["use", "help", "translate", "hackernews", "hn", "hotel", "HN", "cricnews", "cricketnews", "movie", "currency", "holiday", "lyrics", "github"]
def subscribe_all(self):
json = self.client.get_streams()["streams"]
streams = [{"name": stream["name"]} for stream in json]
self.client.add_subscriptions(streams)
def process(self, msg):
content = msg["content"].split()
sender_email = msg["sender_email"]
ttype = msg["type"]
stream_name = msg['display_recipient']
stream_topic = msg['subject']
print(content)
if sender_email == BOT_MAIL:
return
print("Sucessfully heard.")
if content[0].lower() == "technehru" or content[0] == "@**Technehru**":
if content[1].lower() == "help" or content[1].lower() == "use":
message = open("help.txt", "r")
self.client.send_message({
"type": "stream",
"to": stream_name,
"subject": stream_topic,
"content": message.read()
})
if content[1].lower() == "translate":
ip = content[2:]
ip = " ".join(ip)
message = self.trans.translate(ip)
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": message
})
if content[1].lower() == "movie":
ip = content[2:]
ip = " +".join(ip)
message = self.movie.about(ip)
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": message
})
if content[1].lower() == "lyrics":
author = content[2]
title = content[3:]
title = " ".join(title)
message = self.lyrics.about(author, title)
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": message
})
if content[1].lower() == 'holiday':
quote_data = self.holiday.holiday()
self.client.send_message({
"type": "stream",
"to": stream_name,
"subject": stream_topic,
"content": quote_data
})
if content[1].lower() == 'currency':
x = content[2]
y = content[3]
quote_data = self.currency.currency(x,y)
self.client.send_message({
"type": "stream",
"to": stream_name,
"subject": stream_topic,
"content": quote_data
})
if content[1].lower() == "cricnews" or content[1].lower() == "cricketnews":
news = self.cricket.news()
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": news
})
if content[1].lower() == 'hackernews' or content[1].lower() == 'hn':
news = self.hacknews.get_hackernews()
self.client.send_message({
"type": "stream",
"to": stream_name,
"subject": stream_topic,
"content": news
})
if content[1].lower() == 'github':
if content[2].lower() == 'reopen' and content[3].lower() == 'issue':
repo = content[4]
num = int(content[5])
result = self.github.reopen_issue(repo, num)
self.client.send_message({
"type": "stream",
"to": stream_name,
"subject": stream_topic,
"content": result
})
if content[2].lower() == 'comment' and content[3].lower() == 'issue':
repo = content[4]
num = int(content[5])
comment = content[6:]
comment = " ".join(comment)
result = self.github.comment_issue(repo, num, comment)
self.client.send_message({
"type": "stream",
"to": stream_name,
"subject": stream_topic,
"content": result
})
if content[2].lower() == 'close' and content[3].lower() == 'issue':
repo = content[4]
num = int(content[5])
result = self.github.close_issue(repo, num)
self.client.send_message({
"type": "stream",
"to": stream_name,
"subject": stream_topic,
"content": result
})
if content[2].lower() == 'assign' and content[3].lower() == 'issue':
repo = content[4]
num = int(content[5])
assignee = content[6]
result = self.github.assign_issue(repo, num, assignee)
self.client.send_message({
"type": "stream",
"to": stream_name,
"subject": stream_topic,
"content": result
})
if content[1] not in self.subkeys:
ip = content[1:]
ip = " ".join(ip)
message = self.chatbot.get_response(ip).text
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": message
})
elif "technehru" in content and content[0].lower() != "technehru":
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": "Hey there! :blush:"
})
else:
return
