Tinymovr

from typing import List
from enum import IntEnum

from tinymovr.constants import (ControlStates,
    ControlModes, ErrorIDs, error_descriptions)

class StateObj():

    def __init__(self, data):
        self.data = data

    @property
    def state(self):
        return ControlStates(self.data[1])

    @property
    def mode(self):
        return ControlModes(self.data[2])

    @property
    def errors(self):
        if sum(self.data[-5:]) > 0:
            return [ErrorIDs(e) for e in self.data[-5:] if e > 0]
        elif (self.data[0] > 0):
            return [ErrorIDs(self.data[0])]
        return None

    @property
    def error_descriptions(self):
        if self.errors:
            return [error_descriptions[e] for e in self.errors]
        return None

    def __dir__(self):
        return ['state', 'mode', 'errors', 'error_descriptions']

    def _repr_pretty_(self, p, cycle):
        strings = [
            'State: {}; Mode: {}'.format(str(ControlStates(self.data[1])), str(ControlModes(self.data[2]))),
            'Errors:'
        ]
        if self.errors:
            strings.extend(['  ' + str(e) + '(' + str(int(e)) + '): ' + d for e, d in zip(self.errors, self.error_descriptions)])
        else:
            strings.append('  None')
        p.text('\n'.join(strings))