o i"@szUddlmZddlZddlZddlmZddlmZddlmZddlm Z ddl Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZeeZiZ e!e"e!e"e"ffe#d<eej$%ddZ&zddl'm(Z(ddl'm)Z*e*dkrdndZ+Wn e,ydZ(dZ+YnwddZ-ddZ.d e"fd!d"Z/d#d$Z0d>d&e1d'e1fd(d)Z2d>d*e(d&e1d'e1fd+d,Z3 d?d*e(d-e"d.eefd/d0Z4 d?d*e(d1e j5j6d e"d.eefd2d3Z7d4ee8d'e9e:e8e8e8e8e8ffd5d6Z;d7e(d8ee"d'e!e"e9e:e8e8e8e8e8fffd9d:ZF silent_modereturncCs.trt|dSt|ds|stddSdS)Nr:z'Datadog Coverage has not been initiatedFT)r-hasattrlogwarning)r=r?rrr r9Zs   r9 coverage_datacCs,|jr t|jjdkr|stddSdS)Nrz,No coverage collector or data found for itemFT) _collectorlendatarBrC)rDr?rrr _coverage_has_valid_dataes  rHknown_test_name frameworkc Csttj|trt|tjr|dSt|ddsdS|j j z| |WdSt yE}ztt|WYd}~dSd}~ww)NT)r?)rr COVERAGEPYr- isinstancerr. __enter__rHrErGclearswitch_context RuntimeErrorr rBrC)rDrIrJerrrrr _switch_coverage_contextms    rRspancCstr5t|tjr3t|dtd}tj|dd}|sdS|tt d|it t j ||ddddSt|j}t|sCtdSt t j ||tt||||jjdS)Nr,/T)include_importedfiles)r-rLrr.getattrrreport_seen_lines _set_tag_strrjsondumpsr r rK__exit__strtrace_idrHr build_payloadrErGrN)rDrSr'rJr6rVspan_idrrr _report_coverage_to_spans.      ralinescCs\g}ttt|ddD]\}}t|}|dd}|dd}||d|ddfq |S)z3Extract the relevant report data for a single file.cSs|d|dS)Nrr)xrrr szsegments..rrc)r enumeratesortedlistappend)rb _segments_keyggroupr4endrrr segmentss  rpr7contextcCs,|jr|jjs iSddt|jjDS)NcSs0i|]\}}|t|trt|nt|qSr)rLdictrpkeys).0kvrrr sz_lines..)rErGriitems)r7rqrrr _liness rytest_idcCst|}|tvr it|<g}t||D]-\}}|t|vr(t||t||<|r7|t|||dq|dt||iqt|dkrLttt|t d|iS)a Generate a CI Visibility coverage payload, formatted as follows: "files": [ { "filename": , "segments": [ [Int, Int, Int, Int, Int], # noqa:F401 ] }, ... ] For each segment of code for which there is coverage, there are always five integer values: The first number indicates the start line of the code block (index starting in 1) The second number indicates the start column of the code block (index starting in 1). Use value -1 if the column is unknown. The third number indicates the end line of the code block The fourth number indicates the end column of the code block The fifth number indicates the number of executions of the block If the number is >0 then it indicates the number of executions If the number is -1 then it indicates that the number of executions are unknown :param coverage: Coverage object containing coverage data :param root_dir: the directory relative to which paths to covered files should be resolved :param test_id: a unique identifier for the current test run )filenamerpr{rrV) r]rryrxrrjrFr r rZr[)r7r'rz root_dir_str files_datar{rbrrr r_s$    r_)Fr)> itertoolsrrZr1pathlibrtypingrrrddtrace(ddtrace.internal.ci_visibility.constantsr2ddtrace.internal.ci_visibility.telemetry.constantsr1ddtrace.internal.ci_visibility.telemetry.coverager r r r r r$ddtrace.internal.ci_visibility.utilsrddtrace.internal.coverage.coderddtrace.internal.loggerrddtrace.internal.utils.formatsr__name__rBrrrr]__annotations__environgetr-r7rrcoverage_version EXECUTE_ATTR ImportErrorr!r*r8r>boolr9rHrRtraceSpanraintrituplerpryr_rrrr st                        (!4 $