First pass at turning file from pokerstats into class

pyfpdb/FpdbRegex.py

@@ -8,6 +8,9 @@
 # it under the terms of the GNU General Public License as published by
 # the Free Software Foundation, version 3 of the License.
 #
+
+# Modified for use in fpdb by Carl Gherardi
+
 from errors import *
 import re
 import regex
@@ -22,35 +25,41 @@ import regex
 # the hand history title line. Hence, to allow these strangely saved
 # hands to be parsed and imported, there is a conditional "one extra
 # character" allowed at the start of the new hand regex.
-__NEW_HAND_REGEX='^.?PokerStars Game #\d+:\s+Hold\'em'
-__HAND_INFO_REGEX='^.*#(\d+):\s+(\S+)\s([\s\S]+)\s\(\$?([.0-9]+)/\$?([.0-9]+)\)\s-\s(\S+)\s-?\s?(\S+)\s\(?(\w+)\)?'
-__TABLE_INFO_REGEX='^\S+\s+\'.*\'\s+(\d+)-max\s+Seat\s#(\d+)'
-__PLAYER_INFO_REGEX='^Seat\s(\d+):\s(.*)\s\(\$?([.\d]+)\s'
-__POST_SB_REGEX='^(.*):\sposts small blind'
-__POST_BB_REGEX='^(.*):\sposts big blind'
-__POST_BOTH_REGEX='^(.*):\sposts small & big blinds'
-__HAND_STAGE_REGEX='^\*{3}\s(.*)\s\*{3}'
-__HOLE_CARD_REGEX='^\*{3}\sHOLE CARDS'
-__FLOP_CARD_REGEX='^\*{3}\sFLOP\s\*{3}\s\[(\S{2})\s(\S{2})\s(\S{2})\]'
-__TURN_CARD_REGEX='^\*{3}\sTURN\s\*{3}\s\[\S{2}\s\S{2}\s\S{2}\]\s\[(\S{2})\]'
-__RIVER_CARD_REGEX='^\*{3}\sRIVER\s\*{3}\s\[\S{2}\s\S{2}\s\S{2}\s\S{2}\]\s\[(\S{2})\]'
-__SHOWDOWN_REGEX='^\*{3}\sSHOW DOWN'
-__SUMMARY_REGEX='^\*{3}\sSUMMARY'
-__UNCALLED_BET_REGEX='^Uncalled bet \(\$([.\d]+)\) returned to (.*)'
-__POT_AND_RAKE_REGEX='^Total\spot\s\$([.\d]+).*\|\sRake\s\$([.\d]+)'
-__COLLECT_POT_REGEX='^(.*)\scollected\s\$([.\d]+)\sfrom\s((main|side)\s)?pot'
-__POCKET_CARDS_REGEX='^Dealt\sto\s(.*)\s\[(\S{2})\s(\S{2})\]'
-__SHOWN_CARDS_REGEX='^(.*):\sshows\s\[(\S{2})\s(\S{2})\]'
-__ACTION_STEP_REGEX='^(.*):\s(bets|checks|raises|calls|folds)((\s\$([.\d]+))?(\sto\s\$([.\d]+))?)?'
-
-__SHOWDOWN_ACTION_REGEX='^(.*):\s(shows|mucks)'
-__SUMMARY_CARDS_REGEX='^Seat\s\d+:\s(.*)\s(showed|mucked)\s\[(\S{2})\s(\S{2})\]'
-__SUMMARY_CARDS_EXTRA_REGEX='^Seat\s\d+:\s(.*)\s(\(.*\)\s)(showed|mucked)\s\[(\S{2})\s(\S{2})\]'
 
 
-def regexes():
-    m = regex.RegexMatch()
-### Compile the regexes
+class FpdbRegex:
+	def __init__(self):
+		__NEW_HAND_REGEX='^.?PokerStars Game #\d+:\s+Hold\'em'
+		__HAND_INFO_REGEX='^.*#(\d+):\s+(\S+)\s([\s\S]+)\s\(\$?([.0-9]+)/\$?([.0-9]+)\)\s-\s(\S+)\s-?\s?(\S+)\s\(?(\w+)\)?'
+		__TABLE_INFO_REGEX='^\S+\s+\'.*\'\s+(\d+)-max\s+Seat\s#(\d+)'
+		__PLAYER_INFO_REGEX='^Seat\s(\d+):\s(.*)\s\(\$?([.\d]+)\s'
+		__POST_SB_REGEX='^(.*):\sposts small blind'
+		__POST_BB_REGEX='^(.*):\sposts big blind'
+		__POST_BOTH_REGEX='^(.*):\sposts small & big blinds'
+		__HAND_STAGE_REGEX='^\*{3}\s(.*)\s\*{3}'
+		__HOLE_CARD_REGEX='^\*{3}\sHOLE CARDS'
+		__FLOP_CARD_REGEX='^\*{3}\sFLOP\s\*{3}\s\[(\S{2})\s(\S{2})\s(\S{2})\]'
+		__TURN_CARD_REGEX='^\*{3}\sTURN\s\*{3}\s\[\S{2}\s\S{2}\s\S{2}\]\s\[(\S{2})\]'
+		__RIVER_CARD_REGEX='^\*{3}\sRIVER\s\*{3}\s\[\S{2}\s\S{2}\s\S{2}\s\S{2}\]\s\[(\S{2})\]'
+		__SHOWDOWN_REGEX='^\*{3}\sSHOW DOWN'
+		__SUMMARY_REGEX='^\*{3}\sSUMMARY'
+		__UNCALLED_BET_REGEX='^Uncalled bet \(\$([.\d]+)\) returned to (.*)'
+		__POT_AND_RAKE_REGEX='^Total\spot\s\$([.\d]+).*\|\sRake\s\$([.\d]+)'
+		__COLLECT_POT_REGEX='^(.*)\scollected\s\$([.\d]+)\sfrom\s((main|side)\s)?pot'
+		__POCKET_CARDS_REGEX='^Dealt\sto\s(.*)\s\[(\S{2})\s(\S{2})\]'
+		__SHOWN_CARDS_REGEX='^(.*):\sshows\s\[(\S{2})\s(\S{2})\]'
+		__ACTION_STEP_REGEX='^(.*):\s(bets|checks|raises|calls|folds)((\s\$([.\d]+))?(\sto\s\$([.\d]+))?)?'
+
+		__SHOWDOWN_ACTION_REGEX='^(.*):\s(shows|mucks)'
+		__SUMMARY_CARDS_REGEX='^Seat\s\d+:\s(.*)\s(showed|mucked)\s\[(\S{2})\s(\S{2})\]'
+		__SUMMARY_CARDS_EXTRA_REGEX='^Seat\s\d+:\s(.*)\s(\(.*\)\s)(showed|mucked)\s\[(\S{2})\s(\S{2})\]'
+		self.m = regex.RegexMatch()
+
+	def getRegexes():
+		return self.m
+
+	def compileRegexes():
+		### Compile the regexes
 		m.hand_start_re = re.compile(__NEW_HAND_REGEX)
 		m.hand_info_re = re.compile(__HAND_INFO_REGEX)
 		m.table_info_re = re.compile(__TABLE_INFO_REGEX)
