def items_raw(count):
results = []
items_count = math.floor(count * 0.5)
if (items_count > max_items_stuff):
items_count = max_items_stuff
beers_count = count - items_count
items = csv.parse_file(fpath_items, items_count, filtr_item)
for item in items:
item["type"] = "food"
results.append(item)
beers = csv.parse_file(fpath_beers, beers_count, filtr_beer)
for beer in beers:
del beer["idx"]
del beer["abv"]
del beer["ibu"]
del beer["id"]
del beer["style"]
del beer["ounces"]
beer["manu"] = csv.retreive(fpath_brew,
int(beer["brewery_id"]))["name"]
del beer["brewery_id"]
beer["type"] = "beer"
price = round(float(random.randint(1, 12)) + random.random(), 2)
beer["highest_price"] = price
beer["lowest_price"] = price
results.append(beer)
return results
def read_sells_table(): results = [] sells = csv.parse_file(f_sells, 3000) for sell in sells: result = gen.Sell( sell["bar"], sell["item"], sell["price"] ) results.append(result) return results
def read_frequents_table(): results = [] frequents = csv.parse_file(f_frequents, 2500) for freq in frequents: result = gen.Frequent( freq["drinker"], freq["bar"] ) results.append(result) return results
def read_transactions_table(): results = [] transactions = csv.parse_file(f_transactions, 20000) for trans in transactions: result = gen.Transaction( trans["trans_id"], trans["bar"], trans["drinker"], int(trans["day"]), gen.time_from_int_str(trans["time"]), float(trans["tip"]), float(trans["total"]) ) results.append(result) return results
def read_likes_table(): results = [] likes = csv.parse_file(f_likes, 10000) for like in likes: result = gen.Like( like["drinker"], like["item"] ) results.append(result) return results
def read_bars_table(compress=False): results = [] bars = csv.parse_file(f_bars, gen.max_bars*7) for bar in bars: if compress: if int(bar["day"]) != 1: continue result = gen.Bar( bar["name"], bar["license"], bar["city"], bar["state"], bar["phone"], bar["addr"], int(bar["day"]), gen.time_from_int_str(bar["open"]), gen.time_from_int_str(bar["closed"]) ) results.append(result) return results
def read_drinkers_table(): results = [] drinkers = csv.parse_file(f_drinkers, gen.max_drinkers) for drinker in drinkers: results.append(gen.Drinker( drinker["name"], drinker["city"], drinker["state"], drinker["phone"], drinker["addr"] )) return results