a g@s6dZddlZddlmZddlmZGdddZdS)z9 Command line for PyThaiNLP's dataset/corpus management. N)corpus)get_pythainlp_data_pathc@s<eZdZddZddZddZddZd d Zd d Zd S)AppcCsVtjdddtdd}|jdtgddd ||d d }t||j|dS) NdatazManage dataset/corpus.aethainlp data subcommands: catalog show list of available datasets info show information about the dataset get download the dataset rm remove the dataset path show full path to data directory Example: thainlp data get thai2fit_wv Current data path: z To change PyThaiNLP data path, set the operating system's PYTHAINLP_DATA_DIR environment variable. For more information about corpora that PyThaiNLP use, see: https://github.com/PyThaiNLP/pythainlp-corpus/ --)prog descriptionusage subcommand)cataloginfogetrmpathzaction on dataset/corpus)typechoiceshelp)argparseArgumentParserr add_argumentstr parse_argsgetattrr selfargvparserargsrrr dataset_namedataset/corpus's namerrrzDownloaded successfully. Not found.) rrrrrrdownloadr#printrrrr r .s  zApp.getcCsRtjddd}|jdtdd||dd}t|jrFtdntd dS) NzRemove a datasetzthainlp data rm r"r#r$r%rzRemoved successfully.r&) rrrrrrremover#r(rrrr r >s  zApp.rmcCsVtjddd}|jdtdd||dd}t|j}|rJt|ntddS) Nz!Print information about a datasetz thainlp data info r"r#r$r%rr&) rrrrrrget_corpus_db_detailr#r()rrrrr rrr r Ns  zApp.infocCstt}|}t|}td|D]N}td|d||dddt|}|rvtd|dd q.tq.td d S) z,Print dataset/corpus available for download.z&Dataset/corpus available for download:z-  Zlatest_version)endz (Local: version)zO Use subcommand 'get' to download a dataset. Example: thainlp data get crfcut N)rZ get_corpus_dbZ corpus_db_urljsonsortedkeysr(r*)rrZ corpus_dbZ corpus_namesnameZ corpus_inforrr r _s   z App.catalogcCsttdS)zPrint path of local dataset.N)r(r)rrrrr rrszApp.pathN) __name__ __module__ __qualname__r!r r r r rrrrr r s  r)__doc__rZ pythainlprZpythainlp.toolsrrrrrr s