import httplib, sys
import myparser

class search_google_labs:
	def __init__(self,list):
		self.results=""
		self.totalresults=""
		self.server="labs.google.com"
		self.hostname="labs.google.com"
		self.userAgent="(Mozilla/5.0 (Windows; U; Windows NT 6.0;en-US; rv:1.9.2) Gecko/20100115 Firefox/3.6"
		id=0
		self.set=""
		for x in list:
			id+=1
			if id==1:
				self.set=self.set+"q"+str(id)+"="+str(x)
			else:
				self.set=self.set+"&q"+str(id)+"="+str(x)
		
	def do_search(self):
		h = httplib.HTTP(self.server)
		h.putrequest('GET', "/sets?hl=en&"+self.set)
		h.putheader('Host', self.hostname)
		h.putheader('User-agent', self.userAgent)	
		h.endheaders()
		returncode, returnmsg, headers = h.getreply()
		self.results = h.getfile().read()
		self.totalresults+= self.results

	def get_set(self):
		rawres=myparser.parser(self.totalresults,list)
		return rawres.set()
	
	def process(self):
		self.do_search()
				
