uYfcddlmZddlmZddlmZddlmZddlZddlmZddlm Z ee Z d Z d e zZ Gd d eZy) ) TYPE_CHECKING)Any)Dict)OptionalN) get_logger)wrapt _datadog_pin_self_ceZdZdZgdZ ddZedZfdZdZ e dZ e dZ e dd Zd Zdd Zd Z dd ZxZS)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 _initializedc|xstj}||_||_d|_|xsi|_||jd<d|_y)N service_nameT)ddtracerrrrr)selfservicerrrs K/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/ddtrace/pin.py__init__z Pin.__init__#sJ)7>>   }" '. ^$ c |jdS)zoBackward compatibility: accessing to `pin.service` returns the underlying configuration value. r)rrs rrz Pin.service6s ||N++rcht|ddr|dk7r tdtt|||y)NrFrz5can't mutate a pin, use override() or clone() instead)getattrAttributeErrorsuperr __setattr__)rnamevalue __class__s rr!zPin.__setattr__>s3 4 /DI4E !XY Y c4$T51rcVd|jd|jd|jdS)Nz Pin(service=z, tags=z , tracer=)rrrrs r__repr__z Pin.__repr__Cs8< diiQUQ\Q\]]rcH|D]}tj|}|s|cSy)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._findFs. C,,s#C  rct|dr|jSt|tjrt nt }t||d}|9|jt|k7r!|j}|j||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) hasattrr0 isinstancer ObjectProxy_DD_PIN_PROXY_NAME _DD_PIN_NAMErridcloneonto)r,pin_namer-s rr*z Pin.get_fromZsr 3 '##% %)3C9J9J)K%Q]c8T* ?s{{bg5))+C HHSM rc|sy|j|}|t|||j|y|j|||j|y)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,rrrr-s roverridez Pin.overridetsQ$ ll3 ; d6 : ? ? D IIgDI @ E Ec Jrc\t|jxr|jjS)z,Return true if this pin's tracer is enabled.)boolrenabledrs rr?z Pin.enableds"DKK 8T[[%8%88rcz t|dr|j|St|tjrt nt }t||_|jr)tjj|jt|||S#t$rtj!ddYywxYw)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)r1rAr2r r3r4r5r6rrrconfig_add_extra_servicesetattrrlogdebug)rr,sendr9s rr8zPin.ontos HsN+''---7U=N=N-O)UaHc7DL||11$,,?3$/ / H II7$I G HsBA7B B:9B:c t|tjrtnt}t j |}| t||yy#t$rtjddYywxYw)Nz&can't remove pin from object. skippingTrB) r2r r3r4r5r r*delattrrrGrH)rr,r9r-s r remove_fromzPin.remove_fromsd O-7U=N=N-O)UaH,,s#CX& O II>I N OsA A A0/A0c|s&|jr|jj}|jj}t|xs |j||xs |j |S)z=Return a clone of the pin with the given attributes replaced.)rrrr)rcopyrr rr)rrrrrDs rr7z Pin.clones\ 99>>#D""$+t||(T[[   r)NNNN)NNN)T)__name__ __module__ __qualname____doc__ __slots__rpropertyrr!r( staticmethodr.r* classmethodr<r?r8rLr7 __classcell__)r$s@rr r s II  !&,,2 ^&2  KK49 H( O   rr )typingrrrrrinternal.loggerrvendorr rOrGr5r4objectr rrr]sF '   ,x &x r