mirror of https://github.com/explosion/spaCy.git
59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
from __future__ import print_function
|
|
|
|
import sys
|
|
import os
|
|
import shutil
|
|
|
|
import plac
|
|
import sputnik
|
|
from sputnik.package_list import (PackageNotFoundException,
|
|
CompatiblePackageNotFoundException)
|
|
|
|
from .. import about
|
|
|
|
|
|
def migrate(path):
|
|
data_path = os.path.join(path, 'data')
|
|
if os.path.isdir(data_path):
|
|
if os.path.islink(data_path):
|
|
os.unlink(data_path)
|
|
else:
|
|
shutil.rmtree(data_path)
|
|
for filename in os.listdir(path):
|
|
if filename.endswith('.tgz'):
|
|
os.unlink(os.path.join(path, filename))
|
|
|
|
|
|
@plac.annotations(
|
|
force=("Force overwrite", "flag", "f", bool),
|
|
)
|
|
def main(data_size='all', force=False):
|
|
if force:
|
|
sputnik.purge(about.__title__, about.__version__)
|
|
|
|
try:
|
|
sputnik.package(about.__title__, about.__version__, about.__default_model__)
|
|
print("Model already installed. Please run 'python -m "
|
|
"spacy.en.download --force' to reinstall.", file=sys.stderr)
|
|
sys.exit(1)
|
|
except (PackageNotFoundException, CompatiblePackageNotFoundException):
|
|
pass
|
|
|
|
package = sputnik.install(about.__title__, about.__version__, about.__default_model__)
|
|
|
|
try:
|
|
sputnik.package(about.__title__, about.__version__, about.__default_model__)
|
|
except (PackageNotFoundException, CompatiblePackageNotFoundException):
|
|
print("Model failed to install. Please run 'python -m "
|
|
"spacy.en.download --force'.", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
# FIXME clean up old-style packages
|
|
migrate(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
print("Model successfully installed.", file=sys.stderr)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
plac.call(main)
|