add name, parameters fields to Workflow and WorkflowDefinition

Author: andrii-i

jupyter_scheduler/orm.py

@@ -116,6 +116,8 @@ class Workflow(Base):
     tasks = Column(JsonType)
     status = Column(String(64), default=Status.CREATED)
     active = Column(Boolean, default=False)
+    name = Column(String(256))
+    parameters = Column(JsonType(1024))
     # All new columns added to this table must be nullable to ensure compatibility during database migrations.
     # Any default values specified for new columns will be ignored during the migration process.
 
@@ -129,6 +131,8 @@ class WorkflowDefinition(Base):
     active = Column(Boolean, default=False)
     schedule = Column(String(256))
     timezone = Column(String(36))
+    name = Column(String(256))
+    parameters = Column(JsonType(1024))
     # All new columns added to this table must be nullable to ensure compatibility during database migrations.
     # Any default values specified for new columns will be ignored during the migration process.
 

jupyter_scheduler/workflows.py

@@ -1,5 +1,5 @@
 import json
-from typing import List, Optional
+from typing import Dict, List, Optional
 
 from jupyter_server.utils import ensure_async
 from tornado.web import HTTPError, authenticated
@@ -291,9 +291,13 @@ async def post(self, workflow_definition_id: str):
 
 class CreateWorkflow(BaseModel):
     tasks: List[str] = []
+    name: str
+    parameters: Optional[Dict[str, str]] = None
 
 
 class DescribeWorkflow(BaseModel):
+    name: str
+    parameters: Optional[Dict[str, str]] = None
     workflow_id: str
     tasks: List[str] = None
     status: Status = Status.CREATED
@@ -304,6 +308,8 @@ class Config:
 
 
 class UpdateWorkflow(BaseModel):
+    name: str
+    parameters: Optional[Dict[str, str]] = None
     tasks: Optional[List[str]] = None
     status: Optional[Status] = None
     active: Optional[bool] = None
@@ -315,6 +321,8 @@ class Config:
 class CreateWorkflowDefinition(BaseModel):
     tasks: List[str] = []
     # any field added to CreateWorkflow should also be added to this model as well
+    name: str
+    parameters: Optional[Dict[str, str]] = None
     schedule: Optional[str] = None
     timezone: Optional[str] = None
 
@@ -323,6 +331,8 @@ class Config:
 
 
 class DescribeWorkflowDefinition(BaseModel):
+    name: str
+    parameters: Optional[Dict[str, str]] = None
     workflow_definition_id: str
     tasks: List[str] = None
     schedule: Optional[str] = None
@@ -335,6 +345,8 @@ class Config:
 
 
 class UpdateWorkflowDefinition(BaseModel):
+    name: str
+    parameters: Optional[Dict[str, str]] = None
     tasks: Optional[List[str]] = None
     schedule: Optional[str] = None
     timezone: Optional[str] = None