setup.py 1.0 KB
Newer Older
O
Olli-Pekka Heinisuo 已提交
1
from setuptools import setup
2 3
from setuptools.dist import Distribution
import pip
O
Olli-Pekka Heinisuo 已提交
4 5 6
import os
import sys

7 8 9 10 11 12 13 14 15 16
opencv_version = ""

if "--opencv-version" in sys.argv:
    index = sys.argv.index('--opencv-version')
    sys.argv.pop(index)
    opencv_version = sys.argv.pop(index)
else:
    print("Error: no version info (--opencv-version missing), exiting.")
    exit(1)

17 18 19 20 21 22 23 24 25 26 27 28
numpy_version = ""

# Get required numpy version
for package in pip.get_installed_distributions():
    if package.key == "numpy":
        numpy_version = package.version

class BinaryDistribution(Distribution):
    """ Forces BinaryDistribution. """
    def has_ext_modules(asd):
        return True

O
Olli-Pekka Heinisuo 已提交
29 30 31
package_data = {}

if os.name == 'posix':
O
fixes  
Olli-Pekka Heinisuo 已提交
32
    package_data['cv2'] = ['*.so']
O
Olli-Pekka Heinisuo 已提交
33
else:
O
fixes  
Olli-Pekka Heinisuo 已提交
34
    package_data['cv2'] = ['*.pyd']
O
Olli-Pekka Heinisuo 已提交
35 36 37 38

setup(name='opencv-python',
      version=opencv_version,
      description='OpenCV',
39
      distclass=BinaryDistribution,
O
Olli-Pekka Heinisuo 已提交
40 41
      packages=['cv2'],
      package_data=package_data,
42
      install_requires="numpy==%s" % numpy_version,
O
Olli-Pekka Heinisuo 已提交
43
      )