source: trollforge/dsp-trollforge/Library/Libtroll.py @ 249

#!/usr/bin/env python



#     _     ___  _       _____ ____   ___  _     _

#    | |   / _ \| |     |_   _|  _ \ / _ \| |   | |

#    | |  | | | | |       | | | |_) | | | | |   | |

#    | |__| |_| | |___    | | |  _ <| |_| | |___| |___

#    |_____\___/|_____|   |_| |_| \_\\___/|_____|_____|

# 



# TODO: webmail stuff

#  http://www.dodgeit.com/

#  http://spam.la/



# Domains that accept email for @mailinator.com

mailinator = [ 'mailinator.com',

               'fastacura.com',

               'fastchevy.com',

               'fastchrysler.com',

               'fastkawasaki.com',

               'fastmazda.com',

               'fastmitsubishi.com',

               'fastnissan.com',

               'fastsubaru.com',

               'fastsuzuki.com',

               'fasttoyota.com',

               'fastyamaha.com' ]



from urllib import FancyURLopener, urlencode

from re import compile



def sanitize_cookie(cookie):

    str = ''

    regex = compile(' *([^;]*);[^;]*;[^;]*;[^;,]*,*(.*)')

    while True:

        m = regex.match(cookie)

        if not m:

            return str

        str += m.group(1) + '; '

        cookie = m.group(2)



# SO LITTLE SO TINY SO PRETTY SO ZOMFGLOLPWND

class make_troll:

    # HACK CLASS BECAUSE PYTHON FAILS IT AT USING COOKIES

    class urlopener(FancyURLopener):

        _cookie = ''

        def http_error_302(self, *args):

            headers = args[4]

            cookie = headers.get('set-cookie')

            if cookie:

                self._cookie = sanitize_cookie(cookie)

                self.addheaders.append(('Cookie', self._cookie))

            return FancyURLopener.http_error_302(self, *args)



    # Private vars

    _proxy = {}

    _referer = ''

    _uagent = 'Mozilla/4.0'

    _cookie = ''

    _data = {}



    def set_proxy(self, proxy):

        if proxy:

            self._proxy = {'http': proxy}

        else:

            self._proxy = {}



    def set_uagent(self, uagent):

        self._uagent = uagent



    def set_postdata(self, data):

        self._data = data



    def set_cookie(self, cookie):

        self._cookie = cookie



    def get_cookie(self):

        return self._cookie



    def get_url(self, url):

        u = self.urlopener(self._proxy)

        if self._uagent:

            # This needs to be our first addheader call

            u.addheaders = []

            u.addheader('User-agent', self._uagent)

        if self._cookie:

            u.addheader('Cookie', self._cookie)

        if self._referer:

            u.addheader('Referer', self._referer)

        try:

            if self._data:

                f = u.open(url, urlencode(self._data))

            else:

                f = u.open(url)

        except KeyboardInterrupt:

            raise

        except:

            # LOL @ FAILURE

            return False

        gotcookie = []

        if u._cookie:

            self._cookie = u._cookie

            u._cookie = ''

        info = f.info()

        if info.has_key('set-cookie'):

            self._cookie = sanitize_cookie(info['set-cookie'])

        try:

            from operator import add

            text = reduce(add, f.readlines())

        except KeyboardInterrupt:

            raise

        except:

            # LOL @ FAILURE

            return False

        f.close()

        # Update our status

        self._data = {}

        self._referer = url

        return text



    def get_mail(self, address, subject):

        m = compile("(.*)@(.*)").match(address)

        login = m.group(1)

        domain = m.group(2)

        if domain in mailinator:

            return -1 # Disabled for now, they refuse email from IMDb

            url = "http://www.mailinater.com/mailinator/CheckMail.do?email=" + login

            banned = "nefarious, mysterious"

            link = '(http://[^>]*ShowMail.do[^>]*)>[^<]*' + subject + '[^<]*<'

            prefix = ''

        elif domain == "spam.la":

            url = "http://spam.la/?f=" + login

            banned = False

            link = 'href="(.id=[0-9]+)">[^<]*' + subject + '[^<]*<'

            prefix = "http://spam.la/"

        elif domain == "tanya.com":

            url = "http://www.tanya.com/cgi-bin/checkmail.pl?username=" + login + "&button=Check+Mail&terms=on"

            banned = False

            link = 'href="/([^"]*id=[0-9]+)">[^<]*' + subject + '[^<]*<'

            prefix = "http://www.tanya.com/"

        elif domain == "dodgeit.com":

            url = "http://dodgeit.com/run/checkmail?mailbox=" + login

            banned = False

            link = "href='([^']*id=[0-9]+)'>[^<]*" + subject + "[^<]*<"

            prefix = ''

        else:

            return -1

        # Now really get this email

        output = self.get_url(url)

        if not output:

            return -2 # Site unavailable

        if banned and output.find(banned) != -1:

            return -4 # Banned...

        mbox = False

        for m in compile(link).finditer(output):

            mbox = m.group(1)

        if not mbox:

            return -3 # No mail for us

        output = self.get_url(prefix + mbox)

        if not output:

            return -2 # Site unavailable

        if banned and output.find(banned) != -1:

            return -4 # Banned...

        return output