From ea7031e5426b655ba2b8a204bebac66c48ef6c6d Mon Sep 17 00:00:00 2001 From: Bruno Almeida Date: Thu, 3 Jul 2025 15:06:47 -0300 Subject: [PATCH] Use cronjob type correctly --- kubernetes/workloads.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/kubernetes/workloads.py b/kubernetes/workloads.py index a104667..770cb3f 100644 --- a/kubernetes/workloads.py +++ b/kubernetes/workloads.py @@ -575,11 +575,10 @@ def body(self): workload = StatefulSet(name=name, config=config.model_dump()) elif config.type == WorkloadTypes.DAEMONSET: workload = DaemonSet(name=name, config=config.model_dump()) + elif config.type == WorkloadTypes.CRONJOB: + workload = CronJob(name=name, config=config.model_dump()) elif config.type == WorkloadTypes.JOB: - if config.schedule: - workload = CronJob(name=name, config=config.model_dump()) - else: - workload = Job(name=name, config=config.model_dump()) + workload = Job(name=name, config=config.model_dump()) elif config.type == WorkloadTypes.CLOUD_RUN_SERVICE: workload = CloudRunService(name=name, config=config.model_dump()) else: @@ -637,8 +636,8 @@ def body(self): self._add_component(FrontendConfig, "frontend_config", spec=self.config.frontend_config) - # Handling a special case where pdb_min_available or auto_pdb is set, but config.type isn't "job" - if self.config.type != "job" and ( + # Handling a special case where pdb_min_available or auto_pdb is set, but config.type isn't "job" or "cronjob" + if (self.config.type != WorkloadTypes.JOB and self.config.type != WorkloadTypes.CRONJOB) and ( self.config.pdb_min_available or self.config.auto_pdb ): config_attr = "pdb_min_available" if self.config.pdb_min_available else "auto_pdb"