mirror of
https://github.com/explosion/spaCy.git
synced 2024-09-21 19:39:13 +03:00
55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
import sys
|
|
import os
|
|
import shutil
|
|
|
|
import plac
|
|
from sputnik import Sputnik
|
|
|
|
|
|
def migrate(path):
|
|
data_path = os.path.join(path, 'data')
|
|
if os.path.isdir(data_path) and not os.path.islink(data_path):
|
|
shutil.rmtree(data_path)
|
|
for filename in os.listdir(path):
|
|
if filename.endswith('.tgz'):
|
|
os.unlink(os.path.join(path, filename))
|
|
|
|
|
|
def link(package, path):
|
|
if os.path.exists(path):
|
|
os.unlink(path)
|
|
os.symlink(package.dir_path('data'), path)
|
|
|
|
|
|
@plac.annotations(
|
|
force=("Force overwrite", "flag", "f", bool),
|
|
)
|
|
def main(data_size='all', force=False):
|
|
# TODO read version from the same source as the setup
|
|
sputnik = Sputnik('spacy', '0.99.0', console=sys.stdout)
|
|
|
|
path = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
data_path = os.path.abspath(os.path.join(path, '..', 'data'))
|
|
if not os.path.isdir(data_path):
|
|
os.mkdir(data_path)
|
|
|
|
command = sputnik.make_command(
|
|
data_path=data_path,
|
|
repository_url='https://index.spacy.io')
|
|
|
|
if force:
|
|
command.purge()
|
|
|
|
package = command.install('en_default')
|
|
|
|
# FIXME clean up old-style packages
|
|
migrate(path)
|
|
|
|
# FIXME supply spacy with an old-style data dir
|
|
link(package, os.path.join(path, 'data'))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
plac.call(main)
|