#!/usr/bin/env?python
#?coding:?utf8
from?__future__?import?unicode_literals

import?io
import?json
from?os?import?path?walk
from?shutil?import?copy
from?setuptools?import?setup


def?load_meta（fp）:
????with?io.open（fp?encoding=‘utf8‘）?as?f:
????????return?json.load（f）


def?list_files（data_dir）:
????output?=?[]
????for?root?_?filenames?in?walk（data_dir）:
????????for?filename?in?filenames:
????????????if?not?filename.startswith（‘.‘）:
????????????????output.append（path.join（root?filename））
????output?=?[path.relpath（p?path.dirname（data_dir））?for?p?in?output]
????output.append（‘meta.json‘）
????return?output


def?list_requirements（meta）:
????parent_package?=?meta.get（‘parent_package‘?‘spacy‘）
????requirements?=?[parent_package?+?meta[‘spacy_version‘]]
????if?‘setup_requires‘?in?meta:
????????requirements?+=?meta[‘setup_requires‘]
????return?requirements


def?setup_package（）:
????root?=?path.abspath（path.dirname（__file__））
????meta_path?=?path.join（root?‘meta.json‘）
????meta?=?load_meta（meta_path）
????model_name?=?str（meta[‘lang‘]?+?‘_‘?+?meta[‘name‘]）
????model_dir?=?path.join（model_name?model_name?+?‘-‘?+?meta[‘version‘]）

????copy（meta_path?path.join（model_name））
????copy（meta_path?model_dir）

????setup（
????????name=model_name
????????description=meta[‘description‘]
????????author=meta[‘author‘]
????????author_email=meta[‘email‘]
????????url=meta[‘url‘]
????????version=meta[‘version‘]
????????license=meta[‘license‘]
????????packages=[model_name]
????????package_data={model_name:?list_files（model_dir）}
????????install_requires=list_requirements（meta）
????????zip_safe=False
????）


if?__name__?==?‘__main__‘:
????setup_package（）