o i@sddlZddlZddlmZddlmZmZmZmZm Z ddl Z ddl m Z m Z ddlmZmZddlmZdd lmZmZmZmZdd lmZmZed edd d eddddd edddddd eddddd eddddd fdeededed ed!ef d"d#Z d5dd$ddd%deededed eeed!ed&e eeff d'd Z d&eeeffd(d)Z!d$d*deded&eeeffd+d,Z"ded&eefd-d.Z#ded&eeeffd/d0Z$  d6d1eeefd2eed eeed&efd3d4Z%dS)7N)Path)AnyDictListOptionalUnion)MarkdownRendererPrinter)aboututil)importlib_metadata)ArgOptappstring_to_list)get_latest_versionget_model_filenameinfoz Optional loadable spaCy pipeline)helpFz --markdownz-mdz#Generate Markdown for GitHub issuesz--silentz-sz-Sz"Don't print anything (just return)labelsz --excludez-ez2Comma-separated keys to exclude from the print-outz--urlz-uzLPrint the URL to download the most recent compatible version of the pipelinemodelmarkdownsilentexcludeurlcCst|}t|||||ddS)aa Print info about spaCy installation. If a pipeline is specified as an argument, print its meta information. Flag --markdown prints details in Markdown for easy copy-pasting to GitHub issues. Flag --url prints only the download URL of the most recent compatible version of the pipeline. DOCS: https://spacy.io/api/cli#info rrrrN)rr)rrrrrrB/home/ubuntu/.local/lib/python3.10/site-packages/spacy/cli/info.pyinfo_clis r Trreturnc st|| d}s g|r+|dur#d|d}t|}t|d|S|jdddn|r:d|d}t||d }nd }t}d d |D}d |vrct|d trcd dd|d D|d <t ||d} |rt|srt| | S|sfdd |D} |j | |d|S)Nno_printprettyzDownload info for pipeline '' download_urlz%--url option requires a pipeline namerexitszInfo about pipeline 'rzInfo about spaCycSs"i|] \}}|dd|qS) _)lowerreplace.0kvrrr Fs"zinfo.. Pipelinesz, css$|] \}}|d|dVqdS)z ()Nr)r/nr1rrr Hs zinfo..)titlercsi|] \}}|vr||qSrrr.rrrr2Qs)r7) r info_model_urlprintfail info_model info_spacyitems isinstancedictjoin get_markdowntable) rrrrrmsgr7dataraw_data markdown_data table_datarr8rr-s:      cCsRi}tD]}|dd}t|||<qtjtttj j t t |dS)z\Generate info about the current spaCy intallation. RETURNS (dict): The spaCy info. -r+)z spaCy versionLocationPlatformzPython versionr3) r get_installed_modelsr-get_package_versionr __version__strr__file__parentplatformpython_version) all_modelspkg_namepackagerrrr=Vs  r=r)cCst|| d}t|rt|}nt|}|d}|s&|jd|ddt|}| |kr:t | |d<nt ||d<t |}|rJ||d<dd | DS) zGenerate info about a specific model. model (str): Model name of path. silent (bool): Don't print anything, just return. RETURNS (dict): The model meta. r"z meta.jsonzCan't find pipeline meta.jsonrr'sourcer&cSsi|] \}}|dvr||qS))accuracy performancespeedrr.rrrr2szinfo_model..) r r is_packageget_package_pathris_filer;srsly read_jsonresolverOinfo_installed_model_urlr>)rrrD model_path meta_pathmetar&rrrr<hs"     r<cCsNzt|}|d}t|trt|}|dWSWdSty&YdSw)zGiven a pipeline name, get the download URL if available, otherwise return None. This is only available for pipelines installed as modules that have dist-info available. zdirect_url.jsonrN)r distribution read_textr?rOjsonloads Exception)rdisttextrErrrras      racCs<t|}t||}tjd|}d}|j||d}||dS)z=Return the download URL for the latest version of a pipeline./z>https://github.com/explosion/spacy-models/releases/tag/{m}-{v})mr1)r& release_url)rrr __download_url__format)rversionfilenamer& release_tplrnrrrr9s   r9rEr7c Cst}|r||d|g}|D]7\}}|r||vrqt|tr;zt|}Wn ty7d}Ynw|r;q| | |dd|q|| |d|j dS)a Get data in GitHub-flavoured Markdown format for issues etc. data (Dict[str, Any]): Label/value pairs. title (str): Optional title, will be rendered as headline 2. exclude (List[str]): Names of keys to exclude. RETURNS (str): The Markdown string. r F:r* ) raddr7r>r?rOrexistsriappendboldlistrk)rEr7rmdr>keyvalue existing_pathrrrrBs$    "rB)N)NN)&rgrRpathlibrtypingrrrrrr^wasabirr r r compatr _utilrrrrdownloadrrcommandrOboolr r@rr=r<rar9rBrrrrsz       )$