Skip to content

Commit c74ca10

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

1 file changed

Lines changed: 8 additions & 10 deletions

File tree

taskcluster/mozillavpn_taskgraph/parameters.py

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -3,26 +3,24 @@
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

19+
class MozillaVpnParametersSchema(Schema, kw_only=True, rename=None):
20+
pull_request_number: int
21+
version: str
1722

18-
extend_parameters_schema(
19-
{
20-
Required("pull_request_number"): Any(All(int, Range(min=1)), None),
21-
Required("version"): str,
22-
},
23-
defaults_fn=get_defaults,
24-
)
25-
23+
extend_parameters_schema(MozillaVpnParametersSchema, defaults_fn=get_defaults)
2624

2725
def get_decision_parameters(graph_config, parameters):
2826
head_tag = parameters["head_tag"]

0 commit comments

Comments
 (0)