o ip3@slddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl m Z ddl mZddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZeeZ dCdDddZddZddZ dEddZ!dFd!d"Z"d#d$Z#dGd&d'Z$Gd(d)d)eZ%Gd*d+d+eZ&Gd,d-d-eZ'Gd.d/d/eZ(Gd0d1d1eZ)Gd2d3d3eZ*e%j+e&d4d5e%j+e'd6d5e%j+e(d7d5e%j+e)d8d5e%j+e*d9d5e%Z,ee,e\Z-Z.e.se-pd:Z/e0d;e/e1ej2de,_3e \Z4Z5e,j6j3r)e5s)e4pd?Z/e0d@e/e1ej2d@e/d>e,j6_3e$e,j7e,_7dHdAdBZ8dS)I) annotationsN)Env) COMMIT_SHA) MAIN_PACKAGE)REPOSITORY_URL)compat) gitmetadata) get_logger)DDConfig) ValueSource)report_configuration)telemetry_writer)TELEMETRY_LOG_LEVEL)parse_tags_str heap_configrdefault_heap_sample_sizeintreturncCst|tsJ|j}|durtt|S|jsdSzddlm}| j | j }Wnt y@t jd|ddd|YSwd}ttt|||S)Nr)psutilzBUnable to get total memory available, using default value of %d KBiT)exc_infoi) isinstanceProfilingConfigHeap _sample_sizetcastrenabledddtrace.vendorr swap_memorytotalvirtual_memory Exceptionloggerwarningmaxmathceil)rrheap_sample_sizer total_mem max_samplesr*W/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/settings/profiling.py _derive_default_heap_sample_sizes&    r,cCddlm}|j|jfS)Nrddup)"ddtrace.internal.datadog.profilingr/ failure_msg is_availabler.r*r*r+_check_for_ddup_available9s  r3cCr-)Nrstack)r0r5r1r2r4r*r*r+_check_for_stack_availableCs  r6boolcCs0tjd}|dur dStdd|dDS)zBReturn True if DD_INJECTION_ENABLED contains the 'profiler' token.DD_INJECTION_ENABLEDNFcss |] }|dkVqdS)profilerN)striplower).0tokr*r*r+ Psz2_injection_enabled_has_profiler..,)osenvirongetanysplit)injection_enabledr*r*r+_injection_enabled_has_profilerJs rFrawstrcCstrdS|}|dvS)NT)1trueyesonauto)rFr;)rGraw_lcr*r*r+_parse_profiling_enabledSsrOcCs@t|t\}}}|r||t<|r||t<|r||t<|S)z0 Update profiler tags with git metadata )r clean_tags get_git_tagsrrr)tagsrepository_url commit_sha main_packager*r*r+_update_git_metadata_tags\s rVdict[str, str]cCs2ddttttjd|D}|S)NcSsi|] \}}|t|dqS)zutf-8)r ensure_text)r<kvr*r*r+ ms z _enrich_tags..DD_TAGS) itertoolschainrVrr@rArBitems)rRr*r*r+ _enrich_tagslsr`csXeZdZdZejededdddZd5fd d Z ejedddddZ ejedddddZ ejedddddZ eje jedddddZejedddddZejedddddZejed d!d"d#dZejed$ddd%dZejed&ddd'dZejed(d)d"d*dZejed+ddd,dZejed-eid.d/dZejed0ddd1dZejed2d3d"d4dZZS)6ProfilingConfigz dd.profilingrFBooleanz3Enable Datadog profiling when using ``ddtrace-run``)parserdefault help_typehelpargst.AnykwargsrNonecs<tj|i||jstrd|_tj|jd<dSdSdS)NTDD_PROFILING_ENABLED)super__init__rrFr ENV_VAR _value_source)selfrgri __class__r*r+rms  zProfilingConfig.__init__ agentlessrdrerfenable_code_provenanceTz!Whether to enable code provenanceendpoint_collection_enabledz:Whether to enable the endpoint data collection in profiles output_pprofNStringupload_intervalgN@FloatzCThe interval in seconds to wait before flushing out recorded events capture_pctg?zThe percentage of events that should be captured (e.g. memory allocation). Greater values reduce the program execution speed. Must be greater than 0 lesser or equal to 100 max_frames@IntegerzBThe maximum number of frames to capture in stack execution tracingignore_profilerzD**Deprecated**: whether to ignore the profiler in the generated datamax_time_usage_pctzThe percentage of maximum time the stack profiler can use when computing statistics. Must be greater than 0 and lesser or equal to 100api_timeout_msi'zQThe timeout in milliseconds before dropping events if the HTTP API does not replytimeline_enabledzWhether to add timestamp information to captured samples. Adds a small amount of overhead to the profiler, but enables the use of the Timeline view in the UI.rRMappingz^The tags to apply to uploaded profile. Must be a list in the ``key1:value,key2:value2`` formatenable_assertsz7Whether to enable debug assertions in the profiler codesample_pool_capacityzThe number of Sample objects to keep in the pool for reuse. Increasing this can reduce the overhead from frequently allocating and deallocating Sample objects.)rgrhrirhrrj) __name__ __module__ __qualname__ __prefix__r rZr7rOrrmrscode_provenanceendpoint_collectionrOptionalrHrxfloatrzr|rr}rrrrdictrrRrr __classcell__r*r*rqr+raxs      rac@sPeZdZdZZejedddddZejedddddZ ejed ddd dZ d S) ProfilingConfigStackr5rTrbz$Whether to enable the stack profilerruzadaptive_sampling.enabled)rdreprivateuvloopz4Whether to enable uvloop support for async profilingN) rrr__item__rr rZr7radaptive_samplingrr*r*r*r+rs. rc@s<eZdZdZZejedddddZejedddddZ d S) ProfilingConfigLocklockrTrbz#Whether to enable the lock profilerruname_inspect_dirzWhether to inspect the ``dir()`` of local and global variables to find the name of the lock. With this enabled, the profiler finds the name of locks that are attributes of an object.N) rrrrrr rZr7rrr*r*r*r+r$  rc@<eZdZdZZejedddddZeje ddd d dZ d S) ProfilingConfigMemorymemoryrTrbz%Whether to enable the memory profilerru events_bufferrrtN) rrrrrr rZr7rrrr*r*r*r+r9rrc@sNeZdZdZZejedddddZeje j e ddd d dZ e e eZdS) rheaprTrbz*Whether to enable the heap memory profilerru sample_sizeNrzAAverage number of bytes allocated between memory profiler samples)rrrrrr rZr7rrrrrdr,rr*r*r*r+rMs"rc@r) ProfilingConfigPytorchpytorchrFrbz&Whether to enable the PyTorch profilerru events_limiti@Brzsz