fix: handle metadata attempt/dispatch == 0

flamingo-run · May 23, 2023 · 495565f · 495565f
1 parent 0644cc4
commit 495565f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/django_cloud_tasks/tasks/task.py b/django_cloud_tasks/tasks/task.py
@@ -45,12 +45,12 @@ def from_headers(cls, headers: dict) -> Self:
         # Available data: https://cloud.google.com/tasks/docs/creating-http-target-tasks#handler
         cloud_tasks_prefix = "X-Cloudtasks-"
 
-        if attempt_str := headers.get(f"{cloud_tasks_prefix}Taskexecutioncount"):
+        if (attempt_str := headers.get(f"{cloud_tasks_prefix}Taskexecutioncount")) is not None:
             execution_number = int(attempt_str)
         else:
             execution_number = None
 
-        if retry_str := headers.get(f"{cloud_tasks_prefix}Taskretrycount"):
+        if retry_str := headers.get(f"{cloud_tasks_prefix}Taskretrycount") is not None:
             dispatch_number = int(retry_str)
         else:
             dispatch_number = None