diff options
author | emkael <emkael@tlen.pl> | 2024-01-15 12:24:37 +0100 |
---|---|---|
committer | emkael <emkael@tlen.pl> | 2024-01-15 12:26:44 +0100 |
commit | 9917692c976c547a35a027e9e09172d024517eca (patch) | |
tree | 7139d85cb6518a0ce210456af2a66f73b64d4158 /squaredeal | |
parent | 63225aa72d21c2bba6776fcaef2cb16e7eca12c3 (diff) |
Loading and parsing existing SQD/SQK files
Diffstat (limited to 'squaredeal')
-rw-r--r-- | squaredeal/__init__.py | 89 |
1 files changed, 89 insertions, 0 deletions
diff --git a/squaredeal/__init__.py b/squaredeal/__init__.py new file mode 100644 index 0000000..a036f20 --- /dev/null +++ b/squaredeal/__init__.py @@ -0,0 +1,89 @@ +import hashlib, os + + +class SquareDealError(Exception): + pass + + +class SquareDealPhase(object): + def __init__(self): + self.sessions = 0 + self.boards = 0 + self.prefix = '#' + self.info = '' + self.s_keys = [] + + def fromstring(self, phasestr): + parts = phasestr.split(':') + if len(parts) != 4: + raise SquareDealError('Malformed phase definition: %s' % (phasestr)) + self.sessions = int(parts[0]) + self.boards = int(parts[1]) + self.prefix = parts[2] + self.info = parts[3] + self.s_keys = [None] * self.sessions + + +class SquareDeal(object): + def __init__(self): + self.name = '' + self.delayed_info = '' + self.hash = '' + self.phases = [] + self.published = False + + def fromfile(self, sqdpath, encoding='utf-8', sqkpath=None): + with open(sqdpath, encoding=encoding) as sqdfile: + contents = [line.strip() for line in sqdfile.readlines()] + for idx, line in enumerate(contents): + linetype, _, linecontents = line.partition(' ') + if linetype == 'TN': + self.name = linecontents + elif linetype == 'DI': + self.delayed_info = linecontents + elif linetype == 'KH': + self.hash = linecontents + elif linetype == 'PU': + self.published = True + elif linetype == 'SN': + phase = SquareDealPhase() + phase.fromstring(linecontents) + self.phases.append(phase) + else: + raise SquareDealError('Unrecognized tag %s on line %d' % (linetype, idx)) + if sqkpath is None: + sqkpath = list(os.path.splitext(sqdpath)) + sqkpath[-1] = '.sqk' + sqkpath = ''.join(sqkpath) + try: + with open(sqkpath, encoding=encoding) as sqkfile: + contents = [line.strip() for line in sqkfile.readlines()] + except FileNotFoundError: + raise SquareDealError('Unable to locate SQK file for %s' % (sqdpath)) + for line in contents: + lineparts = line.split(':') + if len(lineparts) != 2: + raise SquareDealError('Malformed SQK line: %s' % (line)) + session = lineparts[0].split(',') + if len(session) != 2: + raise SquareDealError('Malformed SQK line: %s' % (line)) + phase_no = int(session[0]) + session_no = int(session[1]) + try: + self.phases[phase_no-1].s_keys[session_no-1] = lineparts[1] + except IndexError: + raise SquareDealError('Session %s from SQK not declared in SQD' % (lineparts[0])) + for ph_idx, phase in enumerate(self.phases): + for s_idx, s_key in enumerate(phase.s_keys): + if s_key is None: + raise SquareDealError('Session %d,%d missing a key in SQK' % (ph_idx+1, s_idx+1)) + with open(sqkpath, 'rb') as sqkfile: + sqk_hash = hashlib.sha256() + while True: + sqk_chunk = sqkfile.read(1024) + if not sqk_chunk: + break + sqk_hash.update(sqk_chunk) + sqk_hash = sqk_hash.hexdigest() + if sqk_hash != self.hash: + raise SquareDealError('SQK hash mismtach: %s in SQD, % actual' % (self.hash, sqk_hash)) |