mhuddlmZddlmZddlmZddlmZddlZddlZddlmZee Z dZ d e zZ Gd d e ZdS) ) TYPE_CHECKING)Any)Dict)OptionalN) get_logger _datadog_pin_self_ceZdZdZgdZ ddZedZfdZdZ e dZ e d Z e dd Zd Zdd ZdZ ddZxZS)PinaPin (a.k.a Patch INfo) is a small class which is used to set tracing metadata on a particular traced connection. This is useful if you wanted to, say, trace two different database clusters. >>> conn = sqlite.connect('/tmp/user.db') >>> # Override a pin for a specific connection >>> pin = Pin.override(conn, service='user-db') >>> conn = sqlite.connect('/tmp/image.db') )tagstracer_target_config _initializedNc|p tj}||_||_d|_|pi|_||jd<d|_dS)N service_nameT)ddtracerr rrr)selfservicer rrs 6/usr/local/lib/python3.11/site-packages/ddtrace/pin.py__init__z Pin.__init__$sL)7>   }" '. ^$ c|jdS)zoBackward compatibility: accessing to `pin.service` returns the underlying configuration value. r)rrs rrz Pin.service7s |N++rct|ddr|dkrtdtt|||dS)NrFrz5can't mutate a pin, use override() or clone() instead)getattrAttributeErrorsuperr __setattr__)rnamevalue __class__s rr zPin.__setattr__?sW 4 / / ZDI4E4E !XYY Y c4$$T511111rc8d|jd|jd|jdS)Nz Pin(service=z, tags=z , tracer=)rr rrs r__repr__z Pin.__repr__Ds'8< diiiQUQ\Q\Q\]]rcP|D]"}t|}|r|cS#dS)a Return the first :class:`ddtrace.pin.Pin` found on any of the provided objects or `None` if none were found >>> pin = Pin._find(wrapper, instance, conn) :param objs: The objects to search for a :class:`ddtrace.pin.Pin` on :type objs: List of objects :rtype: :class:`ddtrace.pin.Pin`, None :returns: The first found :class:`ddtrace.pin.Pin` or `None` is none was found N)r get_from)objsobjpins r_findz Pin._findGs?  C,,s##C   trcFt|dr|St|tjrt nt }t||d}|A|jt|kr)| }| ||S)akReturn the pin associated with the given object. If a pin is attached to `obj` but the instance is not the owner of the pin, a new pin is cloned and attached. This ensures that a pin inherited from a class is a copy for the new instance, avoiding that a specific instance overrides other pins values. >>> pin = Pin.get_from(conn) :param obj: The object to look for a :class:`ddtrace.pin.Pin` on :type obj: object :rtype: :class:`ddtrace.pin.Pin`, None :returns: :class:`ddtrace.pin.Pin` associated with the object, or None if none was found __getddpin__N) hasattrr/ isinstancewrapt ObjectProxy_DD_PIN_PROXY_NAME _DD_PIN_NAMErridcloneonto)r+pin_namer,s rr)z Pin.get_from[s 3 ' ' &##%% %)3C9J)K)K]%%Q]c8T** ?s{bgg55))++C HHSMMM rc|sdS||}|'t||||dS|||||dS)a[Override an object with the given attributes. That's the recommended way to customize an already instrumented client, without losing existing attributes. >>> conn = sqlite.connect('/tmp/user.db') >>> # Override a pin for a specific connection >>> Pin.override(conn, service='user-db') Nr&)r)r r8r7)clsr+rr rr,s roverridez Pin.overrideus|$  Fll3 ; d6 : : : ? ? D D D D D IIgDI @ @ E Ec J J J J JrcZt|jo|jjp |jjS)z,Return true if this pin's tracer is enabled.)boolrenabled _apm_opt_outrs rr?z Pin.enableds(DK  Vdk&9&UT[=UVrTc t|dr||St|tjrt nt }t||_|j r$tj |j t|||S#t$r tddYdSwxYw)zPatch this pin onto the given object. If send is true, it will also queue the metadata to be sent to the server. __setddpin__zcan't pin onto object. skippingTexc_infoN)r0rBr1r2r3r4r5r6rrrconfig_add_extra_servicesetattrrlogdebug)rr+sendr9s rr8zPin.ontos HsN++ .''----7U=N-O-Oa))UaHc77DL| @11$,???3$// / H H H II7$I G G G G G G Hs$BA7B&C C c t|tjrtnt}t |}|t||dSdS#t$r t ddYdSwxYw)Nz&can't remove pin from object. skippingTrC) r1r2r3r4r5r r)delattrrrHrI)rr+r9r,s r remove_fromzPin.remove_froms O-7U=N-O-Oa))UaH,,s##CX&&&&& O O O II>I N N N N N N OsAA&BBc|s |jr|j}|j}t|p|j||p|j|S)z=Return a clone of the pin with the given attributes replaced.)rr rr)r copyrr rr)rrr rrEs rr7z Pin.clonesl $  $9>>##D""$$+t|(T[     r)NNNN)NNN)T)__name__ __module__ __qualname____doc__ __slots__rpropertyrr r' staticmethodr-r) classmethodr<r?r8rMr7 __classcell__)r#s@rr r s_  IHHI  !!!!&,,X,22222 ^^^\&\2  KKK[K4WWW HHHH( O O O          rr )typingrrrrr2rinternal.loggerrrPrHr5r4objectr rrr]s  ''''''j   ,x x x x x &x x x x x r