blob: d0a4c464422bbcd5d8a13a1f916b8e9abfb297e2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
import warnings
from . import DealFormat
from .. import dto
class DGEFormat(DealFormat):
number_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):
warnings.warn(self.number_warning)
dealset = []
number = 1
while True:
deal_str = content.read(128).strip()
if len(deal_str) > 0:
if len(deal_str) < 68:
warnings.warn('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)
deal.hands = self.parse_hands(deal_str[0:68])
dealset.append(deal)
number += 1
else:
break
return dealset
def parse_hands(self, deal_str):
deal = dto.Deal()
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:
raise RuntimeError('invalid .dge line: %s' % (deal_str))
else:
deal.hands[hand][suit].append(char)
return deal.hands
def output_content(self, out_file, dealset):
warnings.warn(self.number_warning)
for deal in dealset:
deal_str = self.single_deal_output(deal)
deal_str += chr(0) * 60
out_file.write(deal_str)
def single_deal_output(self, deal):
deal_str = ''
for hand in deal.hands:
for suit, cards in enumerate(hand):
deal_str += self.suit_indicator(suit)
deal_str += ''.join(cards)
return deal_str
|
