fpdb/pyfpdb/HandHistoryConverter.py

#!/usr/bin/python

#Copyright 2008 Carl Gherardi
#This program is free software: you can redistribute it and/or modify
#it under the terms of the GNU Affero General Public License as published by
#the Free Software Foundation, version 3 of the License.
#
#This program is distributed in the hope that it will be useful,
#but WITHOUT ANY WARRANTY; without even the implied warranty of
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#GNU General Public License for more details.
#
#You should have received a copy of the GNU Affero General Public License
#along with this program. If not, see <http://www.gnu.org/licenses/>.
#In the "official" distribution you can find the license in
#agpl-3.0.txt in the docs folder of the package.

import Configuration
import FpdbRegex
import re
import sys
import traceback
import os
import os.path
import xml.dom.minidom
from xml.dom.minidom import Node

class HandHistoryConverter:
	def __init__(self, config, file, sitename):
		print "HandHistory init called"
		self.c         = config
		self.sitename  = sitename
		self.obs       = ""             # One big string
		self.filetype  = "text"
		self.doc       = None     # For XML based HH files
		self.file      = file
		self.hhbase    = self.c.get_import_parameters().get("hhArchiveBase")
		self.hhbase    = os.path.expanduser(self.hhbase)
		self.hhdir     = os.path.join(self.hhbase,sitename)
		self.gametype  = []
#		self.ofile     = os.path.join(self.hhdir,file)
		self.rexx      = FpdbRegex.FpdbRegex()

	def __str__(self):
		tmp = "HandHistoryConverter: '%s'\n" % (self.sitename)
		tmp = tmp + "\thhbase:     '%s'\n" % (self.hhbase)
		tmp = tmp + "\thhdir:      '%s'\n" % (self.hhdir)
		tmp = tmp + "\tfiletype:   '%s'\n" % (self.filetype)
		tmp = tmp + "\tinfile:     '%s'\n" % (self.file)
#		tmp = tmp + "\toutfile:    '%s'\n" % (self.ofile)
		tmp = tmp + "\tgametype:   '%s'\n" % (self.gametype[0])
		tmp = tmp + "\tgamebase:   '%s'\n" % (self.gametype[1])
		tmp = tmp + "\tlimit:      '%s'\n" % (self.gametype[2])
		tmp = tmp + "\tsb/bb:      '%s/%s'\n" % (self.gametype[3], self.gametype[4])
		return tmp

	def processFile(self):
		if not self.sanityCheck():
			print "Cowardly refusing to continue after failed sanity check"
			return
		self.readFile(self.file)
		self.gametype = self.determineGameType()
		self.hands = self.splitFileIntoHands()
		for hand in self.hands:
			self.readHandInfo(hand)
			self.readPlayerStacks(hand)
			self.markStreets(hand)
			self.readBlinds(hand)
			self.readHeroCards(hand)
			if(hand.involved == True):
				self.writeHand("output file", hand)
			else:
				pass #Don't write out observed hands

	# Functions to be implemented in the inheriting class
	def readSupportedGames(self): abstract

	# should return a list
	#   type  base limit
	# [ ring, hold, nl   , sb, bb ]
	# Valid types specified in docs/tabledesign.html in Gametypes
	def determineGameType(self): abstract

	#TODO: Comment
	def readHandInfo(self, hand): abstract

	# Needs to return a list of lists in the format
	# [['seat#', 'player1name', 'stacksize'] ['seat#', 'player2name', 'stacksize'] [...]]
	def readPlayerStacks(self, hand): abstract

	def markStreets(hand): abstract

	#Needs to return a list in the format
	# ['player1name', 'player2name', ...] where player1name is the sb and player2name is bb, 
	# addtional players are assumed to post a bb oop
	def readBlinds(self, hand): abstract
	def readHeroCards(self, hand): abstract
	def readAction(self, hand, street): abstract

	def sanityCheck(self):
		sane = False
		base_w = False
		#Check if hhbase exists and is writable
		#Note: Will not try to create the base HH directory
		if not (os.access(self.hhbase, os.W_OK) and os.path.isdir(self.hhbase)):
			print "HH Sanity Check: Directory hhbase '" + self.hhbase + "' doesn't exist or is not writable"
		else:
			#Check if hhdir exists and is writable
			if not os.path.isdir(self.hhdir):
				# In first pass, dir may not exist. Attempt to create dir
				print "Creating directory: '%s'" % (self.hhdir)
				os.mkdir(self.hhdir)
				sane = True
			elif os.access(self.hhdir, os.W_OK):
				sane = True
			else:
				print "HH Sanity Check: Directory hhdir '" + self.hhdir + "' or its parent directory are not writable"

		return sane

	# Functions not necessary to implement in sub class
	def setFileType(self, filetype = "text"):
		self.filetype = filetype

	def splitFileIntoHands(self):
		hands = []
		list = self.rexx.split_hand_re.split(self.obs)
		list.pop() #Last entry is empty
		for l in list:
