source: flowtimer/flowtimer/RecurringPhaseSequence.py@ d7a66ad

import json
from copy import deepcopy
from flowtimer.Phase import Phase


class RecurringPhaseSequence:

    @classmethod
    def from_json(cls, a_json_string):
        def custom_object_hook(d):
            if 'title' in d and 'initial_ticks' in d:
                return Phase(d['title'], d['initial_ticks'])
            if 'phase_list' in d and 'initial_repetitions' in d:
                return RecurringPhaseSequence(d["title"], d['phase_list'], d['initial_repetitions'])
            print("Wrong format")
            return d
        return json.loads(a_json_string, object_hook=custom_object_hook)

    @classmethod
    def default_json_string(cls):
        return json.dumps({"title": "default",
                           "phase_list": [{"title": "Tasking", "initial_ticks": 5},
                                          {"title": "Work", "initial_ticks": 45},
                                          {"title": "Break", "initial_ticks": 15}],
                           "initial_repetitions": 3})

    @classmethod
    def default(cls):
        return cls.from_json(cls.default_json_string())

    def __init__(self, title, phase_list, repetitions):
        assert repetitions > 0
        assert phase_list is not []
        self._title = title
        self.state = "initial"
        self.phase_list = phase_list
        self.current_phase = phase_list[0]
        self.initial_repetitions = repetitions
        self.passes_left = repetitions

    def to_json(self):
        return json.dumps(self.__dict__, default=lambda each: each.to_json())

    @property
    def title(self):
        return self._title

    def __str__(self):
        return ("Sequence title:" + self.title + "\n" + str(self.current_phase))

    def is_sequence(self):
        return True

    def current_phase_number(self):
        return self.phase_list.index(self.current_phase)

    def phases_left_in_pass(self):
        return len(self.upcoming_phases_in_pass())

    def upcoming_phases_in_pass(self):
        if self.current_phase_number() < len(self.phase_list) - 1:
            return self.phase_list[self.current_phase_number()+1:]
        return []

    @property
    def initial_ticks(self):
        return sum([each.initial_ticks for each in self.phase_list])

    @property
    def ticks_left(self):
        return (
            (self.passes_left-1) * sum([each.initial_ticks for each in self.phase_list]) +
            self.current_phase.ticks_left +
            sum([each.ticks_left for each in self.upcoming_phases_in_pass()]))

    def completed(self):
        return ((self.passes_left < 1) and
                (not self.upcoming_phases_in_pass() and
                 self.current_phase.completed()))

    def abort(self):
        self.current_phase.abort()
        self.state = "aborted"

    def start(self):
        self.state = "running"

    def skip(self):
        if self.upcoming_phases_in_pass():
            self.current_phase.reset()
            self.current_phase = self.upcoming_phases_in_pass()[0]
            return
        else:
            print("Sequence finished")
            if self.passes_left == 0:
                self.abort()
                return
            else:
                self.passes_left -= 1
                self.current_phase.reset()
                self.current_phase = self.phase_list[0]

    def advance_to_next_phase(self):
        current_index = self.current_phase_number()
        if current_index < len(self.phase_list) - 1:
            # Move to the next phase in the sequence
            self.current_phase.reset()
            self.current_phase = self.phase_list[current_index + 1]
        else:
            # Completed a full sequence; check if more repetitions are needed
            self.passes_left -= 1
            if self.passes_left < 1:
                self.state = "completed"
            else:
                self.current_phase.reset()
                self.current_phase = self.phase_list[0]  # Reset to the first phase

    def tick(self, ticks):
        print("Before tick")
        print(self.current_phase)
        print("Tick:", ticks)
        if not self.completed():
            result = self.current_phase.tick(ticks)
            print("Result", result, "\n")
            if self.current_phase.completed():
                print(self.current_phase.title, "Phase Completed")
                self.advance_to_next_phase()
                print("Advanced to", self.current_phase, "\n")
                result = self.tick(abs(result))
                return
        if self.completed():
            self.advance_to_next_phase()
        print("Passes left:", self.passes_left)

    def unrolled(self):
        return [deepcopy(seq) for seq in [each for each in self.initial_repetitions * self.phase_list]]