@@ -74,6 +83,75 @@ def regexes():
 		m.action_re = re.compile(__ACTION_STEP_REGEX)
 		m.rake_re = re.compile(__POT_AND_RAKE_REGEX)
 		m.showdown_action_re = re.compile(__SHOWDOWN_ACTION_REGEX)
-###
-    return m
+
+	# Set methods for plugins to override
+
+	def setNewHandRegex(self, string):
+		__NEW_HAND_REGEX = string
+
+	def setHandInfoRegex(self, string):
+		__HAND_INFO_REGEX = string
+
+	def setTableInfoRegex(self, string):
+		__TABLE_INFO_REGEX = string
+
+	def setPlayerInfoRegex(self, string):
+		__PLAYER_INFO_REGEX = string
+
+	def setPostSbRegex(self, string):
+		__POST_SB_REGEX = string
+
+	def setPostBbRegex(self, string):
+		__POST_BB_REGEX = string
+
+	def setPostBothRegex(self, string):
+		__POST_BOTH_REGEX = string
+
+	def setHandStageRegex(self, string):
+		__HAND_STAGE_REGEX = string
+
+	def setHoleCardRegex(self, string):
+		__HOLE_CARD_REGEX = string
+
+	def setFlopCardRegex(self, string):
+		__FLOP_CARD_REGEX = string
+
+	def setTurnCardRegex(self, string):
+		__TURN_CARD_REGEX = string
+
+	def setRiverCardRegex(self, string):
+		__RIVER_CARD_REGEX = string
+
+	def setShowdownRegex(self, string):
+		__SHOWDOWN_REGEX = string
+
+	def setSummaryRegex(self, string):
+		__SUMMARY_REGEX = string
+
+	def setUncalledBetRegex(self, string):
+		__UNCALLED_BET_REGEX = string
+
+	def setCollectPotRegex(self, string):
+		__COLLECT_POT_REGEX = string
+
+	def setPocketCardsRegex(self, string):
+		__POCKET_CARDS_REGEX = string
+
+	def setShownCardsRegex(self, string):
+		__SHOWN_CARDS_REGEX = string
+
+	def setSummaryCardsRegex(self, string):
+		__SUMMARY_CARDS_REGEX = string
+
+	def setSummaryCardsExtraRegex(self, string):
+		__SUMMARY_CARDS_EXTRA_REGEX = string
+
+	def setActionStepRegex(self, string):
+		__ACTION_STEP_REGEX = string
+
+	def setPotAndRakeRegex(self, string):
+		__POT_AND_RAKE_REGEX = string
+
+	def setShowdownActionRegex(self, string):
+		__SHOWDOWN_ACTION_REGEX = string