neobots/classes/NeoAccount.py

import requests


class NeoAccount:

    d = 'http://www.neopets.com'
    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 6.3; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/36.0.1951.5 Safari/537.36',
        'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
        'Accept-Language': 'en-us,en;q=0.5',
        'Accept-Encoding': 'gzip, deflate'}

    def __init__(self, un, pw, proxy=""):
        self.un = un
        self.pw = pw
        self.proxy = proxy
        self.referrer = None
        self.result = None

        self.session = requests.Session()
        a = requests.adapters.HTTPAdapter(max_retries=3)
        self.session.mount('http://', a)
        self.session.headers = self.headers

        if (self.proxy != ""):
            self.session.proxies = {'http': 'http://' + self.proxy + '/'}

    def get(self, url, referer=''):
        if url[0] == '/':
            url = self.d + url
        if referer != '':
            self.result = self.session.get(
                url, headers={'Referer': referer}, timeout=20)
        else:
            self.result = self.session.get(url, timeout=20)
        return self.result

    def post(self, url, data, referer=''):
        if url[0] == '/':
            url = self.d + url
        if referer != '':
            self.result = self.session.post(
                url, data=data, headers={'Referer': referer}, timeout=20)
        else:
            self.result = self.session.post(url, data=data, timeout=20)
        return self.result

    def login(self):
        self.result = self.session.get(
            'http://www.neopets.com/index.phtml', timeout=60)
        self.result = self.session.post('http://www.neopets.com/login.phtml',
                                        data={
                                            'username': self.un,
                                            'password': self.pw,
                                            'destination': "http://www.neopets.com/index.phtml"}, timeout=60)
        print self.result.url
        if 'badpassword' in self.result.url:
            return False, 'Bad password'
        elif 'hello' in self.result.url:
            return False, 'Birthday locked'
        elif 'login' in self.result.url:
            return False, 'Frozen'
        elif 'index' in self.result.url:
            return True, 'Logged in'