o i@(@sUdZddlZddlZddlZddlmZddlZddlm Z ddl m Z ddl m Z ddlmZe eZejdkssz._instrument_with_monitoring..rN)rrset_local_eventsrr/r_extract_lines_and_imports co_constsr#updaterraddco_nameco_codeEMPTY_MODULE_BYTES) rrrrr0linesr) nested_coder3 nested_linesr!r!r"rxs     rTr0.cCst}i}d}d}d}d}d} tt|} d} g} tt|j} z t| \}}t| \}}|tkr4q"|| vr\| |} | dur\|r\| | |j dkr\t |dkr\|dur\|df|| <|t urf| |q"|}|}tjg| |ddd }| |tkr| dur|j|}|j|}|dkrd |d d| dn|} | |vr| tt|| d|gf|| <n| |ff|| <|tkr| dur|d |j|}| |vr| tt|| d|gf|| <n| |ff|| <q#tyY||fSw) a, Extract line numbers and import information from bytecode. This parses the bytecode to: 1. Collect all executable line numbers (if track_lines=True) 2. Track IMPORT_NAME and IMPORT_FROM opcodes for dependency tracking Args: code: The code object to analyze package: The package name for resolving relative imports track_lines: Whether to collect line numbers (True for LINE mode, False for PY_START mode) Returns: Tuple of (CoverageLines with executable lines, dict mapping lines to imports) rNTr5r6bigF)signed.)rdictdisfindlinestartsiter enumerater>nextr r<r=len EXTENDED_ARGappendint from_bytesclearr r:co_namesjoinsplittuplelistr StopIteration)rrr0r@r) current_arg previous_arg_previous_previous_argcurrent_import_namecurrent_import_package linestartsr$ext code_iteroffsetopcoder3arg import_depthimport_from_namer!r!r"r9sh        (   8r9)T)3__doc__rHosrtypesrtypingt#ddtrace.internal.bytecode_injectionrddtrace.internal.loggerr/ddtrace.internal.test_visibility.coverage_linesrddtrace.internal.utils.formatsr__name__r version_inforNopmapr r r r bytesr?getenvrreventsPY_STARTLINEr/rDictTuplestrrPOptional__annotations__rVr#Literalr&r,rrboolrGr9r!r!r!r"sZ          B& !   '"