import urllib
import urllib2
import cookielib

class TimedChal:
    """Challenge Class used to interact with hacking challenge sites"""
    def __init__(self,user,pas):
        self.user_name = user
        self.user_pass = pas
        self.login_data = { 'user_name':self.user_name,
                            'user_pass':self.user_pass,
                            'login':'Login' }

    def Opener(self,ref):
        """Creats an opener to store cookies,
        and keep a referer to the site
        Added user-agent to spoof browser"""
        self.refrence = ref
        cj = cookielib.CookieJar()
        self.opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cj))
        self.opener.addheaders.append(('User-agent', 'Mozilla/4.0'))
        self.opener.addheaders.append(('Referer',ref))
        return self.opener

    def GET(self,opnr,url):
        """GET method, notice to data option"""
        get_req = opnr.open(url)
        return get_req.read()

    def POST(self,opnr,url,data):
        """data is a dictinary type like login_data"""
        enData = urllib.urlencode(data)
        get_req = opnr.open(url,enData)
        return get_req.read()