DaveM@4: #!/usr/bin/env python DaveM@4: DaveM@4: import csv DaveM@4: import time DaveM@4: import unicodedata DaveM@4: from geopy.geocoders import Nominatim DaveM@4: from geopy.exc import GeocoderTimedOut DaveM@6: import random DaveM@4: import pdb DaveM@4: DaveM@4: DEFAULT_TIME_H = 12 DaveM@4: DEFAULT_TIME_M = 00 DaveM@4: DEAULT_LOCATION = 'USA' DaveM@4: DaveM@4: def parseCSV(filename): DaveM@4: stream = csv.DictReader(open(filename,'rb')) DaveM@4: dictList = [] DaveM@4: for line in stream: DaveM@4: dictList.append(regulateData(line)) DaveM@4: return dictList DaveM@4: DaveM@4: def regulateData(dataDict): DaveM@4: print("Parse %s"%(str(dataDict['ID']))) DaveM@4: p_DOBQ = "What is your partner's date of birth? Please use the format DD/MM/YYYY (for example, 29/03/1981)." DaveM@4: p_TOBQ = "At what exact time were your partner born? Please use the format HHMM (for example, 2204)." DaveM@4: DOB_DQ = "Which day (numeric) have you been born?" DaveM@4: DOB_MQ = "Which month have you been born?" DaveM@4: DOB_YQ = "Year Of Birth" DaveM@4: TOB_Q = "At what exact time were you born? Please use the format HHMM (for example, 2204)." DaveM@4: COB = "What is your place of birth? Please specify city and country (for example, San Francisco, USA)." DaveM@4: p_COB = "What is your partner's place of birth? Please specify city and country (for example, San Francisco, USA)." DaveM@4: dataDict['DOB'] = parseDOB(dataDict[DOB_DQ],dataDict[DOB_MQ],dataDict[DOB_YQ]) DaveM@4: # print (dataDict[DOB_DQ],dataDict[DOB_MQ],dataDict[DOB_YQ]) DaveM@4: # print dataDict['DOB'] DaveM@4: dataDict['TOB'] = parseTOB(dataDict[TOB_Q]) DaveM@4: dataDict['pDOB'] = parsePartnerDOB(dataDict[p_DOBQ]) DaveM@4: dataDict['pTOB'] = parseTOB(dataDict[p_TOBQ]) DaveM@6: # MAKE RANDOM PLACE DaveM@6: # dataDict['COB'] = (random.uniform(-90, 90),random.uniform(-90, 90)) DaveM@6: # dataDict['pCOB'] = (random.uniform(-90, 90),random.uniform(-90, 90)) DaveM@6: dataDict['COB'] = parseBirthTown(dataDict[COB]) DaveM@6: dataDict['pCOB'] = parseBirthTown(dataDict[p_COB]) DaveM@4: return dataDict DaveM@4: DaveM@4: DaveM@4: def parseBirthTown(s): DaveM@4: try: DaveM@4: s = s.encode('ascii') DaveM@4: except UnicodeDecodeError: DaveM@4: # pdb.set_trace() DaveM@4: s = s.decode('latin-1') DaveM@4: # s = unicodedata.normalize('NFKD',s.decode('utf-8')).encode('ascii','ignore') DaveM@4: timeoutTime = 2 DaveM@4: geolocator = Nominatim(timeout=timeoutTime) DaveM@4: while s is not [] and timeoutTime < 60: DaveM@4: try: DaveM@4: location = geolocator.geocode(s) DaveM@4: if location is not None: DaveM@4: # print(location.raw) DaveM@4: # print (location.latitude, location.longitude) DaveM@4: return (location.latitude, location.longitude, location.raw) DaveM@4: else: DaveM@4: s = s.split(' ',1) DaveM@4: if len(s) == 2: DaveM@4: s = s[1] DaveM@4: # print s DaveM@4: else: DaveM@4: s = DEAULT_LOCATION DaveM@6: except: DaveM@4: timeoutTime += 1 DaveM@6: print("Error: geocode failed on input %s, incrementing timeout time to %d"%(s,timeoutTime)) DaveM@4: time.sleep(5) DaveM@4: geolocator = Nominatim(timeout=timeoutTime) DaveM@4: # places = geograpy.get_place_context(text=s) DaveM@4: DaveM@3: def parsePartnerDOB(dob): DaveM@6: # print dob DaveM@6: # pdb.set_trace() DaveM@4: dob = dob.strip() DaveM@4: if(dob.count('-') == 2): DaveM@4: dob = dob.replace('-','/') DaveM@4: if(dob.count(' ') == 2): DaveM@4: dob = dob.replace(' ','/') DaveM@3: dob_ = dob.split('/') DaveM@3: if(len(dob_) != 3): DaveM@4: dob = dob.replace('/','').strip() DaveM@4: dob_ = [] DaveM@4: # print dob DaveM@4: if len(dob) == 8: # ddmmyyyy DaveM@4: dob_.append(dob[:2]) DaveM@6: dob_.append(dob[2:4]) DaveM@4: dob_.append(dob[4:]) DaveM@4: elif len(dob) == 7 and dob[1] == '1' and (dob[2] == '0' or dob[2] == '1' or dob[2] == '2'): # dmmyyyy DaveM@4: dob_.append(dob[0]) DaveM@6: dob_.append(dob[1:3]) DaveM@6: dob_.append(dob[3:]) DaveM@6: elif(len(dob) == 7): DaveM@6: if int(dob[:2]) > 31:# dmmyyyy DaveM@6: dob_.append(dob[0]) DaveM@6: dob_.append(dob[1:3]) DaveM@6: dob_.append(dob[3:]) DaveM@4: elif len(dob) == 7: # ddmyyyy DaveM@6: dob_.append(dob[0:2]) DaveM@4: dob_.append(dob[2]) DaveM@6: dob_.append(dob[3:]) DaveM@4: elif len(dob) == 6 and dob[3:4] != '19': # ddmmyy DaveM@4: dob_.append(dob[:2]) DaveM@6: dob_.append(dob[2:4]) DaveM@6: dob_.append(dob[4:]) DaveM@4: elif len(dob) == 5 and dob[1] == '1' and (dob[2] == '0' or dob[2] == '1' or dob[2] == '2'): # dmmyy DaveM@4: dob_.append(dob[0]) DaveM@6: dob_.append(dob[1:3]) DaveM@6: dob_.append(dob[3:]) DaveM@4: elif len(dob) == 5: # ddmyy DaveM@4: dob_.append(dob[:2]) DaveM@4: dob_.append(dob[2]) DaveM@6: dob_.append(dob[3:]) DaveM@4: elif len(dob) == 4: # dmyy DaveM@4: dob_.append(dob[0]) DaveM@4: dob_.append(dob[1]) DaveM@4: dob_.append(dob[2:]) DaveM@4: else: DaveM@4: if(len(dob) < 4): DaveM@4: return None DaveM@6: # print dob DaveM@4: # print filter(lambda x: x.isdigit(),dob) DaveM@4: print 'no / partnerDOB issue' DaveM@3: # deal with no /'s DaveM@3: try: DaveM@4: d = int(filter(lambda x: x.isdigit(),dob_[0])) DaveM@4: m = int(filter(lambda x: x.isdigit(),dob_[1])) DaveM@4: y = int(filter(lambda x: x.isdigit(),dob_[2])) DaveM@3: if y < 100: DaveM@3: y = y + 1900 DaveM@6: if (m > 12 and d <= 12): DaveM@6: temp = d DaveM@6: d = m DaveM@6: m = temp DaveM@6: if(d > 31 or d < 1 or m > 12 or m < 1 or y > 2017 or y < 1900): DaveM@3: print 'error with DOB '+d+'/'+m+'/'+y DaveM@6: pdb.set_trace() DaveM@3: except TypeError: DaveM@3: return None DaveM@6: # print (d,m,y) DaveM@3: return (d,m,y) DaveM@3: DaveM@3: def monthStringToNum(s): DaveM@4: # print 'inMonthStringToNum' DaveM@4: m = {'jan':1,'feb':2,'mar':3, DaveM@4: 'apr':4,'may':5,'jun':6,'jul':7,'aug':8, DaveM@4: 'sep':9,'oct':10,'nov':11,'dec':12} DaveM@4: s_ = s.strip()[:3].lower() DaveM@3: try: DaveM@4: out = m[s_] DaveM@3: return out DaveM@3: except: DaveM@3: raise ValueError('Not a month') DaveM@3: DaveM@4: def checkMonthDay(d,m): DaveM@4: if d > 31: # take first two digits of day DaveM@4: d = int(str(d)[:2]) DaveM@4: if d > 31: DaveM@4: d = int(str(d)[1]) DaveM@4: if m > 12 and d < 12: # Day and month wrong way round - American DaveM@4: temp = m DaveM@4: m = d DaveM@4: d = temp DaveM@4: if(m == 2): DaveM@4: if d <= 29: DaveM@4: return (True,d,m) DaveM@4: else: DaveM@4: return (False,d,m) DaveM@4: elif m in [4,6,9,11]: DaveM@4: if d <= 30: DaveM@4: return (True,d,m) DaveM@4: else: DaveM@4: return (False,d,m) DaveM@4: elif m <= 12 and d <= 31: DaveM@4: return (True,d,m) DaveM@4: else: DaveM@4: return (False,d,m) DaveM@4: DaveM@3: def parseDOB(d,m,y): DaveM@4: d = int(filter(lambda x: x.isdigit(),d)) DaveM@4: y = int(filter(lambda x: x.isdigit(),y)) DaveM@3: try: DaveM@4: # print m DaveM@3: m = monthStringToNum(m.strip()) DaveM@3: except ValueError: DaveM@3: m = int(m.strip()) DaveM@3: if(y < 100): DaveM@3: y = y + 1900 DaveM@4: (r,d,m) = checkMonthDay(d,m) DaveM@4: if not r: DaveM@4: print 'error with day month' DaveM@4: print (r,d,m) DaveM@3: return (d,m,y) DaveM@3: DaveM@3: def parseTOB(T): DaveM@4: # pdb.set_trace() DaveM@4: timeFlag = None DaveM@4: T_ = T.replace('.','').lower().strip() DaveM@4: if 'am' in T_: DaveM@4: timeFlag = 0 DaveM@4: T = T_.replace('am','') DaveM@4: if 'pm' in T_: DaveM@4: timeFlag = 1 DaveM@4: T = T_.replace('pm','') DaveM@4: T = T.strip() DaveM@4: if T.count('.') == 1: DaveM@4: T = T.replace('.',':') DaveM@3: try: DaveM@3: if ':' in T: DaveM@4: T_ = T.split(':') DaveM@6: DaveM@4: H = int(T_[0]) DaveM@4: M = int(T_[1]) DaveM@3: else: DaveM@4: if len(T) == 4: DaveM@4: H = int(T[:2]) DaveM@4: M = int(T[2:]) DaveM@4: elif int(T) <= 24 : DaveM@4: H = int(T) DaveM@3: M = 0 DaveM@4: elif int(T) > 100: DaveM@4: H = int(T)/100 DaveM@4: M = int(T)%100 DaveM@3: if timeFlag is not None: DaveM@3: if timeFlag == 0: DaveM@3: H = H%12 DaveM@3: else: DaveM@3: H = H%12 + 12 DaveM@3: except ValueError: DaveM@4: H = DEFAULT_TIME_H DaveM@4: M = DEFAULT_TIME_M DaveM@3: return (H,M) DaveM@3: DaveM@6: def makePayload(dataDict): DaveM@6: if type(dataDict['COB']) is str: DaveM@6: cob_0 = float(dataDict['COB'].split(',')[0][1:]) DaveM@6: cob_1 = float(dataDict['COB'].split(',')[1]) DaveM@6: dataDict['COB'] = (cob_0,cob_1) DaveM@6: if type(dataDict['pCOB']) is str: DaveM@6: pcob_0 = float(dataDict['pCOB'].split(',')[0][1:]) DaveM@6: pcob_1 = float(dataDict['pCOB'].split(',')[1]) DaveM@6: dataDict['pCOB'] = (pcob_0,pcob_1) DaveM@6: if type(dataDict['DOB']) is str: DaveM@6: dataDict['DOB'] = dataDict['DOB'][1:-1].split(',') DaveM@6: if type(dataDict['pDOB']) is str: DaveM@6: dataDict['pDOB'] = dataDict['pDOB'][1:-1].split(',') DaveM@6: if type(dataDict['TOB']) is str: DaveM@6: dataDict['TOB'] = dataDict['TOB'][1:-1].split(',') DaveM@6: if type(dataDict['pTOB']) is str: DaveM@6: dataDict['pTOB'] = dataDict['pTOB'][1:-1].split(',') DaveM@6: # pdb.set_trace() DaveM@3: DaveM@6: print dataDict['pDOB'] DaveM@4: DaveM@6: R = {'send_calculation':'1', #Req DaveM@6: 'muz_narozeni_den':dataDict['DOB'][0], DaveM@6: 'muz_narozeni_mesic':dataDict['DOB'][1], DaveM@6: 'muz_narozeni_rok':dataDict['DOB'][2], DaveM@6: 'muz_narozeni_hodina':dataDict['TOB'][0], DaveM@6: 'muz_narozeni_minuta':dataDict['TOB'][1], DaveM@6: 'muz_narozeni_city':'', DaveM@6: 'muz_narozeni_mesto_hidden':'Manually+place%3A+%C2%B0%27N%2C+%C2%B0%27E',#auto DaveM@6: 'muz_narozeni_stat_hidden':'XX', DaveM@6: 'muz_narozeni_podstat_kratky_hidden':'', DaveM@6: 'muz_narozeni_podstat_hidden':'', DaveM@6: 'muz_narozeni_podstat2_kratky_hidden':'', DaveM@6: 'muz_narozeni_podstat3_kratky_hidden':'', DaveM@6: 'muz_narozeni_input_hidden':'', DaveM@6: 'muz_narozeni_sirka_stupne':str(abs(dataDict['COB'][0])).split('.')[0], DaveM@6: 'muz_narozeni_sirka_minuty':str(float('0.'+str(dataDict['COB'][0]).split('.')[1])*60).split('.')[0], DaveM@6: 'muz_narozeni_sirka_smer': '1' if dataDict['COB'][0]<0 else '0', #address N Dir (0':'N',1':'S) DaveM@6: 'muz_narozeni_delka_stupne':str(abs(dataDict['COB'][1])).split('.')[0], #address E - Main DaveM@6: 'muz_narozeni_delka_minuty':str(float('0.'+str(dataDict['COB'][1]).split('.')[1])*60).split('.')[0], DaveM@6: 'muz_narozeni_delka_smer': '1' if dataDict['COB'][1]<0 else '0', #address E Dir (0':'E',1':'W) DaveM@6: 'muz_narozeni_timezone_form':'auto', DaveM@6: 'muz_narozeni_timezone_dst_form':'auto', DaveM@6: 'send_calculation':'1', DaveM@6: 'zena_narozeni_den':dataDict['pDOB'][0], DaveM@6: 'zena_narozeni_mesic':dataDict['pDOB'][1], DaveM@6: 'zena_narozeni_rok':dataDict['pDOB'][2], DaveM@6: 'zena_narozeni_hodina':dataDict['pTOB'][0], DaveM@6: 'zena_narozeni_minuta':dataDict['pTOB'][1], DaveM@6: 'zena_narozeni_city':'', DaveM@6: 'zena_narozeni_mesto_hidden':'Manually+place%3A+%C2%B0%27N%2C+%C2%B0%27E', DaveM@6: 'zena_narozeni_stat_hidden':'XX', DaveM@6: 'zena_narozeni_podstat_kratky_hidden':'', DaveM@6: 'zena_narozeni_podstat_hidden':'', DaveM@6: 'zena_narozeni_podstat2_kratky_hidden':'', DaveM@6: 'zena_narozeni_podstat3_kratky_hidden':'', DaveM@6: 'zena_narozeni_input_hidden':'', DaveM@6: 'zena_narozeni_sirka_stupne':str(abs(dataDict['pCOB'][0])).split('.')[0], DaveM@6: 'zena_narozeni_sirka_minuty':str(float('0.'+str(dataDict['pCOB'][0]).split('.')[1])*60).split('.')[0], DaveM@6: 'zena_narozeni_sirka_smer': '1' if dataDict['pCOB'][0]<0 else '0', DaveM@6: 'zena_narozeni_delka_stupne':str(abs(dataDict['pCOB'][1])).split('.')[0], DaveM@6: 'zena_narozeni_delka_minuty':str(float('0.'+str(dataDict['pCOB'][1]).split('.')[1])*60).split('.')[0], DaveM@6: 'zena_narozeni_delka_smer': '1' if dataDict['pCOB'][1]<0 else '0', DaveM@6: 'zena_narozeni_timezone_form':'auto', DaveM@6: 'zena_narozeni_timezone_dst_form':'auto', DaveM@6: 'switch_interpretations':'0', DaveM@6: 'house_system':'placidus', DaveM@6: 'uhel_orbis':'#tabs_redraw'} DaveM@6: return R DaveM@6: DaveM@6: DaveM@6: DaveM@6: