uYfddlZddlmZddlmZddlmZddlmZddlmZddlZddl m Z  dd lm Z e eZej d Gd deZy#e $r dd lm Z Y7wxYw)N)deepcopy)Any)Callable)Optional)Set) get_logger) DefaultDict) defaultdictT)slotscheZdZdZej ddeeefZ d dZ d dZ e Z dZ d Zy) Hooksz Hooks configuration object is used for registering and calling hook functions Example:: @config.falcon.hooks.on('request') def on_request(span, request, response): pass Fc4tjtSN) collectionsr setN/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/ddtrace/_hooks.pyzHooks.!s1H1H1Mr)initfactorytypeNcPt}t|j||_|Sr)rr_hooks)selfmemodicthookss r __deepcopy__zHooks.__deepcopy__#s  X6  rcX|sfd}|Sjj|y)a Function used to register a hook for the provided name. Example:: def on_request(span, request, response): pass config.falcon.hooks.register('request', on_request) If no function is provided then a decorator is returned:: @config.falcon.hooks.register('request') def on_request(span, request, response): pass :param hook: The name of the hook to register the function for :type hook: object :param func: The function to register, or ``None`` if a decorator should be returned :type func: function, None :returns: Either a function decorator if ``func is None``, otherwise ``None`` :rtype: function, None c,j||Sr)register)funchookrs rwrapperzHooks.register..wrapperIs dD) rN)radd)rr$r#r%s`` rr"zHooks.register(s.> N Dd#rc~||jvr |j|j|yy#t$rYywxYw)a Function to deregister a function from a hook it was registered under Example:: @config.falcon.hooks.on('request') def on_request(span, request, response): pass config.falcon.hooks.deregister('request', on_request) :param hook: The name of the hook to register the function for :type hook: object :param func: Function hook to register :type func: function N)rremoveKeyError)rr$r#s r deregisterzHooks.deregisterWsE, 4;;   D!((.   s 0 <<c|jj|dD] } ||i| y#t$rtj d||dY3wxYw)aR Function used to call registered hook functions. :param hook: The hook to call functions for :type hook: str :param args: Positional arguments to pass to the hook functions :type args: list :param kwargs: Keyword arguments to pass to the hook functions :type kwargs: dict rz!Failed to run hook %s function %sT)exc_infoN)rget Exceptionlogerror)rr$argskwargsr#s remitz Hooks.emitss`$KKOOD"- ZD Zd%f% Z Z =tTTX Y Zs,"AAr)__name__ __module__ __qualname____doc__attribr strrrrr"onr*r3rrrrrsMTWW%)MT_`ceh`hTi jF'Z B8Zrr)rcopyrtypingrrrrr8internal.loggerr r ImportErrorr r4r/sobjectrrrrrBsq '7"dsZFsZsZ767sA A,+A,