Spaces:
Running
Running
Create registry.py
Browse files- registry.py +25 -0
registry.py
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from functools import wraps
|
2 |
+
import inspect
|
3 |
+
|
4 |
+
class FilterRegistry:
|
5 |
+
def __init__(self):
|
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)
|
13 |
+
params = {}
|
14 |
+
for param in sig.parameters.values():
|
15 |
+
if param.name == 'image':
|
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
|
23 |
+
return decorator
|
24 |
+
|
25 |
+
registry = FilterRegistry()
|