-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy pathcomprehension.py
More file actions
120 lines (98 loc) · 5.81 KB
/
comprehension.py
File metadata and controls
120 lines (98 loc) · 5.81 KB
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
# coding: utf-8
"""
Permify API
Permify is an open source authorization service for creating fine-grained and scalable authorization systems.
The version of the OpenAPI document: v1.4.3
Contact: hello@permify.co
Generated by OpenAPI Generator (https://openapi-generator.tech)
Do not edit the class manually.
""" # noqa: E501
from __future__ import annotations
import pprint
import re # noqa: F401
import json
from typing import Any, ClassVar, Dict, List, Optional
from pydantic import BaseModel, StrictStr
from pydantic import Field
try:
from typing import Self
except ImportError:
from typing_extensions import Self
class Comprehension(BaseModel):
"""
A comprehension expression applied to a list or map. Comprehensions are not part of the core syntax, but enabled with macros. A macro matches a specific call signature within a parsed AST and replaces the call with an alternate AST block. Macro expansion happens at parse time. The following macros are supported within CEL: Aggregate type macros may be applied to all elements in a list or all keys in a map: * `all`, `exists`, `exists_one` - test a predicate expression against the inputs and return `true` if the predicate is satisfied for all, any, or only one value `list.all(x, x < 10)`. * `filter` - test a predicate expression against the inputs and return the subset of elements which satisfy the predicate: `payments.filter(p, p > 1000)`. * `map` - apply an expression to all elements in the input and return the output aggregate type: `[1, 2, 3].map(i, i * i)`. The `has(m.x)` macro tests whether the property `x` is present in struct `m`. The semantics of this macro depend on the type of `m`. For proto2 messages `has(m.x)` is defined as 'defined, but not set`. For proto3, the macro tests whether the property is set to its default. For map and struct types, the macro tests whether the property `x` is defined on `m`.
""" # noqa: E501
iter_var: Optional[StrictStr] = Field(default=None, description="The name of the iteration variable.", alias="iterVar")
iter_range: Optional[Expr] = Field(default=None, alias="iterRange")
accu_var: Optional[StrictStr] = Field(default=None, description="The name of the variable used for accumulation of the result.", alias="accuVar")
accu_init: Optional[Expr] = Field(default=None, alias="accuInit")
loop_condition: Optional[Expr] = Field(default=None, alias="loopCondition")
loop_step: Optional[Expr] = Field(default=None, alias="loopStep")
result: Optional[Expr] = None
__properties: ClassVar[List[str]] = ["iterVar", "iterRange", "accuVar", "accuInit", "loopCondition", "loopStep", "result"]
model_config = {
"populate_by_name": True,
"validate_assignment": True,
"protected_namespaces": (),
}
def to_str(self) -> str:
"""Returns the string representation of the model using alias"""
return pprint.pformat(self.model_dump(by_alias=True))
def to_json(self) -> str:
"""Returns the JSON representation of the model using alias"""
# TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
return json.dumps(self.to_dict())
@classmethod
def from_json(cls, json_str: str) -> Self:
"""Create an instance of Comprehension from a JSON string"""
return cls.from_dict(json.loads(json_str))
def to_dict(self) -> Dict[str, Any]:
"""Return the dictionary representation of the model using alias.
This has the following differences from calling pydantic's
`self.model_dump(by_alias=True)`:
* `None` is only added to the output dict for nullable fields that
were set at model initialization. Other fields with value `None`
are ignored.
"""
_dict = self.model_dump(
by_alias=True,
exclude={
},
exclude_none=True,
)
# override the default output from pydantic by calling `to_dict()` of iter_range
if self.iter_range:
_dict['iterRange'] = self.iter_range.to_dict()
# override the default output from pydantic by calling `to_dict()` of accu_init
if self.accu_init:
_dict['accuInit'] = self.accu_init.to_dict()
# override the default output from pydantic by calling `to_dict()` of loop_condition
if self.loop_condition:
_dict['loopCondition'] = self.loop_condition.to_dict()
# override the default output from pydantic by calling `to_dict()` of loop_step
if self.loop_step:
_dict['loopStep'] = self.loop_step.to_dict()
# override the default output from pydantic by calling `to_dict()` of result
if self.result:
_dict['result'] = self.result.to_dict()
return _dict
@classmethod
def from_dict(cls, obj: Dict) -> Self:
"""Create an instance of Comprehension from a dict"""
if obj is None:
return None
if not isinstance(obj, dict):
return cls.model_validate(obj)
_obj = cls.model_validate({
"iterVar": obj.get("iterVar"),
"iterRange": Expr.from_dict(obj.get("iterRange")) if obj.get("iterRange") is not None else None,
"accuVar": obj.get("accuVar"),
"accuInit": Expr.from_dict(obj.get("accuInit")) if obj.get("accuInit") is not None else None,
"loopCondition": Expr.from_dict(obj.get("loopCondition")) if obj.get("loopCondition") is not None else None,
"loopStep": Expr.from_dict(obj.get("loopStep")) if obj.get("loopStep") is not None else None,
"result": Expr.from_dict(obj.get("result")) if obj.get("result") is not None else None
})
return _obj
from permify.models.expr import Expr
# TODO: Rewrite to not use raise_errors
Comprehension.model_rebuild(raise_errors=False)