grortir.main.model.processes.factories.CallsProcessFactory

Complexity

Total Complexity

6

Size/Duplication

Total Lines	33
Duplicated Lines	0 %

Test Coverage

Coverage

100%

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	6	1
A	_create_stages()	0	4	2
A	construct_process()	0	5	1
A	_update_structure()	0	6	2

"""Contains factory class of process with calls stages."""

from grortir.main.model.processes.calls_process import CallsProcess
from grortir.main.model.stages.calls_stage import CallsStage


class CallsProcessFactory:
    """Class which produce process with the same type of stages.

    Attributes:
        structure_type (str): represent type of structure
        how_many_nodes (int): how many nodes should be in generated process
        process (CallsProcess): process which will be incrementally build
    """

    def __init__(self, structure_type, how_many_nodes):
        """Constructor."""
        self._stages_to_add = []
        self.structure_type = structure_type
        self.how_many_nodes = how_many_nodes
        self.process = CallsProcess()

    def _update_structure(self):
        """Update structure of process."""
        if self.structure_type == "linear":
            self.process.add_path(self._stages_to_add)
        else:
            raise NotImplementedError()

    def _create_stages(self):
        """Create stages which will be injected to process."""
        for i in range(self.how_many_nodes):
            self._stages_to_add += [CallsStage(str(i))]

    def construct_process(self):
        """Construct process."""
        self._create_stages()
        self._update_structure()
        return self.process