From b69eaf300521731198e315cae1c225fc1038e4d7 Mon Sep 17 00:00:00 2001 From: "albert.montes" Date: Thu, 14 Nov 2019 16:43:13 +0100 Subject: [PATCH] Add optional argument so nltk files can be downloaded in a specific directory --- textblob/download_corpora.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/textblob/download_corpora.py b/textblob/download_corpora.py index 47231a80..651502fd 100644 --- a/textblob/download_corpora.py +++ b/textblob/download_corpora.py @@ -12,6 +12,7 @@ $ python -m textblob.download_corpora lite """ +import os import sys import nltk @@ -29,21 +30,23 @@ ALL_CORPORA = MIN_CORPORA + ADDITIONAL_CORPORA -def download_lite(): + +def download_lite(download_dir=None): for each in MIN_CORPORA: - nltk.download(each) + nltk.download(each, download_dir=download_dir) -def download_all(): +def download_all(download_dir=None): for each in ALL_CORPORA: - nltk.download(each) + nltk.download(each, download_dir=download_dir) def main(): + download_dir = os.getenv('NLTK_DATA') if 'lite' in sys.argv: - download_lite() + download_lite(download_dir) else: - download_all() + download_all(download_dir) print("Finished.")