From cfe58d3017b48a8f99d3dd29cabcc6e52bd8d3ae Mon Sep 17 00:00:00 2001 From: "gw0 [http://gw.tnode.com/]" Date: Mon, 26 Jan 2015 22:13:52 +0100 Subject: [PATCH] Add support for TA-Lib binary installed in virtualenv. --- setup.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/setup.py b/setup.py index 64125c5f9..6f6280de2 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,5 @@ from distutils.core import setup +from distutils.command.install import install from distutils.extension import Extension from Cython.Distutils import build_ext @@ -20,6 +21,7 @@ '/usr/local/include', '/opt/include', '/opt/local/include', + os.getenv('VIRTUAL_ENV', './venv') + '/include', ] lib_talib_dirs = [ '/usr/lib', @@ -28,6 +30,7 @@ '/usr/local/lib64', '/opt/lib', '/opt/local/lib', + os.getenv('VIRTUAL_ENV', './venv') + '/lib', ] break @@ -61,6 +64,12 @@ ) ext_modules.append(ext) +# Hack to install ta-lib library loading into virtualenv +def install_virtualenv_lib_loader(): + with open(os.getenv('VIRTUAL_ENV', './venv') + '/bin/activate', 'a') as f: + f.write("\nexport LD_LIBRARY_PATH=$VIRTUAL_ENV/lib:$LD_LIBRARY_PATH\n") +install_virtualenv_lib_loader() + setup( name = 'TA-Lib', version = '0.4.8',