o i4 @sLddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZddlmZddlmZddlmZmZdd lmZdd lmZdd lmZddlZd d hZiddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/id0d1d2d3d4d5d6d7d8d9d:d9d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKd dLdMdNdOdPd dQdRdSdTdUdVZ eGdWdXdXZ!dd]d^Z"eddddeZ#ddgdhZ$ddmdnZ%ddpdqZ&ddudvZ'ddydzZ(dd}d~Z)dddddZ*dddZ+dddZ,dddZ-dS)) annotationsN) Generator)contextmanager) dataclass)cast)ROUNDED)ConsoleGroup)Panel)Rule)Texturllibsqlite3z&opentelemetry-instrumentation-aio_pikaaio_pikaz,opentelemetry-instrumentation-aiohttp-clientaiohttp_clientz,opentelemetry-instrumentation-aiohttp-serveraiohttp_serverz#opentelemetry-instrumentation-aiopgaiopgz%opentelemetry-instrumentation-asyncpgasyncpgz"opentelemetry-instrumentation-botobotoz&opentelemetry-instrumentation-botocorebotocorez$opentelemetry-instrumentation-celeryceleryz-opentelemetry-instrumentation-confluent-kafkaconfluent_kafkaz$opentelemetry-instrumentation-djangodjangoz+opentelemetry-instrumentation-elasticsearch elasticsearchz$opentelemetry-instrumentation-falconfalconz%opentelemetry-instrumentation-fastapifastapiz#opentelemetry-instrumentation-flaskflaskz"opentelemetry-instrumentation-grpcgrpcz#opentelemetry-instrumentation-httpxhttpxz$opentelemetry-instrumentation-jinja2jinja2z*opentelemetry-instrumentation-kafka-python kafka_pythonz#opentelemetry-instrumentation-mysqlmysqlz)opentelemetry-instrumentation-mysqlclient mysqlclientz"opentelemetry-instrumentation-pikapikaz%opentelemetry-instrumentation-psycopgpsycopgz&opentelemetry-instrumentation-psycopg2z(opentelemetry-instrumentation-pymemcache pymemcachez%opentelemetry-instrumentation-pymongopymongoz%opentelemetry-instrumentation-pymysqlpymysqlz%opentelemetry-instrumentation-pyramidpyramidz#opentelemetry-instrumentation-redisredisz'opentelemetry-instrumentation-remoulade remouladez&opentelemetry-instrumentation-requestsrequestsz(opentelemetry-instrumentation-sqlalchemy sqlalchemyz%opentelemetry-instrumentation-sqlite3'opentelemetry-instrumentation-starlette starlettez%opentelemetry-instrumentation-tornadotornado tortoise_ormurllib3 pydantic_ai anthropicopenai openai_agents)z)opentelemetry-instrumentation-tortoiseormz$opentelemetry-instrumentation-urllib%opentelemetry-instrumentation-urllib3zpydantic-ai-slimr3r4z openai-agentsc@s.eZdZUded<ded<ded<ded<dS) InstrumentationContextdict[str, str]instrument_pkg_mapset[str]installed_pkgsinstalled_otel_pkgsset[tuple[str, str]]recommendationsN)__name__ __module__ __qualname____annotations__rCrC_/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/logfire/_internal/cli/run.pyr7Is  r7argsargparse.NamespacereturnNonec Csttt|j}ttt|j}t|}t |j |j }|r5t t jd}t|j ||j}t|||jd|j}t jdt|j}rw|} td|g| d|dd| tj|ddd WddS1spwYdS|r|d} tjtj| } | t jvrt jd| t|d d|tj| dd WddS1swYdSt d t !d dS)N)file)consoleinstrumented_packages_textr>rz-m z python -m  __main__T)run_name alter_syszpython )rNzHUsage: logfire run [-m MODULE] [args...] OR logfire run SCRIPT [args...])"logfire configurerboolsummarysetstrexcludecollect_instrumentation_contextinstrument_packagesr<r9rsysstderrrKr;print_otel_summaryr>script_and_argspathinsertosgetcwdmodulealter_sys_argvjoinrunpy run_moduledirnameabspathrun_pathprintexit) rErTrWctxinstrumented_packagesrJinstrumentation_textr] module_name module_args script_path script_dirrCrCrD parse_runQs<    *" "rsargv list[str]cmdrVGenerator[None, None, None]ccs<tj}|t_ztjd|ddVW|t_dS|t_w)NzRunning command: {cmd_str})cmd_str)rZrtcopyrQinfo)rtrv orig_argvrCrCrDrcs rcrScCstdduS)zCCheck if uv package manager is installed and available in the PATH.uvN)shutilwhichrCrCrCrDis_uv_installedsrinstalled_otel_packagesr:r9r8c Cslg}tjddtjdd|D]!}|dd}||}zt|Wn ty-Yqw||q|S)zCall every `logfire.instrument_x()` we can based on what's installed. Returns a list of packages that were successfully instrumented. LANGSMITH_OTEL_ENABLEDtrueLANGSMITH_TRACING_ENABLEDopentelemetry-instrumentation-)r`environ setdefaultreplaceinstrument_package Exceptionappend)rr9 instrumented otel_pkg_namebase_pkg import_namerCrCrDrYs    rYrcCsd|}tt|dS)N instrument_)getattrrQ)rinstrument_attrrCrCrDrs rr<r;r=cCsht}|D]\}}||vrq||vs|tvr|||fqd|vr)|dd|vr2|d|S)aDetermine which OpenTelemetry instrumentation packages are recommended for installation. Args: instrument_pkg_map: Mapping of instrumentation package names to the packages they instrument. installed_otel_pkgs: Set of already installed instrumentation package names. installed_pkgs: Set of all installed package names. Returns: Set of tuples where each tuple is (instrumentation_package, target_package) that is recommended for installation. r)r-r.r+)r6r1)rUitemsSTANDARD_LIBRARY_PACKAGESadddiscard)r9r<r;r>otel_pkg required_pkgrCrCrD,find_recommended_instrumentations_to_installs  rrmr cCsd|vr |dd|vr|dtd}t|D]#}|dd}||vr2|jd|d d d q|jd |d dd q|S)Nrr-r+r6z!Your instrumentation checklist: rru✓ z (installed and instrumented) greenstyleu⚠️ z0 (installed but not automatically instrumented) yellow)rr sortedrr)r<rmr;textpkg_namerrCrCrDrKs    rKr>tuple[Text, Text]cCszt|}t}|D]\}}|jd|d|dddq |dt}|r9|d|jt|dd|d||fS) zEReturn (recommended_packages_text, install_all_text) as Text objects.u☐ z (need to install z) grey50r z3To install all recommended packages at once, run: bold)rr r_full_install_command)r>sorted_recommendationsrecommended_textrinstrumented_pkg install_textrCrCrDget_recommendation_textss    r)rKrJr Text | Nonec Cstddd}|jddd|jdddtdd}t|\}}|r&||}n|}t||||}t|ddtd d } |d || |dS) Nz To hide this summary box, use: italicrzlogfire run --no-summaryz italic bold.bluez&[bold blue]Logfire Summary[/bold blue])rP)title border_styleboxpaddingr)r rr rr r rrj) rJrKr> hide_note footer_rulerecommended_packages_textinstall_all_text packages_textcontentpanelrCrCrDr\s0       r\c Csz ddtjDWSttfy@ttjdtdddl }dd|j DWdYS1s8wYYdSw)z$Get a set of all installed packages.cSsh|] }|jdqS)Name)metadatalower).0distrCrCrD ,sz%installed_packages..ignore)categoryrNcSsh|]}|jqSrC)keyrpkgrCrCrDr3s) importlibr distributions ImportErrorAttributeErrorwarningscatch_warningsfilterwarnings UserWarning pkg_resources working_set)rrCrCrDinstalled_packages(s *rlist[tuple[str, str]]cCs<|sdSdd|D}trdd|Sdd|S)z?Generate a command to install all recommended packages at once.rcSsg|]\}}|qSrCrC)rr_rCrCrD ;sz)_full_install_command..zuv add rLz pip install )rrd)r> package_namesrCrCrDr6s rrWcsNfddtD}tfdd|D}t||}t|||dS)zFCollects all relevant context for instrumentation and recommendations.csi|] \}}|vr||qSrCrC)rrr)rWrCrD Gsz3collect_instrumentation_context..csh|]}|vr|qSrCrCr)r;rCrDrIsz2collect_instrumentation_context..)r9r;r<r>)OTEL_INSTRUMENTATION_MAPrrkeysrr7)rWr9r<r>rC)rWr;rDrXEsrX)rErFrGrH)rtrurvrVrGrw)rGrS)rr:r9r8rGru)rrV)r9r8r<r:r;r:rGr=)r<r:rmrur;r:rGr )r>r=rGr)rJrrKrr>r=rGrH)rGr:)r>rrGrV)rWr:rGr7). __future__r _annotationsargparserimportlib.metadatar`rer}rZrcollections.abcr contextlibr dataclassesrtypingrrich.boxr rich.consolerr rich.panelr rich.ruler rich.textr rQrrr7rsrcrrYrrrKrr\rrrXrCrCrCrDs               !"#. 3     %  -