class Bot(threading.Thread):
def __init__(self, wa_phone, wa_password, contacts, irc_server, irc_port, owner_nick, log_file):
threading.Thread.__init__(self)
self.must_run = True
self.irc_server = irc_server
self.irc_port = irc_port
self.owner_nick = owner_nick
self.wa_phone = wa_phone
self.log_file = log_file
irc_nick = contacts[wa_phone]
self.irc_nick = irc_nick
self.wa_password = wa_password
self.contacts = contacts
self.irc_i = IRCInterface(self.irc_server, self.irc_port, self.irc_nick, channels_from_contacts(self.contacts), self.irc_msg_received, self.stop)
self.wa_i = WAInterface(self.wa_phone, self.wa_password, self.wa_msg_received, self.stop)
@catch_them_all
def run(self):
try:
self.must_run = True
info("Starting IRC")
self.irc_i.start()
info("Waiting for IRC")
self.irc_i.wait_connected()
info("Starting WA")
self.wa_i.start()
info("Waiting for WA")
self.wa_i.wait_connected()
info("Main loop running")
except:
info("Main loop closing")
self.stop()
def stop(self):
info("Stop instructed, about to stop main loop")
self.must_run = False
self.irc_i.stop()
self.wa_i.stop()
def get_group_from_chan(self, contacts, irc_channel):
for k,v in contacts.items():
if v.lower() == irc_channel.lower():
return k
raise Exception("Channel not found in contact list")
@catch_them_all
def irc_msg_received(self, message):
store_msg(message, self.log_file)
info(" <<< Received IRC message: %s" %message)
if message.chan == self.irc_nick:
info("Private message")
if message.target is None:
raise Exception("Private message sent to no one?")
try:
wa_target = self.contacts[message.target] #try by phone
except KeyError:
wa_target = self.get_group_from_chan(self.contacts, message.target) #try by nick
wa_target += "@s.whatsapp.net"
msg = "<%s> %s" %(message.get_nick(), message.msg.split(":", 1)[1])
self.wa_i.send(wa_target, msg)
else:
info("Group message")
msg = "<%s> %s" %(message.get_nick(), message.msg)
try:
group = self.get_group_from_chan(self.contacts, message.chan)
self.wa_i.send(group, msg)
except:
info("Cannot send message to channel %s" %message.chan)
@catch_them_all
def wa_msg_received(self, message):
store_msg(message, self.log_file)
info(" <<< Received WA message: %s" %message)
if message.chan == self.wa_phone:
#private message
if message.target is None:
# directed to bot itself
nick = self.contacts[message.get_nick()]
irc_msg = "<%s> %s" %(nick, message.msg)
irc_target = self.contacts[message.nick_full.split("@")[0]]
self.irc_i.send(self.owner_nick, irc_msg)
else:
# directed to someone
try:
phone = message.get_nick()
nick = self.contacts[phone]
target = self.get_group_from_chan(self.contacts, message.target)
msg = "<%s> %s" %(target, message.msg)
self.irc_i.send(target, msg)
except:
error("Couldn't relay directed WA msg to IRC")
else:
#group message
try:
msg = "<%s> %s" %(self.contacts[message.get_nick()], message.msg)
except:
error("Contact not recognized")
msg = "<%s> %s" %(message.get_nick(), message.msg)
try:
self.irc_i.send(self.contacts[message.chan], msg)
except:
error("Channel %s not recognized" %(message.chan))
