104 строки
3.4 KiB
Python
104 строки
3.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License. See License.txt in the project root for
|
|
# license information.
|
|
###########################################################################
|
|
|
|
import os
|
|
import sys
|
|
import pathlib
|
|
import setuptools
|
|
|
|
from setuptools import setup, find_packages
|
|
|
|
TOP_DIR = os.path.dirname(__file__) or os.getcwd()
|
|
PACKAGE_NAME = "onnxruntime_extensions"
|
|
|
|
# setup.py cannot be debugged in pip command line, so the command classes are refactored into another file
|
|
cmds_dir = pathlib.Path(TOP_DIR) / ".pyproject"
|
|
sys.path.append(str(cmds_dir))
|
|
# noinspection PyUnresolvedReferences
|
|
import cmdclass as _cmds # noqa: E402
|
|
|
|
_cmds.prepare_env(TOP_DIR)
|
|
|
|
|
|
# read version from the package file.
|
|
def read_version():
|
|
version_str = "0.1.0"
|
|
with open(os.path.join(TOP_DIR, "version.txt"), "r") as f:
|
|
version_str = f.readline().strip()
|
|
|
|
# special handling for Onebranch building
|
|
if os.getenv("BUILD_SOURCEBRANCHNAME", "").startswith("rel-"):
|
|
return version_str
|
|
|
|
# is it a dev build or release?
|
|
rel_br, cid = _cmds.read_git_refs(TOP_DIR) if os.path.isdir(os.path.join(TOP_DIR, ".git")) else (True, None)
|
|
|
|
if rel_br:
|
|
return version_str
|
|
|
|
build_id = os.getenv("BUILD_BUILDID", None)
|
|
if build_id is not None:
|
|
version_str += ".{}".format(build_id)
|
|
else:
|
|
version_str += "+" + cid[:7]
|
|
return version_str
|
|
|
|
|
|
def write_py_version(ext_version):
|
|
text = ["# Generated by setup.py, DON'T MANUALLY UPDATE IT!\n", '__version__ = "{}"\n'.format(ext_version)]
|
|
with open(os.path.join(TOP_DIR, "onnxruntime_extensions/_version.py"), "w") as _fver:
|
|
_fver.writelines(text)
|
|
|
|
|
|
ext_modules = [setuptools.extension.Extension(name=str("onnxruntime_extensions._extensions_pydll"), sources=[])]
|
|
|
|
packages = find_packages()
|
|
package_dir = {k: os.path.join(".", k.replace(".", "/")) for k in packages}
|
|
package_data = {
|
|
"onnxruntime_extensions": ["*.so", "*.pyd"],
|
|
}
|
|
|
|
long_description = ""
|
|
with open(os.path.join(TOP_DIR, "README.md"), "r", encoding="utf-8") as _f:
|
|
long_description += _f.read()
|
|
start_pos = long_description.find("# Introduction")
|
|
start_pos = 0 if start_pos < 0 else start_pos
|
|
end_pos = long_description.find("# Contributing")
|
|
long_description = long_description[start_pos:end_pos]
|
|
ortx_version = read_version()
|
|
write_py_version(ortx_version)
|
|
|
|
setup(
|
|
name=PACKAGE_NAME,
|
|
version=ortx_version,
|
|
packages=packages,
|
|
package_dir=package_dir,
|
|
package_data=package_data,
|
|
description="ONNXRuntime Extensions",
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
license="MIT License",
|
|
author="Microsoft Corporation",
|
|
author_email="onnxruntime@microsoft.com",
|
|
url="https://github.com/microsoft/onnxruntime-extensions",
|
|
ext_modules=ext_modules,
|
|
cmdclass=_cmds.ortx_cmdclass,
|
|
include_package_data=True,
|
|
install_requires=[],
|
|
classifiers=[
|
|
"Development Status :: 4 - Beta",
|
|
"Environment :: Console",
|
|
"Intended Audience :: Developers",
|
|
"Operating System :: MacOS :: MacOS X",
|
|
"Operating System :: Microsoft :: Windows",
|
|
"Operating System :: POSIX :: Linux",
|
|
"Programming Language :: C++",
|
|
"Programming Language :: Python",
|
|
"Programming Language :: Python :: Implementation :: CPython",
|
|
"License :: OSI Approved :: MIT License",
|
|
],
|
|
)
|