summaryrefslogtreecommitdiff
path: root/continuedev/src/continuedev/steps/steps_on_startup.py
blob: 365cbe1a85bf769fb3ad746a7ca5cac5d45a1697 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from ..core.main import Step
from ..core.sdk import Models, ContinueSDK
from .main import UserInputStep
from ..recipes.CreatePipelineRecipe.main import CreatePipelineRecipe
from ..recipes.DDtoBQRecipe.main import DDtoBQRecipe
from ..recipes.DeployPipelineAirflowRecipe.main import DeployPipelineAirflowRecipe
from ..recipes.DDtoBQRecipe.main import DDtoBQRecipe
from ..recipes.AddTransformRecipe.main import AddTransformRecipe
from ..libs.util.step_name_to_steps import get_step_from_name


class StepsOnStartupStep(Step):
    hide: bool = True

    async def describe(self, models: Models):
        return "Running steps on startup"

    async def run(self, sdk: ContinueSDK):
        steps_on_startup = sdk.config.steps_on_startup

        for step_name, step_params in steps_on_startup.items():
            step = get_step_from_name(step_name, step_params)
            await sdk.run_step(step)