sosaku-py 0.6.2

Python bindings for the sosaku JSON Query language
Documentation 
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

import typing as t

import sosaku

JSONValue = int | float | str | bool | None | t.Sequence["JSONValue"] | t.Mapping[str, "JSONValue"]

CURSED_STUFF = """
    len(y) == 5 &&
    startsWith(y, "he") &&
    endsWith(y, "lo") &&
    contains(y, "ell") &&
    matches(y, r"^h.*o$") &&
    foo.bar > 100 &&
    foo.bar >= 123 &&
    foo.bar <= 123 &&
    foo.bar != 0 &&
    foo.baz == "world" &&
    contains(foo, "bar") &&
    !contains(foo, "missing") &&
    foo.qux.nested[0] == 1 &&
    foo.qux.nested[1] == 2 &&
    foo.qux.nested[2] == 3 &&
    contains(foo.qux.nested, 2) &&
    !contains(foo.qux.nested, 42) &&
    foo.qux.nested == [1, 2, 3] &&
    (x > 0 && x < 100 && x >= 42 && x <= 42 && x == 42 && x != 0) &&
    (z && !false && (false || true)) &&
    (("" || y) == "hello") &&
    ({"bar": 123, "baz": "world", "qux": {"nested": [1, 2, 3]}} == foo) &&
    len(foo.qux.nested) == 3 &&
    contains(["a", "b", "c"], "b") &&
    (1 < 2 && 2 < 3 && 3 <= 3 && 4 > 3 && 4 >= 4) &&
    matches("abc123", r"^[a-z]+[0-9]+$") &&
    contains({"k": 1, "m": 2}, "k")
    """


def test_exp_parse():
    exp = sosaku.Exp(CURSED_STUFF)
    assert exp


def test_exp_eval():
    test_json: dict[str, JSONValue] = {
        "x": 42,
        "y": "hello",
        "z": True,
        "foo": {"bar": 123, "baz": "world", "qux": {"nested": [1, 2, 3]}},
    }

    exp = sosaku.Exp(CURSED_STUFF)
    assert exp.eval(test_json) is True


def test_var_access():
    test_json: dict[str, JSONValue] = {
        "x": 42,
        "y": "hello",
        "z": True,
        "foo": {"bar": 123, "baz": "world", "qux": {"nested": [1, 2, 3]}},
    }

    var_access = sosaku.VarAccess("foo.bar")
    assert var_access.access(test_json) == 123


def test_var_replace():
    test_json: dict[str, JSONValue] = {
        "x": 42,
        "y": "hello",
        "z": True,
        "foo": {"bar": 123, "baz": "world", "qux": {"nested": [1, 2, 3]}},
    }

    var_access = sosaku.VarAccess("foo.bar")
    test_json = var_access.replace(test_json, {"new": "value"})
    assert var_access.access(test_json) == {"new": "value"}
    assert test_json["foo"]["bar"] == {"new": "value"}


def test_var_replace_retains_order():
    test_json: dict[str, JSONValue] = {
        "x": 42,
        "a": "hello",
        "b": True,
        "foo": {"bar": 123, "baz": "world", "qux": {"nested": [1, 2, 3]}},
    }

    var_access = sosaku.VarAccess("x")
    test_json = var_access.replace(test_json, True)
    assert var_access.access(test_json) is True
    assert list(test_json.keys()) == ["x", "a", "b", "foo"]