comparison V4/runme.py @ 16:b11cff4b7f83

implement compatibility scores

 def savePick(filename,data):
 	with open(filename, 'wb') as handle:
 		pickle.dump(data,handle)
 
 def parseHoriscope(people,saveFile):
-	horiscopeList = []
 	for person in people:
 		issue = person.identifyIssues()
 		if issue is not None:
 			print 'SKIPPING person '+ person.id + ' error with ' + issue 
 		else:

 			person.horiscope = parsePage(person.resp)
 			# person.horiscope.printPositions()
 		if saveFile is not None:
 				savePick(saveFile,people)
 
-def testMain():
-	restartDataFile = 1
-	if(restartDataFile):
-		pickFile = 'outData.pick'
-		# people = makePeople('individuals.csv')
-		# savePick(pickFile,people)
-		people = loadPick(pickFile)
-		parseSaveFile = pickFile.split('.')[0]+'_collect.pick'
-		parseHoriscope(people,parseSaveFile)
-	else:
-		people = loadPick('onlineDatacollect.pick')
-	for p in people:
-		if p.horiscope is None:
-			print p.id	
-		else:
-			p.horiscope.calcAllAspects()
-
-	# horiscopeData = presentResults(parseSaveFile)
-	# comRules = comp.parseCompatDef('compatibilityRules.csv')
-	# applyCompatScore(horiscopeData,rules)
+# def testMain():
+# 	pickFile = 'outData.pick'
+# 	restartDataFile = 0
+# 	if(restartDataFile):
+# 		# people = makePeople('individuals.csv')
+# 		# savePick(pickFile,people)
+# 		people = loadPick(pickFile)
+# 		parseSaveFile = pickFile.split('.')[0]+'_collect.pick'
+# 		parseHoriscope(people,parseSaveFile)
+# 	else:
+# 		people = loadPick('outData_collect.pick')
+# 	comp = syn.compatibility()
+# 	comp.parseCompatRules('compatibilityRules.csv')
+# 	for person in people:
+# 		if person.issue is None:
+# 			person.score = comp.calcCompatibility(person.horiscope)
+# 			if person.score is None:
+# 				person.issue = 'None Planet Locations'
+# 			else:
+# 				print person.id,person.score
+# 	pdb.set_trace()
 
 def _main():
 	pickFile = 'outData.pick'
-	# people = dict()
 	if not os.path.exists(pickFile):
-		print 'reParse file'
 		people = makePeople('individuals.csv')
 		savePick(pickFile,people)
 	else:
-		print 'read in ' + pickFile
 		people = loadPick(pickFile)
 	parseSaveFile = pickFile.split('.')[0]+'_collect.pick'
-	parseHoriscope(people,parseSaveFile)
-	horiscopeData = presentResults(parseSaveFile)
-	comRules = comp.parseCompatDef('compatibilityRules.csv')
-	applyCompatScore(horiscopeData,rules)
+	if not os.path.exists(parseSaveFile):
+		parseHoriscope(people,parseSaveFile)
+	else:
+		people = loadPick(parseSaveFile)
+	comp = syn.compatibility()
+	comp.parseCompatRules('compatibilityRules.csv')
+	for person in people:
+		person.score = None
+		if person.issue is None:
+			person.score = comp.calcCompatibility(person.horiscope)
+			if person.score is None:
+				person.issue = 'None Planet Locations'
+	savePick('fullResults.pick',people)
 
 if __name__ == "__main__":
-	testMain()
+	_main()