def command_handler(message):
status = ['/todo', '/doing', '/done']
command_bot = Command()
if message.command == '/new':
command_bot.command_new(message.chat, message.msg)
elif message.command == '/rename':
command_bot.command_rename(message.msg, message.user_name,
message.chat)
elif message.command == '/duplicate':
command_bot.command_duplicate(message.msg, message.user_name,
message.chat)
elif message.command == '/delete':
command_bot.command_delete(message.msg, message.user_name,
message.chat)
elif message.command in status:
command_bot.command_status(message.msg, message.user_name,
message.chat, message.command)
elif message.command == '/list':
command_bot.command_list(message.chat)
elif message.command == '/dependson':
command_bot.command_dependson(message.msg, message.user_name,
message.chat)
elif message.command == '/priority':
command_bot.command_priotiry(message.msg, message.user_name,
message.chat)
elif message.command == '/start':
helper = Help()
send_message("Welcome! Here is a list of things you can do.",
message.chat)
send_message(helper.get_help(), message.chat)
elif message.command == '/help':
helper = Help()
send_message("Here is a list of things you can do.", message.chat)
send_message(helper.get_help(), message.chat)
elif message.command == '/showPriority':
command_bot.command_show_priority(message.chat)
else:
send_message(
"I'm sorry " + message.user_name + ". I'm afraid I can't do that.",
message.chat)
class ZulipBot(object):
def __init__(self):
self.client = zulip.Client(site="https://saharsh.zulipchat.com/api/")
self.subscribe_all()
self.trans = Translate()
self.tw = Twimega()
self.pnr = Pnr()
self.weather = Weather()
self.geo = Geocode()
self.searching = Places()
self.help = Help()
print("Initialization Done ...")
self.subkeys = [
"translate", "weather", "pnr", "post", "post_image", "twitter",
"help", "search"
]
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']
if sender_email == BOT_MAIL:
return
if content[0].lower() == "ninjas33" or content[0] == "@**ninjas33**":
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() == "pnr":
message = self.pnr.get_pnr(content[2])
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": message
})
if content[1].lower() == "twitter":
if len(content) > 2 and content[2] == "post":
if self.tw.stream == msg["display_recipient"]:
status = self.tw.post(" ".join(content[3:]))
x = json.dumps(status._json)
x = json.loads(x)
message = "https://twitter.com/{}/status/{}".format(
x["user"]["screen_name"], x["id_str"])
message = "Tweet Posted\n" + message
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": message
})
else:
message = "Use the stream **{}** to post a tweet".format(
self.tw.stream)
self.client.send_message({
"type": "stream",
"to": sender_email,
"content": message
})
if len(content) > 2 and content[2] == "post_image":
if self.tw.stream == msg["display_recipient"]:
status = self.tw.post_image(content[3],
" ".join(content[4:]))
if isinstance(status, str):
message = status
else:
x = json.dumps(status._json)
x = json.loads(x)
message = "https://twitter.com/{}/status/{}".format(
x["user"]["screen_name"], x["id_str"])
message = "Tweet Posted\n" + message
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": message
})
else:
message = "Use the stream **{}** to post a tweet".format(
self.tw.stream)
self.client.send_message({
"type": "private",
"to": sender_email,
"content": message
})
if content[1].lower() == "help" and len(content) == 2:
message = self.help.get_help()
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": message
})
if content[1].lower() == "weather":
place = " ".join(content[2:])
try:
result = self.weather.getWeather(self.geo.convert(place))
message = "**" + "Weather update of " + place + "**" + "\n" + "Summary : " + "**" + result[
"currently"][
"summary"] + "**" + "\n" + "Temparature : " + "**" + str(
result["currently"]["temperature"]
) + "**" + '\n' + "Apparent Temparature : " + "**" + str(
result["currently"]["apparentTemperature"]
) + "**" + "\n" + "Dew Point : " + "**" + str(
result["currently"]["dewPoint"]
) + "**" + "\n" + "Humidity : " + "**" + str(
result["currently"]["humidity"]) + "**"
except KeyError:
message = "Not Working Right Now"
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": message
})
if content[1].lower() == "search":
place = " ".join(content[2:])
try:
result = self.searching.getPlaces(place)
y = result['results']
message = "Result for search \n"
#print(result)
for i in range(len(y)):
message += str(
i + 1) + ". " + y[i]['name'] + " , Rating:" + str(
y[i]['rating']
) + " \n Address : " + y[i]['formatted_address']
message += "\n"
except KeyError:
message = "Not Working Right Now"
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": message
})
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 "ninjas33" in content and content[0] != "ninjas33":
self.client.send_message({
"type": "stream",
"subject": msg["subject"],
"to": msg["display_recipient"],
"content": "Hey there! :blush:"
})
else:
return
