2017-03-16 19:01:51 +03:00
|
|
|
# coding: utf8
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
|
2017-05-22 13:28:58 +03:00
|
|
|
import plac
|
2017-03-18 14:59:21 +03:00
|
|
|
from pathlib import Path
|
2017-05-22 13:28:58 +03:00
|
|
|
|
2017-05-08 00:25:29 +03:00
|
|
|
from ..compat import symlink_to, path2str
|
|
|
|
from ..util import prints
|
2017-03-18 17:14:48 +03:00
|
|
|
from .. import util
|
2017-03-16 19:01:51 +03:00
|
|
|
|
|
|
|
|
2017-05-22 13:28:58 +03:00
|
|
|
@plac.annotations(
|
|
|
|
origin=("package name or local path to model", "positional", None, str),
|
|
|
|
link_name=("name of shortuct link to create", "positional", None, str),
|
|
|
|
force=("force overwriting of existing link", "flag", "f", bool)
|
|
|
|
)
|
2017-08-09 12:52:38 +03:00
|
|
|
def link(cmd, origin, link_name, force=False, model_path=None):
|
2017-05-27 21:01:46 +03:00
|
|
|
"""
|
|
|
|
Create a symlink for models within the spacy/data directory. Accepts
|
2017-05-22 13:28:58 +03:00
|
|
|
either the name of a pip package, or the local path to the model data
|
|
|
|
directory. Linking models allows loading them via spacy.load(link_name).
|
|
|
|
"""
|
2017-05-08 00:25:29 +03:00
|
|
|
if util.is_package(origin):
|
2017-06-04 14:45:50 +03:00
|
|
|
model_path = util.get_package_path(origin)
|
2017-03-16 19:01:51 +03:00
|
|
|
else:
|
2017-08-09 12:52:38 +03:00
|
|
|
model_path = Path(origin) if model_path is None else Path(model_path)
|
2017-04-16 21:53:25 +03:00
|
|
|
if not model_path.exists():
|
2017-05-08 00:25:29 +03:00
|
|
|
prints("The data should be located in %s" % path2str(model_path),
|
2017-05-22 13:28:58 +03:00
|
|
|
title="Can't locate model data", exits=1)
|
2017-03-18 20:57:31 +03:00
|
|
|
link_path = util.get_data_path() / link_name
|
|
|
|
if link_path.exists() and not force:
|
2017-05-08 00:25:29 +03:00
|
|
|
prints("To overwrite an existing link, use the --force flag.",
|
2017-05-22 13:28:58 +03:00
|
|
|
title="Link %s already exists" % link_name, exits=1)
|
2017-03-18 20:57:31 +03:00
|
|
|
elif link_path.exists():
|
|
|
|
link_path.unlink()
|
2017-04-15 13:11:16 +03:00
|
|
|
try:
|
|
|
|
symlink_to(link_path, model_path)
|
|
|
|
except:
|
2017-05-08 00:25:29 +03:00
|
|
|
# This is quite dirty, but just making sure other errors are caught.
|
|
|
|
prints("Creating a symlink in spacy/data failed. Make sure you have "
|
|
|
|
"the required permissions and try re-running the command as "
|
|
|
|
"admin, or use a virtualenv. You can still import the model as a "
|
|
|
|
"module and call its load() method, or create the symlink manually.",
|
|
|
|
"%s --> %s" % (path2str(model_path), path2str(link_path)),
|
|
|
|
title="Error: Couldn't link model to '%s'" % link_name)
|
2017-04-16 23:28:16 +03:00
|
|
|
raise
|
2017-05-08 00:25:29 +03:00
|
|
|
prints("%s --> %s" % (path2str(model_path), path2str(link_path)),
|
2017-06-04 14:45:50 +03:00
|
|
|
"You can now load the model via spacy.load('%s')" % link_name,
|
2017-05-08 00:25:29 +03:00
|
|
|
title="Linking successful")
|