mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-11-30 10:59:43 +08:00
322 lines
12 KiB
C++
322 lines
12 KiB
C++
#!/usr/bin/env python
|
|
#-*- coding:utf-8 -*-
|
|
|
|
from setuptools import setup, find_packages
|
|
import sys
|
|
import os
|
|
import shutil
|
|
import platform
|
|
|
|
def print_help():
|
|
print("python setup.py help -- 查看帮助")
|
|
print("python setup.py -- 执行编译")
|
|
print("python setup.py build -- 执行编译")
|
|
print("python setup.py install -- 执行安装")
|
|
print("python setup.py uninstall -- 删除已安装的Hikyuu")
|
|
print("python setup.py clear -- 清除本地编译结果")
|
|
print("python setup.py bdist_wheel -- 生成wheel安装包")
|
|
|
|
current_plat = sys.platform
|
|
current_bits = 64 if sys.maxsize > 2**32 else 32
|
|
|
|
if len(sys.argv) < 1 or (len(sys.argv) > 1 and sys.argv[1] not in ('bdist_wheel', 'build', 'install', 'uninstall', 'clear')):
|
|
print("参数错误，请参考以下命令执行：")
|
|
print_help()
|
|
exit(0)
|
|
|
|
print("\nchecking python version ...")
|
|
|
|
py_version = platform.python_version_tuple()
|
|
py_version = int(py_version[0])*10 + int(py_version[1])
|
|
if py_version < 31:
|
|
print("Python version must >= 3.1 !")
|
|
sys.exit(0)
|
|
|
|
py_version_changed = True
|
|
if os.path.exists('py_version'):
|
|
with open('py_version', 'r+') as f:
|
|
old_py_version = f.read()
|
|
if old_py_version == str(py_version):
|
|
py_version_changed = False
|
|
else:
|
|
f.seek(0)
|
|
f.write(str(py_version))
|
|
print('old python version:', int(old_py_version)*0.1)
|
|
else:
|
|
with open('py_version', 'w') as f:
|
|
f.write(str(py_version))
|
|
|
|
print('current python version:', int(py_version)*0.1, '\n')
|
|
|
|
if sys.argv[-1] == 'uninstall':
|
|
if current_plat == 'win32':
|
|
site_lib_dir = sys.base_prefix + "/lib/site-packages"
|
|
else:
|
|
usr_dir = os.path.expanduser('~')
|
|
site_lib_dir = '{}/.local/lib/python{:>.1f}/site-packages'.format(usr_dir, py_version*0.1)
|
|
for dir in os.listdir(site_lib_dir):
|
|
if dir == 'hikyuu' or (len(dir) > 6 and dir[:6] == 'Hikyuu'):
|
|
print('delete', site_lib_dir + '/' + dir)
|
|
shutil.rmtree(site_lib_dir + '/' + dir)
|
|
print("Uninstall finished!")
|
|
exit(0)
|
|
|
|
if sys.argv[-1] == 'clear':
|
|
if os.path.lexists('.xmake'):
|
|
print('delete .xmake')
|
|
shutil.rmtree('.xmake')
|
|
if os.path.lexists('build'):
|
|
print('delete build')
|
|
shutil.rmtree('build')
|
|
if os.path.lexists('Hikyuu.egg-info'):
|
|
print('delete Hikyuu.egg-info')
|
|
shutil.rmtree('Hikyuu.egg-info')
|
|
if os.path.exists('py_version'):
|
|
print('delete py_version')
|
|
os.remove('py_version')
|
|
for r, d, f_list in os.walk('hikyuu'):
|
|
for name in f_list:
|
|
if (len(name) > 4 and name[-4:] in ('.dll','.exe','.pyd')) \
|
|
or (len(name) > 3 and name[-3:] == '.so') \
|
|
or (len(name) > 6 and name[-6:] == '.dylib'):
|
|
print('delete', r + '/' + name)
|
|
os.remove(os.path.join(r, name))
|
|
print('clear finished!')
|
|
exit(0)
|
|
|
|
print("checking xmake ...")
|
|
xmake = os.system("xmake --version")
|
|
|
|
if xmake != 0:
|
|
print("Please read https://hikyuu.readthedocs.io/en/latest/developer.html")
|
|
exit(0)
|
|
|
|
print("checking boost ...")
|
|
current_dir = os.getcwd()
|
|
current_boost_root = None
|
|
current_boost_lib = None
|
|
if 'BOOST_ROOT' not in os.environ or 'BOOST_LIB' not in os.environ:
|
|
for dir in os.listdir():
|
|
if len(dir) >= 5 and dir[:5] == 'boost' and os.path.isdir(dir):
|
|
current_boost_root = current_dir + '/' + dir
|
|
current_boost_lib = current_dir + '/' + dir + '/stage/lib'
|
|
os.environ['BOOST_ROOT'] = current_boost_root
|
|
os.environ['BOOST_LIB'] = current_boost_lib
|
|
else:
|
|
current_boost_root = os.environ['BOOST_ROOT']
|
|
current_boost_lib = os.environ['BOOST_LIB']
|
|
|
|
if current_boost_root is None or current_boost_lib is None:
|
|
print("Please read https://hikyuu.readthedocs.io/en/latest/developer.html")
|
|
exit(0)
|
|
|
|
print('BOOST_ROOT:', current_boost_root)
|
|
print('BOOST_LIB:', current_boost_lib)
|
|
|
|
|
|
if py_version_changed:
|
|
if current_plat == 'win32' and current_bits == 64:
|
|
build_hikyuu_pywrap_dir = 'build\\release\\windows\\x64\\.objs\\windows\\x64\\release\\hikyuu_pywrap'
|
|
elif current_plat == 'win32' and current_bits == 32:
|
|
build_hikyuu_pywrap_dir = 'build\\release\\windows\\x64\\.objs\\windows\\x64\\release\\hikyuu_pywrap'
|
|
elif current_plat == 'linux' and current_bits == 64:
|
|
build_hikyuu_pywrap_dir = 'build/release/linux/x86_64/.objs/linux/x86_64/release/hikyuu_pywrap'
|
|
else:
|
|
print("************不支持的平台打包**************")
|
|
exit(0)
|
|
|
|
if os.path.lexists(build_hikyuu_pywrap_dir):
|
|
shutil.rmtree(build_hikyuu_pywrap_dir)
|
|
|
|
print('\ncompile boost ...')
|
|
if py_version_changed and os.path.lexists('{}/bin.v2/libs/python'.format(current_boost_root)):
|
|
shutil.rmtree('{}/bin.v2/libs/python'.format(current_boost_root))
|
|
if current_plat == 'win32':
|
|
os.chdir(current_boost_root)
|
|
if not os.path.exists('b2.exe'):
|
|
os.system('bootstrap.bat')
|
|
os.system('b2 release link=shared address-model=64 -j 4 --with-python --with-date_time --with-filesystem --with-system --with-serialization --with-test')
|
|
os.chdir(current_dir)
|
|
else:
|
|
cmd = 'cd {boost} ; if [ ! -f "b2" ]; then ./bootstrap.sh ; fi; ./b2 release link=shared address-model=64 -j 4 --with-python --with-date_time --with-filesystem --with-system --with-serialization --with-test; cd {current}'.format(boost=current_boost_root, current=current_dir)
|
|
#print(cmd)
|
|
os.system(cmd)
|
|
|
|
|
|
if len(sys.argv) == 1 or sys.argv[-1] == 'build':
|
|
if py_version_changed:
|
|
os.system("xmake f -c")
|
|
os.system("xmake f --with-unit-test=y")
|
|
os.system("xmake")
|
|
os.system("xmake r unit-test")
|
|
sys.exit(0)
|
|
|
|
if sys.argv[-1] == 'install':
|
|
if py_version_changed:
|
|
os.system("xmake f -c")
|
|
if current_plat == 'win32':
|
|
install_dir = sys.base_prefix + "\\Lib\\site-packages\\hikyuu"
|
|
else:
|
|
usr_dir = os.path.expanduser('~')
|
|
if not os.path.lexists(usr_dir + '/.local'):
|
|
os.mkdir(usr_dir + '/.local')
|
|
if not os.path.lexists('{}/.local/lib'.format(usr_dir)):
|
|
os.mkdir('{}/.local/lib'.format(usr_dir))
|
|
if not os.path.lexists('{}/.local/lib/python{:>.1f}'.format(usr_dir, py_version*0.1)):
|
|
os.mkdir('{}/.local/lib/python{:>.1f}'.format(usr_dir, py_version*0.1))
|
|
os.mkdir('{}/.local/lib/python{:>.1f}/site-packages'.format(usr_dir, py_version*0.1))
|
|
if not os.path.lexists('{}/.local/lib/python{:>.1f}/site-packages'.format(usr_dir, py_version*0.1)):
|
|
os.mkdir('{}/.local/lib/python{:>.1f}/site-packages'.format(usr_dir, py_version*0.1))
|
|
install_dir = '{}/.local/lib/python{:>.1f}/site-packages/hikyuu'.format(usr_dir, py_version*0.1)
|
|
os.system("xmake install -o " + install_dir)
|
|
sys.exit(0)
|
|
|
|
if sys.argv[-1] == 'bdist_wheel':
|
|
if py_version_changed:
|
|
os.system("xmake f -c")
|
|
os.system("xmake")
|
|
sys.argv.append("--python-tag")
|
|
sys.argv.append("cp{}".format(py_version))
|
|
sys.argv.append("-p")
|
|
if current_plat == 'win32' and current_bits == 64:
|
|
sys.argv.append("win-amd64")
|
|
elif current_plat == 'win32' and current_bits == 32:
|
|
sys.argv.append("win32")
|
|
elif current_plat == 'linux' and current_bits == 64:
|
|
sys.argv.append("manylinux1_x86_64")
|
|
elif current_plat == 'linux' and current_bits == 32:
|
|
sys.argv.append("manylinux1_i386")
|
|
else:
|
|
print("*********尚未实现*******")
|
|
|
|
|
|
if os.path.lexists('Hikyuu.egg-info'):
|
|
shutil.rmtree('Hikyuu.egg-info')
|
|
|
|
if os.path.lexists('build/lib'):
|
|
shutil.rmtree('build/lib')
|
|
|
|
for bdist in os.listdir('build'):
|
|
if len(bdist) >= 5 and bdist[:5] == 'bdist' and os.path.lexists(bdist):
|
|
shutil.rmtree(bdist)
|
|
|
|
for x in os.listdir('hikyuu'):
|
|
if x[:12] == 'boost_python':
|
|
if x[12:14] != str(py_version):
|
|
os.remove('hikyuu/{}'.format(x))
|
|
|
|
hku_version = ''
|
|
with open('xmake.lua', 'r') as f:
|
|
for line in f:
|
|
if len(line) > 15 and line[:11] == 'set_version':
|
|
pre_pos = line.find('"') + 1
|
|
end_pos = line.find('"', pre_pos)
|
|
hku_version = line[pre_pos : end_pos]
|
|
break
|
|
|
|
if not hku_version:
|
|
print("无法在 xmake.lua 中找到 set_version 语句，获取版本号失败！")
|
|
exit(0)
|
|
|
|
print('current hikyuu version:', hku_version)
|
|
|
|
hku_name = "Hikyuu"
|
|
#hku_version = "1.0.9"
|
|
hku_author = "fasiondog"
|
|
hku_author_email = "fasiondog@sina.com"
|
|
|
|
hku_license = "MIT"
|
|
hku_keywords = ("quant", "trade", "System Trading", "backtester", "量化", "程序化交易", "量化交易", "系统交易")
|
|
hku_platforms = "Independant"
|
|
hku_url = "http://hikyuu.org/"
|
|
|
|
hku_description = "Hikyuu Quant Framework for System Trading Analysis and backtester"
|
|
with open("README.rst", encoding='utf-8') as f:
|
|
hku_long_description = f.read()
|
|
|
|
hku_data_files = []
|
|
|
|
packages = ['hikyuu',
|
|
'hikyuu/config',
|
|
'hikyuu/data',
|
|
'hikyuu/data/mysql_upgrade',
|
|
'hikyuu/data/sqlite_upgrade',
|
|
'hikyuu/data_driver',
|
|
'hikyuu/examples',
|
|
'hikyuu/gui',
|
|
'hikyuu/gui/data',
|
|
'hikyuu/indicator',
|
|
'hikyuu/interactive',
|
|
'hikyuu/interactive/drawplot',
|
|
#'hikyuu/test',
|
|
'hikyuu/tools',
|
|
'hikyuu/trade_instance',
|
|
'hikyuu/trade_manage',
|
|
'hikyuu/trade_sys',
|
|
'hikyuu/util',
|
|
]
|
|
|
|
setup(
|
|
name = hku_name,
|
|
version = hku_version,
|
|
description = hku_description,
|
|
long_description = hku_long_description,
|
|
author = hku_author,
|
|
author_email = hku_author_email,
|
|
|
|
license = hku_license,
|
|
keywords = hku_keywords,
|
|
platforms = hku_platforms,
|
|
url = hku_url,
|
|
|
|
packages = packages, #find_packages(),
|
|
zip_safe = False,
|
|
include_package_data=True,
|
|
package_data = {
|
|
'': ['*.rst', '*.pyd', '*.ini', '*.sql', '*.properties', '*.xml', 'LICENSE.txt', '*.dll', '*.exe', '*.ico'],
|
|
},
|
|
|
|
data_files = hku_data_files,
|
|
|
|
classifiers=[
|
|
# How mature is this project? Common values are
|
|
# 3 - Alpha
|
|
# 4 - Beta
|
|
# 5 - Production/Stable
|
|
'Development Status :: 3 - Alpha',
|
|
|
|
# Indicate who your project is intended for
|
|
'Intended Audience :: Developers',
|
|
'Topic :: Software Development :: Libraries :: Python Modules',
|
|
'Topic :: Office/Business :: Financial',
|
|
'Topic :: Office/Business :: Financial :: Investment',
|
|
'Topic :: Scientific/Engineering :: Mathematics',
|
|
|
|
# Pick your license as you wish (should match "license" above)
|
|
'License :: OSI Approved :: MIT License',
|
|
|
|
'Operating System :: Microsoft :: Windows',
|
|
|
|
# Specify the Python versions you support here. In particular, ensure
|
|
# that you indicate whether you support Python 2, Python 3 or both.
|
|
'Programming Language :: Python :: 3',
|
|
'Programming Language :: Python :: 3.5',
|
|
],
|
|
|
|
entry_points={
|
|
'gui_scripts': ['HikyuuTDX=hikyuu.gui.HikyuuTDX:start',],
|
|
'console_scripts': ['importdata=hikyuu.gui.importdata:main',]
|
|
},
|
|
|
|
install_requires=[
|
|
'matplotlib>=1.5.0',
|
|
'pandas>=0.17.1',
|
|
#'tushare>=0.8.2',
|
|
'pytdx',
|
|
'pyperclip>=1.5.27',
|
|
'pyecharts>=0.2.7',
|
|
'pyecharts_snapshot',
|
|
'PyQt5',
|
|
],
|
|
)
|