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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
import re, subprocess
from squaredeal.sqd import SQD, SQDPhase, generate_session_key, validate_board_range_str
class SquareDealError(Exception):
pass
def squaredeal_board_range(arg_str):
ranges = []
for range_str in arg_str.split(','):
range_match = re.match(r'^([0-9]+)x([0-9]+)$', range_str)
if range_match:
subrange_count = int(range_match.group(2))
ranges += ['%d-%d' % (i*subrange_count+1, (i+1)*subrange_count) for i in range(0, int(range_match.group(1)))]
continue
ranges += [validate_board_range_str(range_str)]
return ','.join(ranges)
class SquareDeal(object):
# TODO: docstrings for command parameters
BIGDEALX_PATH = None
def __init__(self):
self.sqd = SQD()
def create(self, **arguments):
self.sqd.name = arguments.get('event_name')
self.sqd.delayed_info = arguments.get('delayed_information')
self.sqd.tofile(arguments.get('sqd_file'))
def set_name(self, **arguments):
self.sqd.fromfile(arguments.get('sqd_file'), sqkpath=arguments.get('sqk_file'))
if self.sqd.published:
raise SquareDealError('Cannot change name: event already published')
self.sqd.name = arguments.get('event_name')
self.sqd.tofile(arguments.get('sqd_file'))
def set_di(self, **arguments):
self.sqd.fromfile(arguments.get('sqd_file'), sqkpath=arguments.get('sqk_file'))
if self.sqd.published:
raise SquareDealError('Cannot change delayed information description: event already published')
self.sqd.delayed_info = arguments.get('delayed_information')
self.sqd.tofile(arguments.get('sqd_file'))
def add_phase(self, **arguments):
self.sqd.fromfile(arguments.get('sqd_file'), sqkpath=arguments.get('sqk_file'))
if self.sqd.published:
raise SquareDealError('Cannot add phase: event already published')
sdphase = SQDPhase()
sdphase.sessions = arguments.get('sessions')
sdphase.boards = arguments.get('boards')
sdphase.prefix = arguments.get('prefix')
sdphase.info = arguments.get('description')
self.sqd.phases.append(sdphase)
self.sqd.tofile(arguments.get('sqd_file'))
def publish(self, **arguments):
self.sqd.fromfile(arguments.get('sqd_file'), sqkpath=arguments.get('sqk_file'))
if self.sqd.published:
raise SquareDealError('Cannot mark as published: event already published')
if not self.sqd.name:
raise SquareDealError('Cannot mark as published: event name is not set')
if not self.sqd.delayed_info:
raise SquareDealError('Cannot mark as published: delayed information is not set')
if not self.sqd.phases:
raise SquareDealError('Cannot mark as published: no phases are defined')
for sdphase in self.sqd.phases:
sdphase.s_keys = [generate_session_key() for s in range(0, sdphase.sessions)]
self.sqd.published = True
self.sqd.tofile(arguments.get('sqd_file'), sqkpath=arguments.get('sqk_file'))
def set_dv(self, **arguments):
self.sqd.fromfile(arguments.get('sqd_file'), sqkpath=arguments.get('sqk_file'))
if not self.sqd.published:
raise SquareDealError('Cannot set delayed information value: event not published')
self.sqd.delayed_value = arguments.get('delayed_information')
self.sqd.tofile(arguments.get('sqd_file'), sqkpath=arguments.get('sqk_file'))
def generate(self, **arguments):
if not SquareDeal.BIGDEALX_PATH:
raise SquareDealError('Path to BigDeal is not set, initialize SquareDeal.BIGDEALX_PATH value')
self.sqd.fromfile(arguments.get('sqd_file'), sqkpath=arguments.get('sqk_file'))
if not self.sqd.published:
raise SquareDealError('Cannot generate PBN files: event info is not marked as published')
if not self.sqd.delayed_value:
raise SquareDealError('Cannot generate PBN files: delayed information value not set')
try:
self.sqd.generate(arguments.get('phase'), arguments.get('session'),
reserve=arguments.get('reserve'), bigdealx_path=SquareDeal.BIGDEALX_PATH)
except subprocess.CalledProcessError as ex:
raise SquareDealError('BigDeal invocation failed: %s' % (ex.stderr))
|
