Briggs Softworks

Code: Select all

def get_hands_pnl(player_name, hands):
	'''
	Input is a player name string and a dictionary of hands
	Output is a pandas dataframe with hand numbers and pnl
	'''

	import pandas as pd
	
	hands_pnl_df = pd.DataFrame()
	
	### Loop through each hand
	for hand_number in hands.keys():
		hand_text = hands[hand_number]['text']
		dollars_list = re.findall('\((.*?)\)', hand_text)
		player_winners = re.findall('\n(.*?) wins Pot', hand_text)
		if len(player_winners) < 1:
			player_winners = re.findall('\n(.*?) splits Pot', hand_text)
		pot_amount = float(dollars_list[-2]) / len(player_winners)
		
		### Try to match player_name with players who invested money into the pot
		players_outlay_str = dollars_list[-1]
		try:
			my_outlay = float(re.findall(player_name + '\: (.*?)\,', players_outlay_str + ',')[0])
		except:
			my_outlay = 0
		my_pnl = int(player_name in player_winners)*pot_amount - my_outlay
		hand_pnl_df = pd.DataFrame({'hand_number': hand_number, 'pnl': my_pnl}, index = [0])
		hands_pnl_df = pd.concat([hands_pnl_df, hand_pnl_df], ignore_index = True)
	
	return hands_pnl_df.sort_values('pnl', ascending = False)