Python IRC Bot Using Twisted (Rollbot)

I wrote my first IRC bot recently. It turned out to be a simpler task than I thought it would be.

Here’s the back story: At work, our programming team uses an IRC channel for all our internal communication throughout the day. What more would you expect from a bunch of programmers?

Recently I’ve been taking delight in Rebecca Rolling (worse than being Rick Roll’d) every each Friday. Someone made a comment that it would be nice to have a bot to detect such things. I had been looking for an excuse to try and write an IRC bot, so I took the challenge.

Basically, the majority of the script was created following Eric Florenzano’s blog post, so I won’t cover all that again. Just go check it out yourself.

Basically, the part I modified was the privmsg function:

    def privmsg(self, user, channel, msg):
        # use regex to find posted URLs
        matches = re.findall(r'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', msg)
        if matches:
            for url in matches:
                u = urllib.urlopen(url)
                # get the mime type
                urltype = u.headers.gettype()
                #print urltype
                    # use BeautifulSoup to get the page title
                    soup = BeautifulSoup.BeautifulSoup(u)
                    title = re.sub("\s+", ' ', soup.title.string).strip()
                    self.msg(, "Title: %s" % str(title))
                except (AttributeError, HTMLParseError):
                    # if we have an error getting the title, show the mime type
                    self.msg(, \
                        "NO TITLE FOUND (%s)" % urltype)

Comments in the code above should explain what is going on. If you want the full source to the Rollbot, check it out on github.

Hackers: Note that it is entirely possible to hack this. Just create a basic html page, set your own title, and redirect with JavaScript. My co-workers still don’t trust me…