o i@slddlmZddlmZddlZddlmZddlmZddl m Z e e Z dZ d e ZGd d d eZdS) )Any)OptionalN) is_wrapted)config) get_logger _datadog_pin_self_c s`eZdZdZgdZ   d$deedeeeefdeeeefddfdd Z e defd d Z d ed eddffdd Z defddZ edededfddZedededfddZe  d%dedeedeeeefddfddZdefddZd&dededdfddZdeddfd d!Z  d%deedeeeefddfd"d#ZZS)'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') tags_target_config _initializedNservicer rreturncCs.|pi|_d|_|p i|_||jd<d|_dS)N service_nameTr )selfrr rrF/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/_trace/pin.py__init__"s    z Pin.__init__cCs |jdS)zoBackward compatibility: accessing to `pin.service` returns the underlying configuration value. r)rrrrrr1s z Pin.servicenamevaluecs2t|ddr|dkrtdtt|||dS)NrFr z5can't mutate a pin, use override() or clone() instead)getattrAttributeErrorsuperr __setattr__)rrr __class__rrr8szPin.__setattr__cCsd|j|jfS)NzPin(service=%s, tags=%s)rr rrrr__repr__=sz Pin.__repr__objscGs$|D] }t|}|r|SqdS)a Return the first :class:`ddtrace.trace.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.trace.Pin` on :type objs: List of objects :rtype: :class:`ddtrace.trace.Pin`, None :returns: The first found :class:`ddtrace.trace.Pin` or `None` is none was found N)r get_from)r"objpinrrr_find@s  z Pin._findr$cCsZt|dr |St|rtnt}t||d}|dur+|jt|kr+|}| ||S)aqReturn 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.trace.Pin` on :type obj: object :rtype: :class:`ddtrace.trace.Pin`, None :returns: :class:`ddtrace.trace.Pin` associated with the object, or None if none was found __getddpin__N) hasattrr'r_DD_PIN_PROXY_NAME _DD_PIN_NAMErr idcloneonto)r$pin_namer%rrrr#Ss   z Pin.get_fromcCsD|sdS||}|durt||d}n|j||d}||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 r,r-)clsr$rr r%rrroverridels z Pin.overridecCstjjptjS)N)ddtracetracerenabled asm_config _apm_opt_outrrrrr3sz Pin.enabledTsendcCsvz)t|dr ||WSt|rtnt}t||_|jr#tj |jt |||WSt y:t jdddYdSw)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)r(r7rr)r*r+r rr1r_add_extra_servicesetattrrlogdebug)rr$r6r.rrrr-s    zPin.ontocCs\zt|rtnt}t|}|durt||WdSWdSty-tjdddYdSw)Nz&can't remove pin from object. skippingTr8) rr)r*r r#delattrrr<r=)rr$r.r%rrr remove_froms  zPin.remove_fromcCs6|s |jr |j}|j}t|p|j||d}|S)z=Return a clone of the pin with the given attributes replaced.)rr r)r copyrr r)rrr rr%rrrr,s   z Pin.clone)NNN)NN)T)__name__ __module__ __qualname____doc__ __slots__rstrdictrrpropertyrrr! staticmethodr&r# classmethodr0boolr3r-r?r, __classcell__rrrrr s^    r )typingrrr1ddtrace.internal.compatrddtrace.internal.settings.asmrr4internal.loggerrrAr<r*r)objectr rrrrs