mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-14 13:47:13 +03:00
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
# coding: utf8
|
|
from __future__ import unicode_literals
|
|
|
|
from pathlib import Path
|
|
|
|
from .converters import conllu2json
|
|
from .. import util
|
|
|
|
|
|
# Converters are matched by file extension. To add a converter, add a new entry
|
|
# to this dict with the file extension mapped to the converter function imported
|
|
# from /converters.
|
|
|
|
CONVERTERS = {
|
|
'.conllu': conllu2json
|
|
}
|
|
|
|
|
|
def convert(input_file, output_dir, *args):
|
|
input_path = Path(input_file)
|
|
output_path = Path(output_dir)
|
|
check_dirs(input_path, output_path)
|
|
file_ext = input_path.suffix
|
|
if file_ext in CONVERTERS:
|
|
CONVERTERS[file_ext](input_path, output_path, *args)
|
|
else:
|
|
util.sys_exit("Can't find converter for {}".format(input_path.parts[-1]),
|
|
title="Unknown format")
|
|
|
|
|
|
def check_dirs(input_file, output_path):
|
|
if not input_file.exists():
|
|
util.sys_exit(input_file.as_posix(), title="Input file not found")
|
|
if not output_path.exists():
|
|
util.sys_exit(output_path.as_posix(), title="Output directory not found")
|