#			print "'" + l + "'"
			hands = hands + [Hand(self.sitename, self.gametype, l)]
		return hands

	def readFile(self, filename):
		"""Read file"""
		print "Reading file: '%s'" %(filename)
		if(self.filetype == "text"):
			infile=open(filename, "rU")
			self.obs = infile.read()
			infile.close()
		elif(self.filetype == "xml"):
			try:
				doc = xml.dom.minidom.parse(filename)
				self.doc = doc
			except:
				traceback.print_exc(file=sys.stderr)

	def writeHand(self, file, hand):
		"""Write out parsed data"""
		print "%s Game #%s: %s ($%s/$%s) - %s" %(hand.sitename, hand.handid, "XXXXhand.gametype", hand.sb, hand.bb, hand.starttime)
		print "Table '%s' %d-max Seat #%s is the button" %(hand.tablename, hand.maxseats, hand.buttonpos)

		for player in hand.players:
			print "Seat %s: %s ($%s)" %(player[0], player[1], player[2])

		if(hand.posted[0] == "FpdbNBP"):
			print "No small blind posted"
		else:
			print "%s: posts small blind $%s" %(hand.posted[0], hand.sb)

		#May be more than 1 bb posting
		print "%s: posts big blind $%s" %(hand.posted[1], hand.bb)
		if(len(hand.posted) > 2):
			# Need to loop on all remaining big blinds - lazy
			print "XXXXXXXXX FIXME XXXXXXXX"

		print "*** HOLE CARDS ***"
		print "Dealt to %s [%s]" %(hand.hero ,hand.holecards)
#
##		ACTION STUFF
#
		print "*** SUMMARY ***"
#		print "Total pot $%s | Rake $%s)" %(hand.totalpot  $" + hand.rake)
#		print "Board [" + boardcards + "]"
#
##		SUMMARY STUFF

#takes a poker float (including , for thousand seperator and converts it to an int
	def float2int (self, string):
		pos=string.find(",")
		if (pos!=-1): #remove , the thousand seperator
			string=string[0:pos]+string[pos+1:]

		pos=string.find(".")
		if (pos!=-1): #remove decimal point
			string=string[0:pos]+string[pos+1:]

		result = int(string)
		if pos==-1: #no decimal point - was in full dollars - need to multiply with 100
			result*=100
		return result
#end def float2int

class Hand:
#    def __init__(self, sitename, gametype, sb, bb, string):
    def __init__(self, sitename, gametype, string):
	self.sitename = sitename
	self.gametype = gametype
	self.string = string

	self.streets = {} # Index into string for where street starts { 'RIVER': 49 }
			  # Value in characters.

	self.handid = 0
	self.sb = gametype[3]
	self.bb = gametype[4]
	self.tablename = "Slartibartfast"
	self.maxseats = 10
	self.counted_seats = 0
	self.buttonpos = 0
	self.seating = []
	self.players = []
	self.posted = []
	self.involved = True
	self.hero = "Hiro"
	self.holecards = "Xx Xx"
	self.action = []
	self.totalpot = 0
	self.rake = 0

    def printHand(self):
	print self.sitename
	print self.gametype
	print self.string
	print self.handid
	print self.sb
	print self.bb
	print self.tablename
	print self.maxseats
	print self.counted_seats
	print self.buttonpos
	print self.seating
	print self.players
	print self.posted
	print self.action
	print self.involved
	print self.hero