| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071 |
- import re
- from datetime import date
- # from pprint import pprint
- from xml.etree.ElementTree import Element, SubElement, tostring, ElementTree
- # from xml.etree.ElementTree import dump
- import xml.dom.minidom
- def process_pbn(xml_root):
- order_of_suits = ["Spades", "Hearts", "Diamonds", "Clubs"]
- dict_of_hands = {"N": "North", "E": "East", "S": "South", "W": "West"}
- with open('hands.pbn', 'r') as f_pbn:
- for line in f_pbn:
- if not line.strip():
- dealer = ''
- vulnerable = ''
- if '[Dealer' in line:
- dealer = line[9]
- if '[Vulnerable' in line:
- reg = m = re.search('NS|EW|All|None', line)
- vulnerable = reg.group(0)
- if '[Deal ' in line:
- # print('{}'.format(line))
- # print('{}'.format(line[7]))
- # print('{}'.format(line[9:-3]))
- first_hand = line[7]
- assert first_hand in ['N', 'E', 'S', 'W'], \
- "Cannot determine first hand. Got {} but expected 'N', 'E', 'S' or 'W'".format(first_hand)
- order_of_hands = []
- if first_hand == 'N':
- order_of_hands = ["North", "East", "South", "West"]
- hands = line[9:-3].split(' ')
- assert len(hands) == 4, "There are no 4 hands in PBN string {}".format(line[9:-3])
- xml_deal = SubElement(xml_root, "Deal")
- assert dealer in ["N", "E", "S", "W", ''], "Unknown Dealer {}".format(dealer)
- if dealer:
- xml_deal.attrib["Dealer"] = dict_of_hands[dealer]
- assert vulnerable in ["NS", "EW", "All", "None", ''], "Unknown vulnerable {}".format(vulnerable)
- if vulnerable:
- xml_deal.attrib['Vulnerable'] = vulnerable
- i = 0
- for hand in hands:
- if '.' in hand:
- xml_hand = SubElement(xml_deal, "Hand")
- xml_hand.attrib["Position"] = order_of_hands[i]
- suits = hand.split('.')
- assert len(suits) == 4, "There are no 4 suits in PBN hand {}".format(suits)
- j = 0
- for suit in suits:
- xml_hand.attrib[order_of_suits[j]] = suit
- j += 1
- # dump(xml_hand)
- # dump(xml_deal)
- i += 1
- if __name__ == '__main__':
- xml_dealsets = Element("DealSets")
- xml_dealset = SubElement(xml_dealsets, "DealSet")
- xml_dealset.attrib["ComplexityLevel"] = "1"
- xml_dealset.attrib["IBIS-ID"] = "007-Foppe_Hemminga-{}".format(date.today().strftime('%Y%m%d'))
- xml_dealset.attrib["VulnerabilityDefault"] = "None"
- SubElement(xml_dealset, "Parameters")
- process_pbn(xml_dealset)
- # dump(xml_dealsets)
- xml_tree = ElementTree(xml_dealsets)
- dom = xml.dom.minidom.parseString(tostring(xml_dealsets, encoding='utf-8', method='xml'))
- # print(dom.toprettyxml().replace('<?xml version="1.0" ?>', '<?xml version=\'1.0\' encoding=\'utf-8\'?>'))
- with open("output.xml", 'w') as f_xml:
- f_xml.write(dom.toprettyxml().replace('<?xml version="1.0" ?>', '<?xml version=\'1.0\' encoding=\'utf-8\'?>'))
- # xml_tree.write("output.xml", encoding="utf-8", xml_declaration=True)
|
