DGE format support

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)