Example #1
0
    def WorkBook_getArray(self):
        with open(self.filename) as f:
            prisonList = list(csv.DictReader(f))

        prisonArr = []

        print "Reading in data..."
        count = 0
        for prison in prisonList:
            prisonAddress = Address(prison["facility_address1"], 
                prison["city"], prison["state"], prison["zip"][0:5])

            curPrison = Prison(prison["type"], prison["facility_name"],
                prisonAddress)

            # Backup address check if the previous check (by street/
            # address) failed: searches by prison name
            if not prisonAddress.hasAddress:
                prisonAddress.hasAddress = curPrison.Prison_getLocation()

            if not prisonAddress.hasAddress:
                sys.stderr.write("Query overflow: Please try again later\n")
                return None
            prisonArr.append(curPrison)
            
            #Gives status update of reading in the data
            count += 1
            if count % 10 == 0:
                print "Finished reading in {} prisons...".format(count)

        print "Finished reading in prisons"
        self.prisonArr = prisonArr