GlobalNeighbors: globalneighbors/distance.py comparison

we have a data dump

comparison

equal deleted inserted replaced

-:22c384fe954d
+:e69cb496324e
 k)
 return neighbors
+def write_neighbors(fp, neighbors):
+for key, value in neighbors.iteritems():
+fp.write("{key} {value}\n".format(key=key,
+value=json.dumps(value)))
 def main(args=sys.argv[1:]):
 """CLI"""
 # parse command line arguments
 description = """write nearest neighborfiles"""
 parser.add_argument('-k', dest='k',
 type=int, default=50,
 help="number of neighbors to determine [DEFAULT: %(default)s]")
 options = parser.parse_args(args)
-# parse cities
-cities = list(read_cities(options.cities, fields=fields))
 # get locations
-city_locations = locations(cities)
+city_locations = locations(read_cities(options.cities, fields=fields))
+options.cities.close()
+options.output.close()
 # calculate neighbors
 neighbors = calculate_neighbors(city_locations,
 k=options.k,
 lat_tol=options.latlon[0],
 output=options.output_counter)
 # output
 print ("Outputting neighbors")
 sys.stdout.flush()
-options.output.write(json.dumps(neighbors))
+import pdb; pdb.set_trace()
+with open(options.output.name, 'w') as f:
+f.write(json.dumps(neighbors))
 if __name__ == '__main__':
 main()

Mercurial > hg > GlobalNeighbors