"""
Description :   description
Author      :   Ruidi Qiu (r.qiu@tum.de)
Time        :   2024/3/23 11:30:00
LastEdited  :   2024/7/24 11:43:13
"""

from .base_script import BaseScript, BaseScriptStage
from .script_pychecker import WF_pychecker
from .script_directgen import WF_directgen

SCRIPTS_SELECTER = {
    "pychecker": WF_pychecker,
    "directgen": WF_directgen
}

def get_script(script_name:str) -> BaseScript:
    if script_name in SCRIPTS_SELECTER:
        return SCRIPTS_SELECTER[script_name]
    else:
        raise ValueError(f"script name {script_name} is not supported")