rpc-python-toolkit/examples/schema_validation.py at main · n-car/rpc-python-toolkit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
from __future__ import annotations

import json

from rpc_python_toolkit import RpcEndpoint


rpc = RpcEndpoint()

rpc.add_method(
    "math.add",
    lambda _request, _context, params: params["a"] + params["b"],
    schema={
        "type": "object",
        "required": ["a", "b"],
        "properties": {
            "a": {"type": "number"},
            "b": {"type": "number"},
        },
        "additionalProperties": False,
    },
)

valid = rpc.handle_payload(
    {
        "jsonrpc": "2.0",
        "method": "math.add",
        "params": {"a": 2, "b": 3},
        "id": 1,
    }
)

invalid = rpc.handle_payload(
    {
        "jsonrpc": "2.0",
        "method": "math.add",
        "params": {"a": 2},
        "id": 2,
    }
)

print(
    json.dumps(
        {
            "valid": valid.body,
            "invalid": invalid.body,
        },
        indent=2,
    )
)