コード例 #1
0
ファイル: bot.py プロジェクト: badrul/breakbot
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:
            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:
            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 Exception, e:
                error(traceback.print_exc())
                error("Cannot send message to channel %s: %s" %(message.chan, e))
コード例 #2
0
ファイル: bot.py プロジェクト: pezhik/breakbot
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("Connecting IRC client (%s@%s:%s)" %(self.irc_nick, self.irc_server, self.irc_port))
            self.irc_i.start()
            self.irc_i.wait_connected()
            info("Connecting WA client (%s)" %self.wa_phone)
            self.wa_i.start()
            self.wa_i.wait_connected()
            info("Bot ready.")
        except:
            info("Main loop closing")
            self.stop()
    def stop(self):
        info("Bot stopping...")
        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(" <<< IRC %s" %message)

        if message.chan == self.irc_nick:
            if message.target is None:
                raise Exception("Target not specified. Please prefix you private messages with a nickname (e.g. 'person1: hello') or phone number (e.g. '+34555555373: hello')")
            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:
            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 Exception, e:
                error("Cannot send message to channel %s: %s" %(message.chan, e))
コード例 #3
0
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("Connecting IRC client (%s@%s:%s)" %
                 (self.irc_nick, self.irc_server, self.irc_port))
            self.irc_i.start()
            self.irc_i.wait_connected()
            info("Connecting WA client (%s)" % self.wa_phone)
            self.wa_i.start()
            self.wa_i.wait_connected()
            info("Bot ready.")
        except:
            info("Main loop closing")
            self.stop()

    def stop(self):
        info("Bot stopping...")
        self.must_run = False
        self.irc_i.stop()
        self.wa_i.stop()

    def get_wa_id_from_name(self, contacts, name):
        for k, v in contacts.items():
            if v.lower() == name.lower():
                return k
        raise KeyError("Name %s not found in contact list" % name)

    @catch_them_all
    def irc_msg_received(self, message):
        store_msg(message, self.log_file)
        info(" <<< IRC %s" % message)

        if message.chan == self.irc_nick:
            if message.target is None:
                raise Exception(
                    "Target not specified. Please prefix you private messages with a nickname (e.g. 'person1: hello') or phone number (e.g. '+34555555373: hello')"
                )
            wa_target = message.target
            if message.target not in self.contacts:
                try:
                    wa_target = self.get_wa_id_from_name(
                        self.contacts, message.target)  #get by nick
                except KeyError:
                    if not wa_target.isdigit():
                        raise Exception(
                            "Whatsapp identifier '%s' not found in contact list, and does not look like a phone number"
                            % message.target)
                    warning(
                        "Phone number '%s' not found in contact list. Trying to send anyway..."
                        % message.target)
            wa_target += "@s.whatsapp.net"
            msg = "<%s> %s" % (message.get_nick(), message.msg)
            self.wa_i.send(wa_target, msg)
        else:
            msg = "<%s> %s" % (message.get_nick(), message.msg)
            try:
                group = self.get_wa_id_from_name(self.contacts, message.chan)
                self.wa_i.send(group, msg)
            except Exception, e:
                error("Cannot send message to channel %s: %s" %
                      (message.chan, e))
コード例 #4
0
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(" <<< IRC %s" % message)

        if message.chan == self.irc_nick:
            if message.target is None:
                raise Exception(
                    "Target not specified. Please prefix you private messages with a nickname (e.g. 'person1: hello') or phone number (e.g. '+34555555373: hello')"
                )
            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:
            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 Exception, e:
                error(traceback.print_exc())
                error("Cannot send message to channel %s: %s" %
                      (message.chan, e))
