def bootstrap(): Card.delete().execute() def load_dataset(filename, data=[], keys=[]): with open(filename, 'rb') as csvfile: lines = csv.reader(csvfile) dataset = list(lines) keys = dataset[0] for x in range(1, len(dataset)): element = {} for y in range(len(dataset[x])): element[keys[y]] = dataset[x][y] data.append(element) decks_file = "WebSite/hearthstonedecks.csv" raw_data = [] load_dataset(decks_file, raw_data) print("Raw_data: " + repr(len(raw_data))) with db.database.atomic(): step = 100 for idx in range(0, len(raw_data), step): Card.insert_many(raw_data[idx:idx + step]).execute() print("Inserted: " + repr(idx)) cards = Card.select().execute() for card in raw_data: naive_bayes.add_card(card) k_neighbours.update_deck(card) return make_response("Success")
def add_card_entry(card): #card["card_count"] = int(card["card_count"]) naive_bayes.add_card(card) k_neighbours.update_deck(card) Card.insert_many([card]).execute()