o h@sdZddlZddlZddlZddlZddlmZmZmZm Z m Z e dZ e ddZdZe ddZe d ejed Zegd Zeed dd fddZeedddfddZdS)z, Functions for setting up the environments. N)HOME_DIR request_fileunzipget_root_from_zipfileset_logging_levelstanzaCORENLP_MODEL_URLzdhttps://huggingface.co/stanfordnlp/corenlp-{model}/resolve/{tag}/stanford-corenlp-models-{model}.jarzNhttp://nlp.stanford.edu/software/stanford-corenlp-{version}-models-{model}.jarzThttps://huggingface.co/stanfordnlp/CoreNLP/resolve/{tag}/stanford-corenlp-latest.zip CORENLP_HOMEstanza_corenlp) arabicchinesez english-extraz english-kbpfrenchgerman hungarianitalianspanishINFOTc Cstj|}|r |stdtd|d|d||}|tvr2t |dt td|dkr8|nd|}|j |||d }tj |d |d |d } tj | re|setd | ddSz t|| |WdSttfyxty} ztd| d} ~ ww)a A automatic way to download the CoreNLP models. Args: model: the name of the model, can be one of 'arabic', 'chinese', 'english', 'english-kbp', 'french', 'german', 'hungarian', 'italian', 'spanish' version: the version of the model dir: the directory to download CoreNLP model into; alternatively can be set up with environment variable $CORENLP_HOME url: The link to download CoreNLP models. It will need {model} and either {version} or {tag} to properly format the URL logging_level: logging level to use during installation force: Download model anyway, no matter model file exists or not z1Both model and model version should be specified.z Downloading z models (version z) into directory z- is currently not supported. Must be one of: .mainv)tagmodelversionzstanford-corenlp-z-models-z.jarz Model file z? already exists. Please download this model to a new directory.NzpDownloading CoreNLP model file failed. Please try manual downloading at: https://stanfordnlp.github.io/CoreNLP/.)ospath expanduser ValueErrorloggerinfostriplowerAVAILABLE_MODELSKeyErrorlistformatjoinexistswarnrKeyboardInterrupt SystemExit Exception RuntimeError) rrdirurl logging_levelproxiesforcer download_url model_pather4X/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/resources/installation.pydownload_corenlp_models"sJ    r6rc Cstj|}t|ddtj|r&tt|dkr&td|ddSt d|t dtj |d|d krA|nd |}|j ||d }z t |tj |d|Wnttfyctys}ztd |d}~wwt d t|dt d|ttj |d}tj ||}t|D] }ttj |||qt dttj |dt||tkrtd|ddSdS)a A fully automatic way to install and setting up the CoreNLP library to use the client functionality. Args: dir: the directory to download CoreNLP model into; alternatively can be set up with environment variable $CORENLP_HOME url: The link to download CoreNLP models Needs a {version} or {tag} parameter to specify the version logging_level: logging level to use during installation N)r.verboserz Directory z; already exists. Please install CoreNLP to a new directory.z Installing CoreNLP package into zDownload to destination file: z corenlp.ziprr)rrzlDownloading CoreNLP zip file failed. Please try manual installation: https://stanfordnlp.github.io/CoreNLP/.z Unzipping downloaded zip file...z,Moving files into the designated folder at: zRemoving downloaded zip file...zFor customized installation location, please set the `CORENLP_HOME` environment variable to the location of the installation. In Unix, this is done with `export CORENLP_HOME=z`.)rrrrr&lenlistdirrr'rdebugr%r$rr(r)r*r+rrshutilmoveremovermtreeDEFAULT_CORENLP_DIRwarning) r,r-r.r/rrr3corenlp_dirnamefr4r4r5install_corenlpXsP      rC)__doc__rloggingzipfiler;stanza.resources.commonrrrrr getLoggerrgetenvDEFAULT_CORENLP_MODEL_URLBACKUP_CORENLP_MODEL_URLDEFAULT_CORENLP_URLrr%r?setr!r6rCr4r4r4r5s.   6