2017-03-18 14:14:48 +00:00
|
|
|
# coding: utf8
|
2017-03-18 17:14:03 +00:00
|
|
|
from __future__ import print_function
|
|
|
|
# NB! This breaks in plac on Python 2!!
|
2017-03-23 10:08:30 +00:00
|
|
|
#from __future__ import unicode_literals
|
2017-03-18 14:14:48 +00:00
|
|
|
|
|
|
|
import plac
|
|
|
|
from spacy.cli import download as cli_download
|
|
|
|
from spacy.cli import link as cli_link
|
|
|
|
from spacy.cli import info as cli_info
|
2017-03-20 21:50:13 +00:00
|
|
|
from spacy.cli import package as cli_package
|
2017-03-18 14:14:48 +00:00
|
|
|
|
|
|
|
|
|
|
|
class CLI(object):
|
|
|
|
"""Command-line interface for spaCy"""
|
|
|
|
|
2017-03-20 21:50:13 +00:00
|
|
|
commands = ('download', 'link', 'info', 'package')
|
2017-03-18 14:14:48 +00:00
|
|
|
|
|
|
|
@plac.annotations(
|
|
|
|
model=("model to download (shortcut or model name)", "positional", None, str),
|
|
|
|
direct=("force direct download. Needs model name with version and won't "
|
|
|
|
"perform compatibility check", "flag", "d", bool)
|
|
|
|
)
|
|
|
|
def download(self, model=None, direct=False):
|
|
|
|
"""
|
|
|
|
Download compatible model from default download path using pip. Model
|
|
|
|
can be shortcut, model name or, if --direct flag is set, full model name
|
|
|
|
with version.
|
|
|
|
"""
|
|
|
|
|
|
|
|
cli_download(model, direct)
|
|
|
|
|
|
|
|
|
|
|
|
@plac.annotations(
|
|
|
|
origin=("package name or local path to model", "positional", None, str),
|
2017-03-21 01:05:14 +00:00
|
|
|
link_name=("name of shortuct link to create", "positional", None, str),
|
|
|
|
force=("force overwriting of existing link", "flag", "f", bool)
|
2017-03-18 14:14:48 +00:00
|
|
|
)
|
|
|
|
def link(self, origin, link_name, force=False):
|
|
|
|
"""
|
|
|
|
Create a symlink for models within the spacy/data directory. Accepts
|
|
|
|
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).
|
|
|
|
"""
|
|
|
|
|
|
|
|
cli_link(origin, link_name, force)
|
|
|
|
|
|
|
|
|
|
|
|
@plac.annotations(
|
|
|
|
model=("optional: shortcut link of model", "positional", None, str),
|
|
|
|
markdown=("generate Markdown for GitHub issues", "flag", "md", str)
|
|
|
|
)
|
|
|
|
def info(self, model=None, markdown=False):
|
|
|
|
"""
|
|
|
|
Print info about spaCy installation. If a model shortcut link is
|
|
|
|
speficied as an argument, print model information. Flag --markdown
|
|
|
|
prints details in Markdown for easy copy-pasting to GitHub issues.
|
|
|
|
"""
|
|
|
|
|
|
|
|
cli_info(model, markdown)
|
|
|
|
|
|
|
|
|
2017-03-20 21:50:13 +00:00
|
|
|
@plac.annotations(
|
|
|
|
input_dir=("directory with model data", "positional", None, str),
|
2017-03-21 01:05:34 +00:00
|
|
|
output_dir=("output directory", "positional", None, str),
|
2017-03-21 10:19:21 +00:00
|
|
|
force=("force overwriting of existing folder in output directory", "flag", "f", bool)
|
2017-03-20 21:50:13 +00:00
|
|
|
)
|
2017-03-21 01:05:34 +00:00
|
|
|
def package(self, input_dir, output_dir, force=False):
|
2017-03-20 21:50:13 +00:00
|
|
|
"""
|
|
|
|
Generate Python package for model data, including meta and required
|
|
|
|
installation files. A new directory will be created in the specified
|
2017-03-20 21:50:55 +00:00
|
|
|
output directory, and model data will be copied over.
|
2017-03-20 21:50:13 +00:00
|
|
|
"""
|
|
|
|
|
2017-03-21 01:05:34 +00:00
|
|
|
cli_package(input_dir, output_dir, force)
|
2017-03-20 21:50:13 +00:00
|
|
|
|
|
|
|
|
2017-03-18 14:14:48 +00:00
|
|
|
def __missing__(self, name):
|
|
|
|
print("\n Command %r does not exist\n" % name)
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
import plac
|
|
|
|
import sys
|
|
|
|
cli = CLI()
|
|
|
|
sys.argv[0] = 'spacy'
|
|
|
|
plac.Interpreter.call(CLI)
|