-
Notifications
You must be signed in to change notification settings - Fork 1
/
__init__.py
55 lines (47 loc) · 1.74 KB
/
__init__.py
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
from retrack.nodes.base import BaseNode
from retrack.nodes.check import Check
from retrack.nodes.constants import Bool, Constant, IntervalCatV0, List
from retrack.nodes.contains import Contains
from retrack.nodes.datetime import CurrentYear
from retrack.nodes.dynamic import BaseDynamicNode
from retrack.nodes.dynamic import registry as dynamic_registry
from retrack.nodes.endswith import EndsWith
from retrack.nodes.endswithany import EndsWithAny
from retrack.nodes.inputs import Input
from retrack.nodes.logic import And, Not, Or
from retrack.nodes.match import If
from retrack.nodes.math import AbsoluteValue, Math, Round
from retrack.nodes.outputs import Output
from retrack.nodes.start import Start
from retrack.nodes.startswith import StartsWith
from retrack.nodes.startswithany import StartsWithAny
from retrack.nodes.lowercase import LowerCase
from retrack.utils.registry import Registry
_registry = Registry()
def registry() -> Registry:
return _registry
def register(name: str, node: BaseNode) -> None:
registry().register(name, node)
register("Input", Input)
register("Start", Start)
register("Constant", Constant)
register("List", List)
register("Bool", Bool)
register("Output", Output)
register("Check", Check)
register("If", If)
register("And", And)
register("Or", Or)
register("Not", Not)
register("Math", Math)
register("Round", Round)
register("AbsoluteValue", AbsoluteValue)
register("StartsWith", StartsWith)
register("EndsWith", EndsWith)
register("StartsWithAny", StartsWithAny)
register("EndsWithAny", EndsWithAny)
register("Contains", Contains)
register("CurrentYear", CurrentYear)
register("IntervalCatV0", IntervalCatV0)
register("LowerCase", LowerCase)
__all__ = ["registry", "register", "BaseNode", "dynamic_registry", "BaseDynamicNode"]