tor-browser

commit bb3d84f3e03c5e78154937a3517ab50056b37732
parent 08c6e2464d7ac105490e601eec1390e6edef190b
Author: Julien Cristau <jcristau@mozilla.com>
Date:   Mon,  6 Oct 2025 11:31:11 +0000

Bug 1992386 - set task expiration policy by level. r=taskgraph-reviewers,bhearsum

Differential Revision: https://phabricator.services.mozilla.com/D267403

Diffstat:
M
taskcluster/config.yml
 | 
6
+++---
M
taskcluster/gecko_taskgraph/config.py
 | 
4
+++-
M
taskcluster/gecko_taskgraph/transforms/job/common.py
 | 
2
+-
M
taskcluster/gecko_taskgraph/transforms/task.py
 | 
9
++++++---

4 files changed, 13 insertions(+), 8 deletions(-)
diff --git a/taskcluster/config.yml b/taskcluster/config.yml
@@ -961,14 +961,14 @@ mac-signing:
                       - "/"  # The .app
 
 expiration-policy:
-    by-project:
-        try:
+    by-level:
+        '1':
             default: 28 days
             shortest: 7 days
             short: 14 days
             medium: 28 days
             long: 28 days
-        autoland:
+        '3':
             default: 1 year
             shortest: 14 days
             short: 3 months
diff --git a/taskcluster/gecko_taskgraph/config.py b/taskcluster/gecko_taskgraph/config.py
@@ -125,6 +125,8 @@ graph_config_schema = Schema(
                 Length(min=1),
             ),
         },
-        Required("expiration-policy"): optionally_keyed_by("project", {str: str}),
+        Required("expiration-policy"): optionally_keyed_by(
+            "project", "level", {str: str}
+        ),
     }
 )
diff --git a/taskcluster/gecko_taskgraph/transforms/job/common.py b/taskcluster/gecko_taskgraph/transforms/job/common.py
@@ -215,6 +215,6 @@ def get_expiration(config, policy="default"):
     expires = evaluate_keyed_by(
         config.graph_config["expiration-policy"],
         "artifact expiration",
-        {"project": config.params["project"]},
+        {"project": config.params["project"], "level": config.params["level"]},
     )[policy]
     return expires
diff --git a/taskcluster/gecko_taskgraph/transforms/task.py b/taskcluster/gecko_taskgraph/transforms/task.py
@@ -2138,9 +2138,12 @@ def set_task_and_artifact_expiry(config, jobs):
     cap = "28 days" if is_try(config.params) else None
     cap_from_now = fromNow(cap, now) if cap else None
     if cap:
-        for policy, expires in config.graph_config["expiration-policy"]["by-project"][
-            "try"
-        ].items():
+        expiration_policy = evaluate_keyed_by(
+            config.graph_config["expiration-policy"],
+            "task expiration",
+            {"project": config.params["project"], "level": config.params["level"]},
+        )
+        for policy, expires in expiration_policy.items():
             if fromNow(expires, now) > cap_from_now:
                 raise Exception(
                     f'expiration-policy "{policy}" is larger than {cap} '