DUP format

1 files changed, 54 insertions, 0 deletions

diff --git a/dealconvert/formats/dup.py b/dealconvert/formats/dup.py
index 081599a..5b78bfc 100644
--- a/dealconvert/formats/dup.py
+++ b/dealconvert/formats/dup.py
@@ -1,6 +1,60 @@
+import sys
+
 from . import DealFormat
+from .bri import BRIFormat
+from .dge import DGEFormat
+from .. import dto
 
 class DUPFormat(DealFormat):
     @property
     def suffix(self):
         return '.dup'
+
+    def __init__(self):
+        self.bri = BRIFormat()
+        self.dge = DGEFormat()
+
+    def parse_content(self, content):
+        boards = []
+        while True:
+            boards.append(content.read(156))
+            if len(boards[-1]) < 156:
+                if len(boards[-1]) > 0:
+                    print 'WARNING: truncated .dup content: %s' % (boards[-1])
+                boards = boards[0:-1]
+                break
+        boards = [(board[0:78], board[78:146], board[146:]) for board in boards]
+        if boards[0][2][0] == chr(0):
+            print 'ERROR: .dup file header not found'
+            sys.exit()
+        start_board = int(boards[0][2][2:4].strip())
+        board_count = int(boards[0][2][7:9].strip())
+        board_numbers = range(start_board, start_board+board_count)
+        if boards[0][2][1].upper() != 'N':
+            board_numbers.reverse()
+        dealset = []
+        for idx, board in enumerate(boards):
+            deal = dto.Deal()
+            deal.number = board_numbers[idx]
+            deal.dealer = deal.get_dealer(deal.number)
+            deal.vulnerable = deal.get_vulnerability(deal.number)
+            deal.hands = self.bri.parse_hands(board[0])
+            dealset.append(deal)
+        return dealset
+
+    def output_content(self, out_file, dealset):
+        board_numbers = [deal.number for deal in dealset]
+        first_board = min(board_numbers)
+        board_count = len(dealset)
+        for board in range(first_board, first_board+board_count):
+            if board not in board_numbers:
+                print 'ERROR: .dup format requires consequent board numbers'
+                sys.exit()
+        header = 'YN%s 0 %02d ' % (str(first_board).ljust(2, ' '), board_count)
+        for deal in sorted(dealset, key=lambda d:d.number):
+            out_file.write(self.bri.single_deal_output(deal))
+            out_file.write(self.dge.single_deal_output(deal))
+            if deal.number == first_board:
+                out_file.write(header)
+            else:
+                out_file.write(chr(0) * 10)