class WorkspaceState:
def __init__(self, *, allow_start, allow_reboot):
self._allow_start = allow_start
self._allow_reboot = allow_reboot
@property def allow_start(self):
return self._allow_start
@property def allow_reboot(self):
return self._allow_reboot
states = dict(
available = WorkspaceState(allow_start=False, allow_reboot=True),
unhealthy = WorkspaceState(allow_start=False, allow_reboot=False),
stopped = WorkspaceState(allow_start=True, allow_reboot=False),
)
…
workspace = get_workspace(workspace_id)
if states[workspace.state].allow_reboot:
workspace.reboot()