コード例 #1
0
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")
コード例 #2
0
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()