196 lines
6.8 KiB
Python
196 lines
6.8 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""HandHistory.py
|
|
|
|
Parses HandHistory xml files and returns requested objects.
|
|
"""
|
|
# Copyright 2008-2010, Ray E. Barker
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# 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 General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
########################################################################
|
|
# Standard Library modules
|
|
import xml.dom.minidom
|
|
from xml.dom.minidom import Node
|
|
|
|
class HandHistory:
|
|
def __init__(self, xml_string, elements = ('ALL')):
|
|
|
|
doc = xml.dom.minidom.parseString(xml_string)
|
|
if elements == ('ALL'):
|
|
elements = ('BETTING', 'AWARDS', 'POSTS', 'PLAYERS', 'GAME')
|
|
|
|
if 'BETTING' in elements:
|
|
self.BETTING = Betting(doc.getElementsByTagName('BETTING')[0])
|
|
if 'AWARDS' in elements:
|
|
self.AWARDS = Awards (doc.getElementsByTagName('AWARDS')[0])
|
|
if 'POSTS' in elements:
|
|
self.POSTS = Posts (doc.getElementsByTagName('POSTS')[0])
|
|
if 'GAME' in elements:
|
|
self.GAME = Game (doc.getElementsByTagName('GAME')[0])
|
|
if 'PLAYERS' in elements:
|
|
self.PLAYERS = {}
|
|
p_n = doc.getElementsByTagName('PLAYERS')[0]
|
|
for p in p_n.getElementsByTagName('PLAYER'):
|
|
a_player = Player(p)
|
|
self.PLAYERS[a_player.name] = a_player
|
|
|
|
class Player:
|
|
def __init__(self, node):
|
|
self.name = node.getAttribute('NAME')
|
|
self.seat = node.getAttribute('SEAT')
|
|
self.stack = node.getAttribute('STACK')
|
|
self.showed_hand = node.getAttribute('SHOWED_HAND')
|
|
self.cards = node.getAttribute('CARDS')
|
|
self.allin = node.getAttribute('ALLIN')
|
|
self.sitting_out = node.getAttribute('SITTING_OUT')
|
|
self.hand = node.getAttribute('HAND')
|
|
self.start_cards = node.getAttribute('START_CARDS')
|
|
|
|
if self.allin == '' or \
|
|
self.allin == '0' or \
|
|
self.allin.upper() == 'FALSE': self.allin = False
|
|
else: self.allin = True
|
|
|
|
if self.sitting_out == '' or \
|
|
self.sitting_out == '0' or \
|
|
self.sitting_out.upper() == 'FALSE': self.sitting_out = False
|
|
else: self.sitting_out = True
|
|
|
|
def __str__(self):
|
|
temp = "%s\n seat = %s\n stack = %s\n cards = %s\n" % \
|
|
(self.name, self.seat, self.stack, self.cards)
|
|
temp = temp + " showed_hand = %s\n allin = %s\n" % \
|
|
(self.showed_hand, self.allin)
|
|
temp = temp + " hand = %s\n start_cards = %s\n" % \
|
|
(self.hand, self.start_cards)
|
|
return temp
|
|
|
|
class Awards:
|
|
def __init__(self, node):
|
|
self.awards = [] # just an array of award objects
|
|
for a in node.getElementsByTagName('AWARD'):
|
|
self.awards.append(Award(a))
|
|
|
|
def __str__(self):
|
|
temp = ""
|
|
for a in self.awards:
|
|
temp = temp + "%s\n" % (a)
|
|
return temp
|
|
|
|
class Award:
|
|
def __init__(self, node):
|
|
self.player = node.getAttribute('PLAYER')
|
|
self.amount = node.getAttribute('AMOUNT')
|
|
self.pot = node.getAttribute('POT')
|
|
|
|
def __str__(self):
|
|
return self.player + " won " + self.amount + " from " + self.pot
|
|
|
|
class Game:
|
|
def __init__(self, node):
|
|
print node
|
|
self.tags = {}
|
|
for tag in ( ('GAME_NAME', 'game_name'), ('MAX', 'max'), ('HIGHLOW', 'high_low'),
|
|
('STRUCTURE', 'structure'), ('MIXED', 'mixed') ):
|
|
L = node.getElementsByTagName(tag[0])
|
|
if (not L): continue
|
|
print L
|
|
for node2 in L:
|
|
title = ""
|
|
for node3 in node2.childNodes:
|
|
if (node3.nodeType == Node.TEXT_NODE):
|
|
title +=node3.data
|
|
self.tags[tag[1]] = title
|
|
|
|
def __str__(self):
|
|
return "%s %s %s, (%s max), %s" % (self.tags['structure'],
|
|
self.tags['game_name'],
|
|
self.tags['game_name'],
|
|
self.tags['max'],
|
|
self.tags['game_name'])
|
|
|
|
class Posts:
|
|
def __init__(self, node):
|
|
self.posts = [] # just an array of post objects
|
|
for p in node.getElementsByTagName('POST'):
|
|
self.posts.append(Post(p))
|
|
|
|
def __str__(self):
|
|
temp = ""
|
|
for p in self.posts:
|
|
temp = temp + "%s\n" % (p)
|
|
return temp
|
|
|
|
class Post:
|
|
def __init__(self, node):
|
|
self.player = node.getAttribute('PLAYER')
|
|
self.amount = node.getAttribute('AMOUNT')
|
|
self.posted = node.getAttribute('POSTED')
|
|
self.live = node.getAttribute('LIVE')
|
|
|
|
def __str__(self):
|
|
return ("%s posted %s %s %s") % (self.player, self.amount, self.posted, self.live)
|
|
|
|
class Betting:
|
|
def __init__(self, node):
|
|
self.rounds = [] # a Betting object is just an array of rounds
|
|
for r in node.getElementsByTagName('ROUND'):
|
|
self.rounds.append(Round(r))
|
|
|
|
def __str__(self):
|
|
temp = ""
|
|
for r in self.rounds:
|
|
temp = temp + "%s\n" % (r)
|
|
return temp
|
|
|
|
class Round:
|
|
def __init__(self, node):
|
|
self.name = node.getAttribute('ROUND_NAME')
|
|
self.action = []
|
|
for a in node.getElementsByTagName('ACTION'):
|
|
self.action.append(Action(a))
|
|
|
|
def __str__(self):
|
|
temp = self.name + "\n"
|
|
for a in self.action:
|
|
temp = temp + " %s\n" % (a)
|
|
return temp
|
|
|
|
class Action:
|
|
def __init__(self, node):
|
|
self.player = node.getAttribute('PLAYER')
|
|
self.action = node.getAttribute('ACT')
|
|
self.amount = node.getAttribute('AMOUNT')
|
|
self.allin = node.getAttribute('ALLIN')
|
|
|
|
def __str__(self):
|
|
return self.player + " " + self.action + " " + self.amount + " " + self.allin
|
|
|
|
if __name__== "__main__":
|
|
file = open('test.xml', 'r')
|
|
xml_string = file.read()
|
|
file.close()
|
|
|
|
print xml_string + "\n\n\n"
|
|
h = HandHistory(xml_string, ('ALL'))
|
|
print h.GAME
|
|
print h.POSTS
|
|
print h.BETTING
|
|
print h.AWARDS
|
|
|
|
for p in h.PLAYERS.keys():
|
|
print h.PLAYERS[p]
|