DaveM@0: #!/usr/bin/env python
DaveM@0: import requests
DaveM@0: import re
DaveM@0: import time
DaveM@0: import csv
DaveM@0: import random
DaveM@0: from HTMLParser import HTMLParser
DaveM@0: # from lxml import html
DaveM@0: from bs4 import BeautifulSoup
DaveM@0: 
DaveM@0: def parsePage(resp):
DaveM@0: 	soup = BeautifulSoup(resp.content, 'lxml')
DaveM@0: 	tcCell = soup.find_all('div', attrs={'class':'tc'})
DaveM@0: 	for cell in tcCell:
DaveM@0: 		person = dict()
DaveM@0: 		tableCell = cell.find_all('td');
DaveM@0: 		if len(tableCell) > 2:
DaveM@0: 			C = tableCell[0].strong.contents[0].encode('utf-8')
DaveM@0: 			D = tableCell[2].strong.contents[0].encode('utf-8')
DaveM@0: 			print (C,D)
DaveM@0: 			A = re.search("\/>(.*)<br/>.*\(([0-9]*)\\xc2\\xb0([0-9]*)(.*)\)",str(tableCell[1]))
DaveM@0: 			# A0 = A.group(1)
DaveM@0: 			# A1 = A.group(2).split('\xc2\xb0')[0]
DaveM@0: 			# A2 = A.group(2).split('\xc2\xb0')[1].split('\xe2')[0]
DaveM@0: 			print (A.group(1),A.group(2),A.group(3))
DaveM@0: 			person[(C,D)] = (A.group(1),A.group(2),A.group(3))
DaveM@0: 		return person
DaveM@0: 
DaveM@0: 
DaveM@0: def setURL(p):
DaveM@2: 	""" 
DaveM@2: 	Code impacting factors into URL
DaveM@2: 	IMPACTING FACTORS
DaveM@2: 	Date of Birth
DaveM@2: 	Birth Time
DaveM@2: 	Country of birth
DaveM@2: 	City of birth (And state of birth)
DaveM@2: 	"""
DaveM@0: 	## For some reason we need to post men first then women.
DaveM@0: 	# url = "https://horoscopes.astro-seek.com/calculate-love-compatibility/?send_calculation=1&muz_narozeni_den=1&muz_narozeni_mesic=1&muz_narozeni_rok=1970&muz_narozeni_hodina=00&muz_narozeni_minuta=00&muz_narozeni_city=London%2C+United+Kingdom&muz_narozeni_mesto_hidden=London&muz_narozeni_stat_hidden=GB&muz_narozeni_podstat_kratky_hidden=England&muz_narozeni_podstat_hidden=England&muz_narozeni_podstat2_kratky_hidden=Greater+London&muz_narozeni_podstat3_kratky_hidden=undefined&muz_narozeni_input_hidden=&muz_narozeni_sirka_stupne=51&muz_narozeni_sirka_minuty=30&muz_narozeni_sirka_smer=0&muz_narozeni_delka_stupne=0&muz_narozeni_delka_minuty=8&muz_narozeni_delka_smer=1&muz_narozeni_timezone_form=auto&muz_narozeni_timezone_dst_form=auto&send_calculation=1&zena_narozeni_den=1&zena_narozeni_mesic=1&zena_narozeni_rok=1970&zena_narozeni_hodina=00&zena_narozeni_minuta=00&zena_narozeni_city=Berlin%2C+Germany&zena_narozeni_mesto_hidden=Berlin&zena_narozeni_stat_hidden=DE&zena_narozeni_podstat_kratky_hidden=Berlin&zena_narozeni_podstat_hidden=Berlin&zena_narozeni_podstat2_kratky_hidden=undefined&zena_narozeni_podstat3_kratky_hidden=undefined&zena_narozeni_input_hidden=&zena_narozeni_sirka_stupne=52&zena_narozeni_sirka_minuty=31&zena_narozeni_sirka_smer=0&zena_narozeni_delka_stupne=13&zena_narozeni_delka_minuty=24&zena_narozeni_delka_smer=0&zena_narozeni_timezone_form=auto&zena_narozeni_timezone_dst_form=auto&switch_interpretations=0&house_system=placidus&uhel_orbis=#tabs_redraw"
DaveM@0: 	# payload = {'muz_narozeni_den':'1','muz_narozeni_mesic':'1','muz_narozeni_rok':'1970'}
DaveM@0: 	url = 'https://horoscopes.astro-seek.com/calculate-love-compatibility/'
DaveM@0: 	mDay = random.randint(1,29)
DaveM@0: 	mMonth = random.randint(1,12)
DaveM@0: 	mYear = random.randint(1,100)+1917
DaveM@0: 	fDay = random.randint(1,29)
DaveM@0: 	fMonth = random.randint(1,12)
DaveM@0: 	fYear = random.randint(1,100)+1917
DaveM@0: 
DaveM@0: 	payload = {'?send_calculation':'1','muz_narozeni_den':mDay,'muz_narozeni_mesic':mMonth,'muz_narozeni_rok':mYear,'zena_narozeni_den':fDay,'zena_narozeni_mesic':fMonth,'zena_narozeni_rok':fYear}
DaveM@0: 	return (url,payload)
DaveM@0: 
DaveM@0: def requestURL(url,payload):
DaveM@0: 	r = requests.get(url, params=payload)
DaveM@0: 	time.sleep(5)
DaveM@0: 	return r
DaveM@0: 
DaveM@0: def parseCSV(filename):
DaveM@0: 	stream = csv.DictReader(open(filename,'rb'))
DaveM@0: 	dictList = []
DaveM@0: 	for line in stream:
DaveM@0: 		dictList.append(line)
DaveM@0: 	return dictList
DaveM@0: 
DaveM@0: def printToFile(filename,data):
DaveM@0: 	keys = data[0].keys()
DaveM@0: 	with open(filename,'w') as stream:
DaveM@0: 		dict_writer = csv.DictWriter(stream, keys)
DaveM@0: 		dict_writer.writeheader()
DaveM@0: 		dict_writer.writerows(data)
DaveM@0: 
DaveM@0: def main():
DaveM@0: 	# people = dict()
DaveM@0: 	people = parseCSV('individuals.csv')
DaveM@0: 	horiscopeList = []
DaveM@0: 	# people = [1,2,3,4,5]
DaveM@0: 	for person in people:
DaveM@0: 		print 'parsing person '+ person['ID']
DaveM@0: 		url,payload = setURL('')
DaveM@0: 		resp = requestURL(url,payload)
DaveM@0: 		
DaveM@0: 		person['horiscope'] = parsePage(resp)
DaveM@0: 		horiscopeList.append(person)
DaveM@0: 	print horiscopeList
DaveM@0: 	printToFile('outputdata.csv',horiscopeList)
DaveM@0: 
DaveM@0: if __name__ == "__main__":
DaveM@0: 	main()