add missing CronWorkflow message object

jupyter-naas · Jan 12, 2024 · 896199c · 896199c
1 parent c89b948
commit 896199c
Show file tree

Hide file tree

Showing 3 changed files with 92 additions and 31 deletions.
diff --git a/protos/workflow.proto b/protos/workflow.proto
@@ -95,6 +95,22 @@ message Workflow {
     Spec spec = 2;
 }
 
+message CronSpec {
+    string schedule = 1;
+    string timezone = 2;
+    optional string startingDeadlineSeconds = 3;
+    optional string concurrencyPolicy = 4;
+    optional string successfulJobsHistoryLimit = 5;
+    optional string failedJobsHistoryLimit = 6;
+    optional string suspend = 7;
+    Spec workflowSpec = 8;
+}
+
+message CronWorkflow {
+    Metadata metadata = 1;
+    CronSpec spec = 2;
+}
+
 /**
 *   Argo workflow CRUD resources
 */

diff --git a/python/naas_models/pydantic/workflow_p2p.py b/python/naas_models/pydantic/workflow_p2p.py
@@ -166,6 +166,31 @@ class Workflow(BaseModel):
 
 
 
+class CronSpec(BaseModel):
+
+    _one_of_dict = {"CronSpec._concurrencyPolicy": {"fields": {"concurrencyPolicy"}}, "CronSpec._failedJobsHistoryLimit": {"fields": {"failedJobsHistoryLimit"}}, "CronSpec._startingDeadlineSeconds": {"fields": {"startingDeadlineSeconds"}}, "CronSpec._successfulJobsHistoryLimit": {"fields": {"successfulJobsHistoryLimit"}}, "CronSpec._suspend": {"fields": {"suspend"}}}
+    _check_one_of = root_validator(pre=True, allow_reuse=True)(check_one_of)
+
+    schedule: str = FieldInfo(default="") 
+    timezone: str = FieldInfo(default="") 
+    startingDeadlineSeconds: str = FieldInfo(default="") 
+    concurrencyPolicy: str = FieldInfo(default="") 
+    successfulJobsHistoryLimit: str = FieldInfo(default="") 
+    failedJobsHistoryLimit: str = FieldInfo(default="") 
+    suspend: str = FieldInfo(default="") 
+    workflowSpec: Spec = FieldInfo() 
+
+
+
+
+class CronWorkflow(BaseModel):
+
+    metadata: Metadata = FieldInfo() 
+    spec: CronSpec = FieldInfo() 
+
+
+
+
 class WorkflowCreationRequest(BaseModel):
 
     _one_of_dict = {"WorkflowCreationRequest._description": {"fields": {"description"}}, "WorkflowCreationRequest._name": {"fields": {"name"}}, "WorkflowCreationRequest._namespace": {"fields": {"namespace"}}, "WorkflowCreationRequest._serverDryRun": {"fields": {"serverDryRun"}}, "WorkflowCreationRequest._user_uid": {"fields": {"user_uid"}}, "WorkflowCreationRequest._workflow": {"fields": {"workflow"}}}