mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-11 04:08:09 +03:00
98 lines
3.3 KiB
Python
98 lines
3.3 KiB
Python
# coding: utf8
|
|
from __future__ import unicode_literals
|
|
|
|
import json
|
|
import shutil
|
|
import requests
|
|
from pathlib import Path
|
|
|
|
import six
|
|
|
|
from .. import about
|
|
from .. import util
|
|
|
|
if six.PY2:
|
|
json_dumps = lambda data: json.dumps(data, indent=2).decode("utf8")
|
|
elif six.PY3:
|
|
json_dumps = lambda data: json.dumps(data, indent=2)
|
|
|
|
def package(input_dir, output_dir, force):
|
|
input_path = Path(input_dir)
|
|
output_path = Path(output_dir)
|
|
check_dirs(input_path, output_path)
|
|
|
|
template_setup = get_template('setup.py')
|
|
template_manifest = get_template('MANIFEST.in')
|
|
template_init = get_template('en_model_name/__init__.py')
|
|
meta = generate_meta()
|
|
|
|
model_name = meta['lang'] + '_' + meta['name']
|
|
model_name_v = model_name + '-' + meta['version']
|
|
main_path = output_path / model_name_v
|
|
package_path = main_path / model_name
|
|
|
|
create_dirs(package_path, force)
|
|
shutil.copytree(input_path.as_posix(), (package_path / model_name_v).as_posix())
|
|
create_file(main_path / 'meta.json', json_dumps(meta))
|
|
create_file(main_path / 'setup.py', template_setup)
|
|
create_file(main_path / 'MANIFEST.in', template_manifest)
|
|
create_file(package_path / '__init__.py', template_init)
|
|
|
|
util.print_msg(
|
|
main_path.as_posix(),
|
|
"To build the package, run `python setup.py sdist` in that directory.",
|
|
title="Successfully created package {p}".format(p=model_name_v))
|
|
|
|
|
|
def check_dirs(input_path, output_path):
|
|
if not input_path.exists():
|
|
util.sys_exit(input_path.as_poisx(), title="Model directory not found")
|
|
if not output_path.exists():
|
|
util.sys_exit(output_path.as_posix(), title="Output directory not found")
|
|
|
|
|
|
def create_dirs(package_path, force):
|
|
if package_path.exists():
|
|
if force:
|
|
shutil.rmtree(package_path.as_posix())
|
|
else:
|
|
util.sys_exit(package_path.as_posix(),
|
|
"Please delete the directory and try again.",
|
|
title="Package directory already exists")
|
|
Path.mkdir(package_path, parents=True)
|
|
|
|
|
|
def create_file(file_path, contents):
|
|
file_path.touch()
|
|
file_path.open('w', encoding='utf-8').write(contents)
|
|
|
|
|
|
def generate_meta():
|
|
settings = [('lang', 'Model language', 'en'),
|
|
('name', 'Model name', 'model'),
|
|
('version', 'Model version', '0.0.0'),
|
|
('spacy_version', 'Required spaCy version', '>=1.7.0,<2.0.0'),
|
|
('description', 'Model description', False),
|
|
('author', 'Author', False),
|
|
('email', 'Author email', False),
|
|
('url', 'Author website', False),
|
|
('license', 'License', 'CC BY-NC 3.0')]
|
|
|
|
util.print_msg("Enter the package settings for your model.", title="Generating meta.json")
|
|
|
|
meta = {}
|
|
for setting, desc, default in settings:
|
|
response = util.get_raw_input(desc, default)
|
|
meta[setting] = default if response == '' and default else response
|
|
return meta
|
|
|
|
|
|
def get_template(filepath):
|
|
url = 'https://raw.githubusercontent.com/explosion/spacy-dev-resources/master/templates/model/'
|
|
r = requests.get(url + filepath)
|
|
if r.status_code != 200:
|
|
util.sys_exit(
|
|
"Couldn't fetch template files from GitHub.",
|
|
title="Server error ({c})".format(c=r.status_code))
|
|
return r.text
|