#!/usr/bin/env python
"""Downloads the necessary NLTK corpora for TextBlob.

Usage: ::

    $ python -m textblob.download_corpora

If you only intend to use TextBlob's default models, you can use the "lite"
option: ::

    $ python -m textblob.download_corpora lite

"""

import sys

import nltk

MIN_CORPORA = [
    "brown",  # Required for FastNPExtractor
    "punkt_tab",  # Required for WordTokenizer
    "wordnet",  # Required for lemmatization
    "averaged_perceptron_tagger_eng",  # Required for NLTKTagger
]

ADDITIONAL_CORPORA = [
    "conll2000",  # Required for ConllExtractor
    "movie_reviews",  # Required for NaiveBayesAnalyzer
]

ALL_CORPORA = MIN_CORPORA + ADDITIONAL_CORPORA


def download_lite():
    for each in MIN_CORPORA:
        nltk.download(each)


def download_all():
    for each in ALL_CORPORA:
        nltk.download(each)


def main():
    if "lite" in sys.argv:
        download_lite()
    else:
        download_all()
    print("Finished.")


if __name__ == "__main__":
    main()