comparison V4/synastry.py @ 19:ae220e89cb3a

fixing parse bugs, and angle calculation bugs

 import random
 import csv
 import re
+import requests
+import time
+from bs4 import BeautifulSoup
+
 import pdb
 
 class compatibility:
 	def __init__(self):
 		self.rules = []

 				text.append(line.strip())
 		return text	
 
 	def parseCompatRules(self,filename):
 		comList = self.readFile(filename)
-		# compat = {}
 		for com in comList:
 			c = com.split(',')
-			# print c
 			if len(c) == 4:
 				if len(c[1]) > 1:
 					aspectList = []
 					planetTuple = tuple(sorted((c[0].lower(),c[2].lower())))
-					score = c[3]
+					score = int(c[3])
 					for subset in c[1].split(' '):
 						aspectList.append(subset)
 				elif len(c[1]) == 0 :
 					if 'house' not in c[2]:
 						print 'some error where rule is not house'

 					if len(houseNo) != 1:
 						print 'multiple house numbers found - ERROR'
 					aspectList = int(houseNo[0])
 					score = int(c[3])
 			self.addRule(planetTuple,aspectList,score)
-		# return compat
-		# pdb.set_trace()
 
 	def calcCompatibility(self, horiscope):
 		score = 0
 		for r in self.rules:
 			if 'vertex' in r.planet:

 				if aspect is not None:
 					if aspect in r.aspect:
 						score += r.score
 		return score
 
+	
 
 
 class compatibilityRule:
 	def __init__(self,uniqueId,planetTuple,aspectList,score):
 		self.id = uniqueId

 			self.p_dob = self.p_dob[1:-1].split(',')
 		if type(self.tob) is str:
 			self.tob = self.tob[1:-1].split(',')
 		if type(self.p_tob) is str:
 			self.p_tob = self.p_tob[1:-1].split(',')
-		# pdb.set_trace()
-
-		# print dataDict['pDOB']
-
+		
 		self.payload = {'send_calculation':'1', #Req
 			'muz_narozeni_den':self.dob[0],
 			'muz_narozeni_mesic':self.dob[1],
 			'muz_narozeni_rok':self.dob[2],
 			'muz_narozeni_hodina':self.tob[0],

 			'zena_narozeni_timezone_dst_form':'auto',
 			'switch_interpretations':'0',
 			'house_system':'placidus',
 			'uhel_orbis':'#tabs_redraw'}
 
+	def requestURL(self):
+		self.resp = requests.get(self.url, params=self.payload)
+		time.sleep(5)
+		return self.resp
+
+	def parsePage(self):
+		gotLocation = 0
+		self.horiscope = planetPositions()
+		soup = BeautifulSoup(self.resp.content, 'lxml')
+		tcCell = soup.find_all('div', attrs={'class':'right-sedy-banner-svetlejsi'})
+		for cell in tcCell:
+			if "Planets in partner's house" in cell.get_text():
+				gotLocation = 1
+			divList = cell.find_all('div')
+			for i in range(len(divList)):
+				planetName = divList[i].getText().lower().strip().replace(':','').split(' ')[0]
+				if planetName in planetPositions.planetNames:
+					if gotLocation and not '/' in planetName:
+						self.horiscope.planets[planetName].setHouse(divList[i+2].getText(),divList[i+4].getText())
+					else:
+						self.horiscope.planets[planetName].setLocation(divList[i+2].getText(),divList[i+1].img.attrs['alt'],0)
+						self.horiscope.planets[planetName].setLocation(divList[i+4].getText(),divList[i+3].img.attrs['alt'],1)
+		return self.horiscope
+
+
 
 class planetRelation:
 	noHouseList = ['asc','ic','dsc','mc','asc/mc','sun/moon']
+	zodiacAngle = {'aries':0,'taurus':30,'gemini':60,'cancer':90,'leo':120,'virgo':150,'libra':180,'scorpio':210,'sagittarius':240,'capricorn':270,'aquarius':300,'pisces':330}
+
 	def __init__(self,planetName):
 		self.name = planetName
 		self.angleA = None
 		self.angleB = None
 		if planetName not in planetRelation.noHouseList:
 			self.houseA = None
 			self.houseB = None
 
-	def setLocation(self,A,B):
-		self.angleA = float('.'.join(A.encode('ascii','replace').split('?'))[:-1])
-		self.angleB = float('.'.join(B.encode('ascii','replace').split('?'))[:-1])
-		# print self.angleA,self.angleB
-
+	def setLocation(self,value,sign,isB):
+		signVal = planetRelation.zodiacAngle[sign.lower()]
+		signVal+= float('.'.join(value.encode('ascii','replace').split('?'))[:-1])
+		# print self.name,sign.lower(),planetRelation.zodiacAngle[sign.lower()],value,signVal
+		if not isB:
+			self.angleA = signVal
+		else:
+			self.angleB = signVal
+		
 	def setHouse(self,A,B):
 		self.houseA = int(A.encode('ascii','ignore'))
 		self.houseB = int(B.encode('ascii','ignore'))
-		# print self.houseA,self.houseB
 
 	def test_random(self):
 		self.angleA = random.random()*360
 		self.angleB = random.random()*360
 
 
 class planetPositions:
 	aspectDict = {'conjunction':0,'semi-square':45,'sextile':60,'square':90,'trine':120,'opposite':180}
+	# zodiacAngle = {'aries':0,'taurus':30,'gemini':60,'cancer':90,'leo':120,'virgo':150,'libra':180,'scorpio':210,'sagittarius':240,'capricorn':270,'aquarius':300,'pisces':330}
+	# Taken from https://en.wikipedia.org/wiki/Astrological_sign, with reference from https://en.wikipedia.org/wiki/Astrological_symbols#Signs_of_the_zodiac
 	planetNames = ['sun','moon','mercury','venus','mars','jupiter','saturn','uranus','neptune','pluto','node','lilith','chiron','asc','ic','dsc','mc','asc/mc','sun/moon']
 	def __init__(self):
 		self.planets = {}
 		for p in planetPositions.planetNames:
 			self.planets[p] = planetRelation(p)

 	def test_random(self):
 		for planet in self.planets:
 			self.planets[planet].test_random()
 
 	def calcAngle(self,componentA,componentB):
-		self.angle = max(self.planets[componentA].angleA,self.planets[componentB].angleB) - min(self.planets[componentA].angleA,self.planets[componentB].angleB)
+		self.angle = abs(self.planets[componentA].angleA - self.planets[componentB].angleB)
 		self.angleRange = self.angle-10,self.angle+10
 
 	def calcAspect(self,componentA,componentB):
 		self.calcAngle(componentA,componentB)
+		# print componentA,componentB,self.angle,self.planets[componentA].angleA,self.planets[componentB].angleB
 		for aspect in planetPositions.aspectDict:
 			if self.angleRange[0] < planetPositions.aspectDict[aspect] and self.angleRange[1] > planetPositions.aspectDict[aspect]:
-				aspectDiff = abs(self.angle - planetPositions.aspectDict[aspect])
+				# print aspect#,componentA,componentB,self.angle
+				aspectDiff = round(abs(self.angle - planetPositions.aspectDict[aspect]),2)
 				self.aspect[componentA,componentB] = (aspect,aspectDiff)
 				return aspect,aspectDiff
 		return None,None
 
 	def isInHouse(self,component,houseNo):