Skip to content

Commit 0be8ae6

Browse files
committed
Try to update extend_parameters_schema to use a msgspec schema
1 parent 39a77cb commit 0be8ae6

1 file changed

Lines changed: 9 additions & 9 deletions

File tree

taskcluster/mozillavpn_taskgraph/parameters.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,26 +3,26 @@
33
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
44

55
import os
6+
from typing import Annotated, Required
67

7-
from taskgraph.parameters import extend_parameters_schema
8-
from voluptuous import All, Any, Range, Required
8+
import msgspec
99

10+
from taskgraph.parameters import extend_parameters_schema
11+
from taskgraph.util.schema import Schema
1012

1113
def get_defaults(repo_root):
1214
return {
1315
"pull_request_number": None,
1416
"version": "",
1517
}
1618

17-
18-
extend_parameters_schema(
19+
CustomParametersSchema = Schema.from_dict(
1920
{
20-
Required("pull_request_number"): Any(All(int, Range(min=1)), None),
21-
Required("version"): str,
22-
},
23-
defaults_fn=get_defaults,
21+
"pull_request_number": Required[Annotated[int, msgspec.Meta(ge=1)]],
22+
"version": Required[str]
23+
}
2424
)
25-
25+
extend_parameters_schema(CustomParametersSchema, defaults_fn=get_defaults)
2626

2727
def get_decision_parameters(graph_config, parameters):
2828
head_tag = parameters["head_tag"]

0 commit comments

Comments
 (0)