diff --git a/jupyter_scheduler/extension.py b/jupyter_scheduler/extension.py index 49a5d9c0..70e732b7 100644 --- a/jupyter_scheduler/extension.py +++ b/jupyter_scheduler/extension.py @@ -21,6 +21,7 @@ JOB_DEFINITION_ID_REGEX = r"(?P\w+(?:-\w+)+)" JOB_ID_REGEX = r"(?P\w+(?:-\w+)+)" +WORKFLOW_ID_REGEX = r"(?P\w+(?:-\w+)+)" class SchedulerApp(ExtensionApp): @@ -37,6 +38,9 @@ class SchedulerApp(ExtensionApp): (r"scheduler/runtime_environments", RuntimeEnvironmentsHandler), (r"scheduler/config", ConfigHandler), (r"scheduler/worklows", WorkflowHandler), + (r"scheduler/worklows/%s/run" % (WORKFLOW_ID_REGEX, JOB_ID_REGEX), WorkflowHandler), + (r"scheduler/worklows/%s/jobs" % WORKFLOW_ID_REGEX, WorkflowHandler), + (r"scheduler/worklows/%s/jobs/%s" % (WORKFLOW_ID_REGEX, JOB_ID_REGEX), WorkflowHandler), ] drop_tables = Bool(False, config=True, help="Drop the database tables before starting.")