a D øg ã@sjddlmZddlmZddlmZiZddd„ZGdd„deƒZddd œd d „Z d a d d„Z dd„Z dS)é)ÚContextDecorator)Ú itemgetter)Úget_app_submodulesNcs@|dur‡‡fdd„}|Sˆtvr*gtˆ<tˆ |ˆf¡dS)a Register hook for ``hook_name``. Can be used as a decorator:: @register('hook_name') def my_hook(...): pass or as a function call:: def my_hook(...): pass register('hook_name', my_hook) Ncstˆ|ˆd|S)N©Úorder)Úregister)Úfn©Ú hook_namer©úL/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/wagtail/hooks.pyÚ decoratorszregister..decorator)Ú_hooksÚappend)r rrr r r r r s rc@s$eZdZdd„Zdd„Zdd„ZdS)Ú TemporaryHookcCs||_||_dS©N)Úhooksr)Úselfrrr r r Ú__init__'szTemporaryHook.__init__cCs8|jD],\}}|tvrgt|<t| ||jf¡qdSr)rrrr)rr rr r r Ú __enter__+szTemporaryHook.__enter__cCs(|jD]\}}t| ||jf¡qdSr)rrÚremover)rÚexc_typeÚ exc_valueÚ tracebackr rr r r Ú__exit__1szTemporaryHook.__exit__N)Ú__name__Ú __module__Ú __qualname__rrrr r r r r&srrcCs,t|tƒs|dur||fg}n|}t||ƒS)a} Register hook for ``hook_name`` temporarily. This is useful for testing hooks. Can be used as a decorator:: def my_hook(...): pass class TestMyHook(Testcase): @hooks.register_temporarily('hook_name', my_hook) def test_my_hook(self): pass or as a context manager:: def my_hook(...): pass with hooks.register_temporarily('hook_name', my_hook): # Hook is registered here # Hook is unregistered here To register multiple hooks at the same time, pass in a list of 2-tuples: def my_hook(...): pass def my_other_hook(...): pass with hooks.register_temporarily([ ('hook_name', my_hook), ('hook_name', my_other_hook), ]): # Hooks are registered here N)Ú isinstanceÚlistr)Zhook_name_or_hooksrrrr r r Úregister_temporarily6s& r FcCststtdƒƒdadS)NZ wagtail_hooksT)Ú_searched_for_hooksrrr r r r Úsearch_for_hooksgs r"cCs0tƒt |g¡}t|tdƒd}dd„|DƒS)z0Return the hooks function sorted by their order.é)ÚkeycSsg|] }|d‘qS)rr )Ú.0Úhookr r r Ú sózget_hooks..)r"rÚgetÚsortedr)r rr r r Ú get_hooksns r+)Nr)N) Ú contextlibrÚoperatorrZwagtail.utils.appsrrrrr r!r"r+r r r r Ús    .