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"]
    with open('hands.pbn', 'r') as f_pbn:
        for line in f_pbn:
            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")
                # dump(xml_deal)
                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)