zillow.py for Ch7 Programming Collective Intelligence

""" Helped from here: https://github.com/nico/collectiveintelligence-book """

import xml.dom.minidom
import urllib2 as urllib2

zwskey = "YOUR_APIKEY_HERE"

def getaddressdata(address, city):
  escad = address.replace(' ', '+')

  url = 'http://www.zillow.com/webservice/GetDeepSearchResults.htm?'
  url += 'zws-id=%s&address=%s&citystatezip=%s' % (zwskey, escad, city)

  doc = xml.dom.minidom.parseString(urllib2.urlopen(url).read())
  code = doc.getElementsByTagName('code')[0].firstChild.data
  if code != '0': return None

  try:
      zipcode = doc.getElementsByTagName('zipcode')[0].firstChild.data
      use = doc.getElementsByTagName('useCode')[0].firstChild.data
      year = doc.getElementsByTagName('yearBuilt')[0].firstChild.data
      bath = doc.getElementsByTagName('bathrooms')[0].firstChild.data
      bed = doc.getElementsByTagName('bedrooms')[0].firstChild.data
      #rooms = doc.getElementsByTagName('totalRooms')[0].firstChild.data
      price = doc.getElementsByTagName('amount')[0].firstChild.data
  except:
      return None

  return zipcode, use, int(year), float(bath), int(bed), price

def getpricelist():
    l1=[]
    for line in file('addresslist.txt'):
        data=getaddressdata(line.strip(),'Irvington,NJ')
        if data==None: pass
        else: l1.append(data)
        print data
    return l1

if __name__ == '__main__':
    import treepredict as treepredict
    housedata = getpricelist()
    tree = treepredict.buildtree(housedata, scoref=treepredict.variance)
    treepredict.drawtree(tree, 'housetree.jpg')
    print "Created housetree.jpg"

"""
 example api call http://www.zillow.com/webservice/GetDeepSearchResults.htm?zws-id=YOUR_APIKEY_HERE&address=22+Franklin+Ter&citystatezip=Irvington+NJ
"""