ze.py

import urllib.request
import urllib.error
import urllib.parse
import plugin
from random import choice
import re


class Plugin(plugin.Plugin):
    def register_commands(self):
        self.commands = [
            ('ze', self.ze),
        ]

    def prepare(self):
        self.url = 'http://srcommunity.org/w/index.php?action=raw&title=%s'

    def get_wikitext(self, title):
        wikitext = urllib.request.urlopen(self.url % (title)).read()
        wikitext = wikitext.decode("utf-8")
        return wikitext

    def reformat(self, line):
        line = line.strip()
        line = re.sub("<.*?>| *\(.*?\)|''|\[[^\]]+?\||[\[\]]+", '', line)

        line = line.lstrip(' :').strip()

        return line

    def get_quote(self):
        ep_index = self.get_wikitext('A_Show_Episode_Guide').split(
            '=== Notes ===')[0]

        lines = ep_index.split('\n')

        eps = [e.split('[[')[-1].split('|')[0].replace(' ', '_')
               for e in lines
               if len(e) > 0
               and '[[Image:Duckie' in e]

        eps = [e for e in eps if e.startswith('A_show:_')]

        ep = choice(eps)

        print(' :: picking from %s' % ep)
        transcript = self.get_wikitext(ep)

        epno = int(transcript.split(
            'http://ashow.zefrank.com/episodes/')[1].split(' ')[0])

        source = 'http://ashow.zefrank.com/episodes/%i' % epno

        # clumsy, but the transcripts are inconsistent
        transcript = transcript.split('Credits:')[0]
        transcript = transcript.split('Links from sidebar:')[0]

        try:
            transcript = transcript.split('|next]]')[1]
        except IndexError:
            transcript = transcript.split(' | next\n')[1]

        lines = transcript.split('\n')
        lines = [self.reformat(l) for l in lines]
        lines = [l for l in lines
                 if len(l) > 0
                 and not l.startswith(('#', '(', '|', '{', '}'))]

        line = choice(lines)

        return (line, source)

    def ze(self, message, args):
        """
        Returns a random quote from the [a show
        transcriptions](http://srcommunity.org/wiki/A_Show_Episode_Guide).
        """

        quote, source = self.get_quote()
        self.irc.privmsg(message.source, quote+'\x03# |\x03 '+source)