i+dZddlmZddlZddlZddlmZmZddl m Z ddl m Z ddl mZddlmZeeZeGd d Zdd Zdd Zdd ZddZddZddZddZy)a6File discovery and module import utilities for filesystem-based routing. This module provides functions to: 1. Discover Python files in a directory tree 2. Import modules (as packages if __init__.py exists, else directly) 3. Extract decorated components (Tool, Resource, Prompt objects) from imported modules ) annotationsN) dataclassfield)Path) ModuleType)FastMCPComponent) get_loggercJeZdZUdZeeZded<eeZ ded<y)DiscoveryResultzResult of filesystem discovery.)default_factoryz#list[tuple[Path, FastMCPComponent]] componentszdict[Path, str] failed_filesN) __name__ __module__ __qualname____doc__rlistr __annotations__dictr}/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/server/providers/filesystem_discovery.pyr r s&)76M E 6" 99 %  DJJ &  T# %=rc(|dz jS)z;Check if a directory is a Python package (has __init__.py).r)r) directorys r_is_package_dirr+Bs  % - - //rc|j}d}||jk7r-t|r|}|j}n |S||jk7r-|S)zFind the root of the package containing this file. Walks up the directory tree until we find a directory without __init__.py. Returns: The package root directory, or None if not in a package. N)parentr+) file_pathcurrent package_roots r_find_package_rootr1GsUGL W^^ # 7 #"LnnG   W^^ # rc|j|j}t|j}|dj d|d<dj |S)zCompute the dotted module name for a file within a package. Args: file_path: Path to the Python file. package_root: Root directory of the package. Returns: Dotted module name (e.g., "mcp.tools.greet"). r.) relative_tor-rr" removesuffixjoin)r.r0relativer"s r_compute_module_namer9\sN$$\%8%89H  Eb &&u-E"I 88E?rc |j}t|}|t||}t|j}|t j vr t j jd| |t jvr&tjt j|Stj|S|j}t|j}|t j vr t j jd|tjj||}| |j td|tjj#|}|t j|< |j j%||S#t$r}td|d|d||d}~wwxYw#t&$r7}t jj)|dtd|d||d}~wwxYw)aImport a Python file as a module. If the file is part of a package (directory has __init__.py), imports it as a proper package member (relative imports work). Otherwise, imports directly using spec_from_file_location. Args: file_path: Path to the Python file. Returns: The imported module. Raises: ImportError: If the module cannot be imported. NrzFailed to import z from z: zCannot load spec for zFailed to execute module )resolver1r9strr-sysr'insertmodules importlibreload import_module ImportErrorstemutilspec_from_file_locationloadermodule_from_spec exec_module Exceptionpop)r.r0 module_namepackage_parente parent_dirspecmodules rimport_module_from_filerRms !!#I&i0L*9lC \001  ) HHOOA~ . ckk) '' K(@AA**;7 7 nn ))* SXX % HHOOAz *~~55k9M <4;;. 5i[AB B006#) K  S KK # #F +  7 #K=ykA3G  , S KKOOK . 9)BqcJKQR R Ss627F*FG G'F<<G H 2G??Hc0ddl}ddlm}ddlm}ddlm}ddlm}ddl m }ddl m }dd l m}dd lm} dd lm} | |||f} g} t'|D]4} | j)d r t+|| }t/|| r| j1|A||}|Lt/|| r|j2 |j2nd }| j5||j6|j8|j:|j<|j>|j@|jB|jD||jF|jH|jJ }| j1|t/||r~|j2 |j2nd }d|jLvxrd|jLv}||}tO|jQ|jR}|s|r|j5||jL|j6|j8|j:|j<|jT|j>|jB|jD||jJ }n|j5||jL|j6|j8|j:|j<|jT|j>|jB|jD||jJ }| j1|t/||s|j2 |j2nd }|j5||j6|j8|j:|j<|j>|jD||jJ }| j1|7| S#t,$rYFwxYw)aExtract all MCP components from a module. Scans all module attributes for instances of Tool, Resource, ResourceTemplate, or Prompt objects created by standalone decorators, or functions decorated with @tool/@resource/@prompt that have __fastmcp__ metadata. Args: module: The imported module to scan. Returns: List of component objects (Tool, Resource, ResourceTemplate, Prompt). rN)get_fastmcp_meta) PromptMeta)Prompt) ResourceMeta)Resource)ResourceTemplate)without_injected_parameters)ToolMeta)Tool_F) r title descriptioniconstags output_schemarmetatask exclude_args serializerauth{}) fn uri_templater r^r_r` mime_typerarrcrdrg) rjurir r^r_r`rlrarrcrdrg)r r^r_r`rarcrdrg)+inspectfastmcp.decoratorsrTfastmcp.prompts.function_promptrUfastmcp.prompts.promptrV#fastmcp.resources.function_resourcerWfastmcp.resources.resourcerXfastmcp.resources.templaterYfastmcp.server.dependenciesrZfastmcp.tools.function_toolr[fastmcp.tools.toolr\dir startswithgetattrAttributeError isinstancer#rd from_functionr r^r_r`rarbrrcrerfrgrmbool signature parametersrl)rQrnrTrUrVrWrXrYrZr[r\component_typesr r objrc resolved_tasktoolhas_uri_params wrapper_fnhas_func_paramsresourceprompts rextract_componentsrs 3:-@3;G4'X'7@O)+JF  ??3   &$'C c? +   c "  $  $)-1YY-B  ))** $ 0 0**"&"4"4 $ 0 0&!%!2!2#*!!$'D,/-1YY-B  !$!DSDHH_8= "&w'8'8'D'O'O"P!_/==%)XX!YY"jj$($4$4"jj"&..!YY$($4$4!YY*!YY > H (55 HH!YY"jj$($4$4"jj"&..!YY$($4$4!YY*!YY 6 H!!(+D*--1YY-B  --** $ 0 0**&. !!&)mp c   s% N NNc^t}t|D]=} t|}t|}|D]}|jj||f!?|S#t$r"}t ||j |<Yd}~hd}~wt $r"}t ||j |<Yd}~d}~wwxYw)aDiscover files, import modules, and extract components. This is the main entry point for filesystem-based discovery. Args: root: Root directory to scan. Returns: DiscoveryResult with components and any failed files. Note: Files that fail to import are tracked in failed_files, not logged. The caller is responsible for logging/handling failures. Files with no components are silently skipped. N) r r(rRrCr<rrJrr r#)r%resultr.rQrNr components rdiscover_and_importr's  F#D)  ,Y7F(/ #I    $ $i%; <$* M -0VF   *  -0VF   *  s# A B,!A>> B, B''B,)r%rreturnz list[Path])r*rrr~)r.rrz Path | None)r.rr0rrr<)r.rrr)rQrrzlist[FastMCPComponent])r%rrr )r __future__rimportlib.utilr@r= dataclassesrrpathlibrtypesrfastmcp.utilities.componentsrfastmcp.utilities.loggingr rloggerr r(r+r1r9rRrrrrrrsq# (90 H  @@ @D0 *"?Dup r