Spaces:
Sleeping
Sleeping
Update registry.py
Browse files- registry.py +13 -2
registry.py
CHANGED
|
@@ -6,7 +6,15 @@ class FilterRegistry:
|
|
| 6 |
self.filters = {}
|
| 7 |
self.params_map = {}
|
| 8 |
|
| 9 |
-
def register(self, name,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
def decorator(func):
|
| 11 |
self.filters[name] = func
|
| 12 |
sig = inspect.signature(func)
|
|
@@ -16,7 +24,10 @@ class FilterRegistry:
|
|
| 16 |
continue
|
| 17 |
params[param.name] = {
|
| 18 |
'type': param.annotation,
|
| 19 |
-
'default': param.default if param.default != inspect.Parameter.empty else defaults.get(param.name)
|
|
|
|
|
|
|
|
|
|
| 20 |
}
|
| 21 |
self.params_map[name] = params
|
| 22 |
return func
|
|
|
|
| 6 |
self.filters = {}
|
| 7 |
self.params_map = {}
|
| 8 |
|
| 9 |
+
def register(self, name, defaults=None, min_vals=None, max_vals=None, step_vals=None):
|
| 10 |
+
if defaults is None:
|
| 11 |
+
defaults = {}
|
| 12 |
+
if min_vals is None:
|
| 13 |
+
min_vals = {}
|
| 14 |
+
if max_vals is None:
|
| 15 |
+
max_vals = {}
|
| 16 |
+
if step_vals is None:
|
| 17 |
+
step_vals = {}
|
| 18 |
def decorator(func):
|
| 19 |
self.filters[name] = func
|
| 20 |
sig = inspect.signature(func)
|
|
|
|
| 24 |
continue
|
| 25 |
params[param.name] = {
|
| 26 |
'type': param.annotation,
|
| 27 |
+
'default': param.default if param.default != inspect.Parameter.empty else defaults.get(param.name),
|
| 28 |
+
'min': min_vals.get(param.name),
|
| 29 |
+
'max': max_vals.get(param.name),
|
| 30 |
+
'step': step_vals.get(param.name)
|
| 31 |
}
|
| 32 |
self.params_map[name] = params
|
| 33 |
return func
|