コード例 #5
0
ファイル: bot.py プロジェクト: SM-Networkx/breakbot
class Bot(threading.Thread):
    def __init__(self, cfg, contacts):
        threading.Thread.__init__(self)
        self.must_run = True
        self.irc_server = cfg["irc_server_name"]
        self.irc_port = cfg["irc_server_port"]
        self.owner_nick = cfg["bot_owner_nick"]
        self.wa_phone = cfg["wa_phone"]
        self.log_file = cfg["log_file"]
        self.irc_nick = cfg["irc_nick"]
        self.wa_password = cfg["wa_password"]
        self.contacts = contacts
        self.spamcheck = cfg["spamcheck"]
        self.logging = cfg["logging"]
        self.spamcheck_nick = cfg["spamcheck_nick"]
	self.verbose = cfg["verbose"]
	fwd_filters = cfg["filter"]
	if fwd_filters != "":
	    info(2, "Filter messages: "+fwd_filters)
	    filters = fwd_filters.split(",")
	    self.filters = filters
	    info(2, "Found %s Filters" %len(filters))
	else:
	    self.filters = []

	if len(self.filters) == 0:
	    info(2, "Filtering disabled")
        self.irc_i = IRCInterface(self.irc_server, self.irc_port, self.irc_nick, self.owner_nick, channels_from_contacts(self.contacts), self.irc_msg_received, self.stop)
        self.irc_i.spamcheck_enabled = self.spamcheck
        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(1, "Starting IRC")
            self.irc_i.start()
            info(1, "Waiting for IRC")
            self.irc_i.wait_connected()
            info(1, "Starting WA")
            self.wa_i.start()
            info(1, "Waiting for WA")
            self.wa_i.wait_connected()
	    info(1, "Program ready")
            info(3, "Main loop running")
        except:
            info(3, "Main loop closing")
            self.stop()
    def stop(self):
        info(2, "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(3, " <<< IRC %s" %message)
	if message.get_nick() == self.owner_nick and message.msg == "!die":
	    info(1, "Got die from Owner")
	    self.irc_i.cli.send("QUIT :Die from owner")
	    self.stop()
	elif message.msg == "!die":
	    info(1, "%s in %s tried to die me. Slap around him." %(message.get_nick(), message.chan))
	    self.irc_i.send_queue.put("PRIVMSG %s :\001ACTION slaps %s around with a large trout\001" %(message.chan, message.get_nick()))
        if self.spamcheck == 1:
            #if message.target is None:
                #raise Exception("Private message sent to no one?")
	    if message.get_nick() != self.spamcheck_nick and message.target != None:
		try:
		    wa_target = self.contacts[message.target] #try by phone
		except KeyError:
		    try:
			wa_target = self.get_group_from_chan(self.contacts, message.target) #try by nick
		    except:
			wa_target = ""
			info(2, "Group not recognized")
		if wa_target != "":
		    wa_target += "@s.whatsapp.net"
		    msg = "<%s> %s" %(message.get_nick(), message.msg.split(":", 1)[1])
		    self.wa_i.send(wa_target, msg)
	    elif message.get_nick() == self.spamcheck_nick:
		if self.irc_i.server_spamcheck == False:
		    self.irc_i.server_spamcheck = True
		    info(3, "SpamScanner scan is done. Now joining channels.")
		    self.irc_i.join_channels()
	    else:
		msg = "<%s> %s" %(message.get_nick(), message.msg)
		if len(self.filters) > 0:
		    matches = 0
		    for f in self.filters:
			if re.search(f, message.msg, re.I):
			    matches = matches + 1
		    if matches > 0:
			info(2, "IRC Message matching filters. Forwarding Message.")
			try:
			    group = self.get_group_from_chan(self.contacts, message.chan)
			    self.wa_i.send(group, msg)
			except Exception, e:
			    error(traceback.print_exc())
			    error("Cannot send message to channel %s: %s" %(message.chan, e))
		else:
		    info(3, "IRC Message filtering disabled. Forwarding Message.")
		    try:
			group = self.get_group_from_chan(self.contacts, message.chan)
			self.wa_i.send(group, msg)
		    except Exception, e:
			error(traceback.print_exc())
			error("Cannot send message to channel %s: %s" %(message.chan, e))