def load_data(self):
self.item_categories = products.load_products_json()
self.zipcode_objs = load_zipcode_data()
self.first_names, self.last_names = load_names()
name = cols[0]
weight = float(cols[5])
if cols[4] == "1":
first_names.append((name, weight))
if cols[3] == "1":
last_names.append((name, weight))
name_fl.close()
return first_names, last_names
if __name__ == "__main__":
from zipcodes import load_zipcode_data
from stores import StoreGenerator
print "Loading zipcode data..."
zipcode_objs = load_zipcode_data()
print
print "Generating Stores..."
generator = StoreGenerator(zipcode_objs=zipcode_objs)
stores = generator.generate(n=100)
print
print "Generating customers..."
generator = CustomerGenerator(zipcode_objs=zipcode_objs, stores=stores)
customers = generator.generate(10)
print
print "Customers"
for c in customers:
print c
self.zipcode_objs = zipcode_objs
self.zipcode_sampler = ZipcodeSampler(zipcode_objs=zipcode_objs,
income_scaling_factor=income_scaling_factor)
self.current_id = 0
def generate(self, n):
stores = list()
for i in xrange(n):
store = Store()
store.id = self.current_id
self.current_id += 1
store.name = "Store_" + str(i)
store.zipcode = self.zipcode_sampler.sample()
store.coords = self.zipcode_objs[store.zipcode].coords
stores.append(store)
return stores
if __name__ == "__main__":
from zipcodes import load_zipcode_data
zipcode_objs = load_zipcode_data()
generator = StoreGenerator(zipcode_objs=zipcode_objs,
income_scaling_factor=100.0)
stores = generator.generate(100)
for s in stores:
print s
