Add parameter to download() for application to not exit if a Model exists. The default behavior is unchanged.

This commit is contained in:
Mahmoud Lababidi 2016-09-14 10:04:09 -04:00
parent cccadbf2a2
commit 4c9ccc3b8b
1 changed files with 5 additions and 4 deletions

View File

@ -9,15 +9,16 @@ from sputnik.package_list import (PackageNotFoundException,
from . import about
def download(lang, force=False):
def download(lang, force=False, fail_on_exist=True):
if force:
sputnik.purge(about.__title__, about.__version__)
try:
sputnik.package(about.__title__, about.__version__, about.__models__[lang])
print("Model already installed. Please run 'python -m "
"spacy.%s.download --force' to reinstall." % lang, file=sys.stderr)
sys.exit(0)
if fail_on_exist:
print("Model already installed. Please run 'python -m "
"spacy.%s.download --force' to reinstall." % lang, file=sys.stderr)
sys.exit(0)
except (PackageNotFoundException, CompatiblePackageNotFoundException):
pass