diff options
| -rw-r--r-- | dealconvert/formats/dge.py | 65 |
1 files changed, 65 insertions, 0 deletions
diff --git a/dealconvert/formats/dge.py b/dealconvert/formats/dge.py index fa1720a..21d3044 100644 --- a/dealconvert/formats/dge.py +++ b/dealconvert/formats/dge.py @@ -1,6 +1,71 @@ +import sys + from . import DealFormat +from .. import dto class DGEFormat(DealFormat): + number_warning = 'WARNING: .dge file format assumes consequent deal numbers from 1' + suits = { + chr(6): dto.SUIT_SPADES, + chr(3): dto.SUIT_HEARTS, + chr(4): dto.SUIT_DIAMONDS, + chr(5): dto.SUIT_CLUBS + } + + def suit_indicator(self, the_suit): + for indicator, suit in self.suits.iteritems(): + if suit == the_suit: + return indicator + return None + @property def suffix(self): return '.dge' + + def parse_content(self, content): + print self.number_warning + dealset = [] + number = 1 + while True: + deal_str = content.read(128).strip() + if len(deal_str) > 0: + if len(deal_str) < 68: + print 'WARNING: truncated .dge input: %s' % (deal_str) + break + else: + deal = dto.Deal() + deal.number = number + deal.dealer = deal.get_dealer(number) + deal.vulnerable = deal.get_vulnerability(number) + hand = 0 + suit_count = -1 + suit = None + for char in deal_str[0:68]: + if char in self.suits: + suit = self.suits[char] + suit_count += 1 + if suit_count == 4: + suit_count = 0 + hand += 1 + else: + if suit is None: + print 'ERROR: invalid .dge line: %s' % (deal_str) + sys.exit() + else: + deal.hands[hand][suit].append(char) + dealset.append(deal) + number += 1 + else: + break + return dealset + + def output_content(self, out_file, dealset): + print self.number_warning + for deal in dealset: + deal_str = '' + for hand in deal.hands: + for suit, cards in enumerate(hand): + deal_str += self.suit_indicator(suit) + deal_str += ''.join(cards) + deal_str += chr(0) * 60 + out_file.write(deal_str) |