eienmojiki commited on
Commit
4c61c0c
·
verified ·
1 Parent(s): b8f7d24

Update registry.py

Browse files
Files changed (1) hide show
  1. 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, **defaults):
 
 
 
 
 
 
 
 
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