revil-O/platypush
1
0
Fork 0
forked from platypush/platypush
Code Pull requests Activity

Allow for custom procedure names on the @procedure decorator.

Browse source 
```
@procedure("foo")
def bar():
    ...
```

Will now be published as `procedure.foo` instead of
`procedure.<module>.bar`.
This commit is contained in:
Fabio Manganiello 2024-05-24 20:07:24 +02:00
parent cbc58c7330
commit 5c2204f99d
Signed by untrusted user: blacklight
GPG key ID: D90FBA7F76362774
2 changed files with 15 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
platypush/config/__init__.py
View file

@ -355,7 +355,7 @@ class Config:
prefix = modname + '.' if prefix is None else prefix prefix = modname + '.' if prefix is None else prefix
self.procedures.update( self.procedures.update(
**{ **{
prefix + name: obj getattr(obj, 'procedure_name', f'{prefix}.{name}'): obj
for name, obj in inspect.getmembers(module) for name, obj in inspect.getmembers(module)
if is_functional_procedure(obj) if is_functional_procedure(obj)
} }

7
platypush/procedure/__init__.py
View file

@ -4,6 +4,7 @@ import re
from functools import wraps from functools import wraps
from queue import LifoQueue from queue import LifoQueue
from typing import Optional
from ..common import exec_wrapper from ..common import exec_wrapper
from ..config import Config from ..config import Config
@ -557,12 +558,14 @@ class IfProcedure(Procedure):
return response return response
def procedure(f): def procedure(name: Optional[str] = None):
def func_wrapper(f):
""" """
Public decorator to mark a function as a procedure. Public decorator to mark a function as a procedure.
""" """
f.procedure = True f.procedure = True
f.procedure_name = name
@wraps(f) @wraps(f)
def _execute_procedure(*args, **kwargs): def _execute_procedure(*args, **kwargs):
@ -570,5 +573,7 @@ def procedure(f):
return _execute_procedure return _execute_procedure
return func_wrapper
# vim:sw=4:ts=4:et: # vim:sw=4:ts=4:et: