o i@snUddlZddlZddlZddlZddlmZddlZddlZddl Z ddl Z ddl Z ddl mZddlmZddlmZGdddZGdd d Zd Zd Zdae jeed <d e jedefddZdefddZdee jee jeffddZdedefddZdededdfddZ defddZ!dedefddZ"dededefd d!Z#de jefd"d#Z$dS)$N)Path) gitmetadata) Distribution) _package_for_root_module_mappingc @sFeZdZdedededeeddf ddZdeeejffd d Z dS) LibrarykindnameversionpathsreturnNcCs||_||_||_||_dSNrrr r )selfrrr r rf/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/datadog/profiling/code_provenance.py__init__s zLibrary.__init__cCs|j|j|jt|jdS)Nr )rrr listr rrrrto_dictszLibrary.to_dict) __name__ __module__ __qualname__strsetrdicttAnyrrrrrrs  rc@s.eZdZdddZdeeejffddZdS)CodeProvenancer Nc Csg|_tddtttddddgd}tjdkrDtj D]%}zt j |}|r9|j dkr9|jd |jd WqtyCYqw|j|tpNi}i}ttd }|D]G\}}|j}|d rw|td dd dd}||} | durtd||jtd} | ||<||} |ds| r| jt| q\t\} } } || d} rd| _ |j!|"dS)Nzstandard librarystdlibzz&zr ) frozenzpurelibz __pycache__/.rz.pylibrary)# librariesrplatformpython_versionr sysconfigget_pathsys version_infostdlib_module_names importlibutil find_specoriginr addr Exceptionappendrritems startswithlensplitgetr endswithis_dirrr get_git_tagsrextendvalues)r python_stdlibrspecmodule_to_distributionr' site_packagesmoduledistlib module_path_ main_packageinforrrr#sV        zCodeProvenance.__init__cCsddd|jDiS)Nv1cSsg|]}|qSr)r).0rFrrr jsz*CodeProvenance.to_dict..)r'rrrrriszCodeProvenance.to_dict)r N) rrrrrrrrrrrrrr"s Frzddtrace-code-provenancerK_code_provenance_file_pathpathr cCs2|sdSz tt|jWStyYdSw)Nr&)rrstat st_mtime_nsOSError)rOrrr_safe_mtime_nsrs rScCsVtd}tdd}dtttj |t |f}t | d}td|S)Nr#DD_MAIN_PACKAGEr&utf-8-)r*r+osgetenvjoin_CODE_PROVENANCE_CACHE_VERSIONr(r)r,prefixrShashlibsha256encode hexdigest_CODE_PROVENANCE_CACHE_PREFIX)r#rIdatadigestrrr_cache_basename{s   rdc CsJzt}tt}||d||dfWSttfy$YdSw)Nz.jsonz.lock)NN)rdrtempfile gettempdirFileNotFoundErrorrR)basetmpdirrrr_cache_file_and_locks rj cache_filec CsXz|jdd}Wn tyYdSw|sdSzt|WdSttfy+YdSw)NrVencodingFT) read_textrRjsonloads TypeError ValueError)rkrbrrr_is_valid_cache_files  rsrbcCs||jdtd}z@|jddd}|||t|Wdn1s1wYt ||Wz| WdSt yNYdSwz| Wwt y^Yww)Nr$z.tmpwrVrl) with_suffixsuffixrXgetpidopenwriteflushfsyncfilenoreplaceunlinkrR)rkrbtmp_pathfrrr_write_cached_jsons$    rcCst}t|Sr )rrodumpsr)cprrr_compute_json_strsrcCs4t}|sdSzt||WdStyYdSw)NFT)rrrR)rkcomputedrrr_compute_and_writes  r lock_filenamec CszMt|d=}ddl}z |||j|jBWnty(YWdWdSwt|r7 WdWdSt|WdWS1sFwYWdStyWYdSw)Nza+brFT)rxfcntllockfLOCK_EXLOCK_NBrRrsr)rkrrrrrr_ensure_cache_files"   ( rcCsLtrtSt\}}|dus|durdSt|st|t|r$t|atSdSr )rNrjrsrr)rk lock_filerrrget_code_provenance_files r)%r]importlib.utilr/rorXpathlibrr(r,r*retypingrddtrace.internalrddtrace.internal.packagesrrrrrar[rNOptionalr__annotations__rSrdtuplerjboolrsrrrrrrrrrs6     K "