comparison runme.py @ 0:479b128cc52c

make python files original
author DaveM
date Fri, 08 Dec 2017 10:20:44 +0000
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:479b128cc52c
1 #!/usr/bin/env python
2 import requests
3 import re
4 import time
5 import csv
6 import random
7 from HTMLParser import HTMLParser
8 # from lxml import html
9 from bs4 import BeautifulSoup
10
11 def parsePage(resp):
12 soup = BeautifulSoup(resp.content, 'lxml')
13 tcCell = soup.find_all('div', attrs={'class':'tc'})
14 for cell in tcCell:
15 person = dict()
16 tableCell = cell.find_all('td');
17 if len(tableCell) > 2:
18 C = tableCell[0].strong.contents[0].encode('utf-8')
19 D = tableCell[2].strong.contents[0].encode('utf-8')
20 print (C,D)
21 A = re.search("\/>(.*)<br/>.*\(([0-9]*)\\xc2\\xb0([0-9]*)(.*)\)",str(tableCell[1]))
22 # A0 = A.group(1)
23 # A1 = A.group(2).split('\xc2\xb0')[0]
24 # A2 = A.group(2).split('\xc2\xb0')[1].split('\xe2')[0]
25 print (A.group(1),A.group(2),A.group(3))
26 person[(C,D)] = (A.group(1),A.group(2),A.group(3))
27 return person
28
29
30 def setURL(p):
31 ## For some reason we need to post men first then women.
32 # 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"
33 # payload = {'muz_narozeni_den':'1','muz_narozeni_mesic':'1','muz_narozeni_rok':'1970'}
34 url = 'https://horoscopes.astro-seek.com/calculate-love-compatibility/'
35 mDay = random.randint(1,29)
36 mMonth = random.randint(1,12)
37 mYear = random.randint(1,100)+1917
38 fDay = random.randint(1,29)
39 fMonth = random.randint(1,12)
40 fYear = random.randint(1,100)+1917
41
42 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}
43 return (url,payload)
44
45 def requestURL(url,payload):
46 r = requests.get(url, params=payload)
47 time.sleep(5)
48 return r
49
50 def parseCSV(filename):
51 stream = csv.DictReader(open(filename,'rb'))
52 dictList = []
53 for line in stream:
54 dictList.append(line)
55 return dictList
56
57 def printToFile(filename,data):
58 keys = data[0].keys()
59 with open(filename,'w') as stream:
60 dict_writer = csv.DictWriter(stream, keys)
61 dict_writer.writeheader()
62 dict_writer.writerows(data)
63
64 def main():
65 # people = dict()
66 people = parseCSV('individuals.csv')
67 horiscopeList = []
68 # people = [1,2,3,4,5]
69 for person in people:
70 print 'parsing person '+ person['ID']
71 url,payload = setURL('')
72 resp = requestURL(url,payload)
73
74 person['horiscope'] = parsePage(resp)
75 horiscopeList.append(person)
76 print horiscopeList
77 printToFile('outputdata.csv',horiscopeList)
78
79 if __name__ == "__main__":
80 main()