diff options
| -rw-r--r-- | dealconvert/formats/bhg.py | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/dealconvert/formats/bhg.py b/dealconvert/formats/bhg.py index 1542aa1..7ff0704 100644 --- a/dealconvert/formats/bhg.py +++ b/dealconvert/formats/bhg.py @@ -1,6 +1,53 @@ +import sys + from . import DealFormat +from .. import dto class BHGFormat(DealFormat): @property def suffix(self): return '.bhg' + + def parse_content(self, content): + board_lines = [line.strip() for line in content.readlines()] + deals = [] + for board_no, line in enumerate(board_lines): + if board_no > 0: + if len(line) != 52: + print 'WARNING: malformed .bhg line #%d: %s' % (board_no, line) + elif not line.isalpha(): + print 'WARNING: invalid characters in .bhg line #%d: %s' % (board_no, line) + else: + deal = dto.Deal() + deal.number = board_no + deal.vulnerable = deal.get_vulnerability(board_no) + deal.dealer = deal.get_dealer(board_no) + hands = [[ + ord(c) - 65 if ord(c) < 96 else ord(c) - 71 + for c in list( + line[13*i:13*(i+1)] + )] for i in range(0, 4) + ] + for hand, cards in enumerate(hands): + for card in cards: + suit = card / 13 + card = self.cards[card % 13] + deal.hands[(hand + deal.dealer) % 4][suit].append(card) + deals.append(deal) + return deals + + def output_content(self, out_file, dealset): + lines = [''] * (max([board.number for board in dealset])+2) + for deal in dealset: + line = '' + for hand in range(0, 4): + for i, suit in enumerate(deal.hands[(hand + deal.dealer) % 4]): + try: + cards = [13*i + self.cards.index(card) for card in suit] + for card in cards: + line += chr((65 if card < 26 else 71)+card) + except ValueError: + print 'ERROR: invalid suit %s in board #%d' % (''.join(suit), deal.number) + sys.exit() + lines[deal.number] = line + out_file.write('\r\n'.join(lines)) |