from pyboot.application import PybootApplication
from runners.task_generator import TaskGenerator
from runners.data_generator import DataGenerator
from runners.data_recorder import DataRecorder
from runners.task_templates_divider import TaskTemplatesDivider

@PybootApplication("main")
class MainApp:
    @staticmethod
    def start():
        TaskGenerator(config_path="configs/generate_task_config.yaml").run()
        DataGenerator(config_path="configs/generate_data_config.yaml").run()
        DataRecorder(config_path="configs/record_data_config.yaml").run()

@PybootApplication("divide_task")
class DivideTaskApp:
    @staticmethod
    def start():
        TaskTemplatesDivider(config_path="configs/divide_task_config.yaml").run()

@PybootApplication("generate_task")
class GenerateTaskApp:
    @staticmethod
    def start():
        TaskGenerator(config_path="configs/generate_task_config.yaml").run()

@PybootApplication("generate_data")
class GenerateDataApp:
    @staticmethod
    def start():
        DataGenerator(config_path="configs/generate_data_config.yaml").run()

@PybootApplication("record_data")
class RecordDataApp:
    @staticmethod
    def start():
        DataRecorder(config_path="configs/record_data_config.yaml").run()