2017-03-16 19:13:08 +03:00
|
|
|
# coding: utf8
|
2017-04-15 13:05:47 +03:00
|
|
|
from __future__ import unicode_literals
|
2017-03-16 19:13:08 +03:00
|
|
|
|
|
|
|
from pathlib import Path
|
2017-04-15 13:05:47 +03:00
|
|
|
|
2017-03-16 19:13:08 +03:00
|
|
|
from .util import set_lang_class, get_lang_class, parse_package_meta
|
2017-03-17 20:47:05 +03:00
|
|
|
from .deprecated import resolve_model_name
|
2017-03-22 12:02:14 +03:00
|
|
|
from .cli import info
|
2016-01-15 20:01:02 +03:00
|
|
|
|
2017-04-15 13:05:47 +03:00
|
|
|
from . import en, de, zh, es, it, hu, fr, pt, nl, sv, fi, bn, he
|
2017-03-18 21:30:03 +03:00
|
|
|
|
2017-03-15 19:33:57 +03:00
|
|
|
|
2016-04-12 17:31:16 +03:00
|
|
|
set_lang_class(en.English.lang, en.English)
|
|
|
|
set_lang_class(de.German.lang, de.German)
|
2016-11-02 22:05:14 +03:00
|
|
|
set_lang_class(es.Spanish.lang, es.Spanish)
|
|
|
|
set_lang_class(pt.Portuguese.lang, pt.Portuguese)
|
|
|
|
set_lang_class(fr.French.lang, fr.French)
|
|
|
|
set_lang_class(it.Italian.lang, it.Italian)
|
2016-12-08 01:07:43 +03:00
|
|
|
set_lang_class(hu.Hungarian.lang, hu.Hungarian)
|
2016-04-24 19:45:16 +03:00
|
|
|
set_lang_class(zh.Chinese.lang, zh.Chinese)
|
2016-11-29 18:42:05 +03:00
|
|
|
set_lang_class(nl.Dutch.lang, nl.Dutch)
|
2016-12-20 13:05:06 +03:00
|
|
|
set_lang_class(sv.Swedish.lang, sv.Swedish)
|
2017-02-01 01:27:29 +03:00
|
|
|
set_lang_class(fi.Finnish.lang, fi.Finnish)
|
2017-02-28 04:48:37 +03:00
|
|
|
set_lang_class(bn.Bengali.lang, bn.Bengali)
|
2017-03-29 11:34:57 +03:00
|
|
|
set_lang_class(he.Hebrew.lang, he.Hebrew)
|
2017-02-01 01:27:29 +03:00
|
|
|
|
2016-03-25 20:54:45 +03:00
|
|
|
|
2016-10-18 20:23:31 +03:00
|
|
|
def load(name, **overrides):
|
2016-10-20 19:27:48 +03:00
|
|
|
data_path = overrides.get('path', util.get_data_path())
|
2017-03-17 20:47:05 +03:00
|
|
|
model_name = resolve_model_name(name)
|
2017-03-17 21:51:45 +03:00
|
|
|
meta = parse_package_meta(data_path, model_name, require=False)
|
2017-03-17 02:30:02 +03:00
|
|
|
lang = meta['lang'] if meta and 'lang' in meta else name
|
2017-03-16 19:13:08 +03:00
|
|
|
cls = get_lang_class(lang)
|
|
|
|
overrides['meta'] = meta
|
2017-03-17 21:51:45 +03:00
|
|
|
model_path = Path(data_path / model_name)
|
2017-03-17 02:30:02 +03:00
|
|
|
if model_path.exists():
|
|
|
|
overrides['path'] = model_path
|
2017-03-17 21:51:45 +03:00
|
|
|
|
2016-12-19 00:15:29 +03:00
|
|
|
return cls(**overrides)
|