def update_db(self, mongo_db_inst):
for event_type_name, stats_dict in self.event_stats.iteritems():
mongo_collection = mongo_db_inst[event_type_name]
for full_key, gain_stats_obj in sorted(stats_dict.iteritems()):
existing_raw_obj = mongo_collection.find_one({'_id': full_key})
if existing_raw_obj:
existing_stat = SmallGainStat()
existing_stat.from_primitive_object(
existing_raw_obj['vals'])
gain_stats_obj.merge(existing_stat)
key_wrap_obj = {'vals': gain_stats_obj.to_primitive_object()}
utils.write_object_to_db(key_wrap_obj, mongo_collection,
full_key)
def update_db(self, mongo_db_inst):
for event_type_name, stats_dict in self.event_stats.iteritems():
log.debug('Updating database for event type %s, %d stats',
event_type_name, len(stats_dict))
mongo_collection = mongo_db_inst[event_type_name]
inserts = 0
updates = 0
for full_key, gain_stats_obj in sorted(stats_dict.iteritems()):
existing_raw_obj = mongo_collection.find_one({'_id': full_key})
if existing_raw_obj:
updates += 1
existing_stat = SmallGainStat()
existing_stat.from_primitive_object(
existing_raw_obj['vals'])
gain_stats_obj.merge(existing_stat)
else:
inserts += 1
key_wrap_obj = {'vals': gain_stats_obj.to_primitive_object()}
utils.write_object_to_db(key_wrap_obj, mongo_collection,
full_key)
log.debug('Database update results for %s: %d inserts, %d updates',
event_type_name, inserts, updates)
def fetch_conditional_stats(self, target_inds, interaction_tuples):
db = utils.get_mongo_database()
card_stats = []
count_searched = 0
for target_ind in target_inds:
for interaction_tuple in interaction_tuples:
count_searched += 1
if count_searched > 1000:
return card_stats
key = target_ind + ';' + (','.join(interaction_tuple))
db_val = db.card_supply.find_one({'_id': key})
if db_val:
small_gain_stat = SmallGainStat()
small_gain_stat.from_primitive_object(db_val['vals'])
def name_getter(ind_str):
return dominioncards.index_to_card(int(ind_str)).singular
card_name = name_getter(target_ind)
condition = map(name_getter, interaction_tuple)
stat_with_context = {'card_name': card_name,
'condition': condition,
'stats': small_gain_stat}
card_stats.append(stat_with_context)
return card_stats
def accumulate_card_stats(games_stream, stats_accumulator, max_games=-1):
ct = 0
for game in games_stream:
detected_events = detect_events(game)
per_player_accum = game.cards_accumalated_per_player().iteritems()
for player, accum_dict in per_player_accum:
avail = analysis_util.available_cards(game, accum_dict.keys())
win_points = game.get_player_deck(player).WinPoints()
for card in avail:
count = accum_dict.get(card, 0)
small_gain_stat = SmallGainStat()
if count:
small_gain_stat.win_given_any_gain.add_outcome(win_points)
else:
small_gain_stat.win_given_no_gain.add_outcome(win_points)
small_gain_stat.win_weighted_gain.add_many_outcomes(
win_points, count)
card_index = str(card_info.card_index(card))
stats_accumulator.merge_stats(detected_events, card_index,
small_gain_stat)
max_games -= 1
if max_games == 0:
break
