from setuptools import setup, Distribution, Extension

import subprocess

class BinaryDistribution(Distribution):

def has_ext_modules(foo):

return True

MAJOR = 0

MINOR = 11

PATCH = 0

RC = 0

ISTAGED = False

def git_commit():

try:

cmd = ['git', 'rev-parse', 'HEAD']

git_commit = subprocess.Popen(cmd, stdout = subprocess.PIPE).communicate()[0].strip()

except:

git_commit = 'Unknown'

return git_commit

def write_version_py(filename='paddle/version.py'):

cnt = '''

# THIS FILE IS GENERATED FROM PADDLEPADDLE SETUP.PY

#

full_version = '%(major)d.%(minor)d.%(patch)d'

major = '%(major)d'

minor = '%(minor)d'

patch = '%(patch)d'

rc = '%(rc)d'

istaged = %(istaged)s

commit = '%(commit)s'

with_mkl = '%(with_mkl)s'

def show():

if istaged:

print 'full_version:', full_version

print 'major:', major

print 'minor:', minor

print 'patch:', patch

print 'rc:', rc

else:

print 'commit:', commit

def mkl():

return with_mkl

'''

commit = git_commit()

with open(filename, 'w') as f:

f.write(cnt % {

'major': MAJOR,

'minor': MINOR,

'patch': PATCH,

'rc': RC,

'version': '${PADDLE_VERSION}',

'commit': commit,

'istaged': ISTAGED,

'with_mkl': '@WITH_MKL@'})

write_version_py(filename='@PADDLE_SOURCE_DIR@/python/paddle/version.py')

packages=['paddle',

'paddle.proto',

'paddle.trainer',

'paddle.trainer_config_helpers',

'paddle.utils',

'paddle.v2',

'paddle.v2.dataset',

'paddle.v2.reader',

'paddle.v2.master',

'paddle.v2.plot',

'paddle.v2.fluid',

'paddle.v2.fluid.proto',

'paddle.v2.fluid.layers',

'py_paddle']

with open('@PADDLE_SOURCE_DIR@/python/requirements.txt') as f:

setup_requires = f.read().splitlines()

if '${CMAKE_SYSTEM_PROCESSOR}' not in ['arm', 'armv7-a', 'aarch64']:

setup_requires+=['opencv-python']

# the prefix is sys.prefix which should always be usr

paddle_bin_dir = 'opt/paddle/bin'

paddle_bins = ['${PADDLE_BINARY_DIR}/paddle/trainer/paddle_trainer',

'${PADDLE_BINARY_DIR}/paddle/trainer/paddle_merge_model',

'${PADDLE_BINARY_DIR}/paddle/pserver/paddle_pserver_main',

'${PADDLE_BINARY_DIR}/paddle/scripts/paddle']

paddle_rt_lib_dir = 'lib'

paddle_rt_libs = ['${WARPCTC_LIBRARIES}']

if '${MKL_SHARED_LIBS}'!= '':

paddle_rt_libs += '${MKL_SHARED_LIBS}'.split(';')

setup(name='${PACKAGE_NAME}',

version='${PADDLE_VERSION}',

description='Parallel Distributed Deep Learning',

install_requires=setup_requires,

packages=packages,

ext_modules=[Extension('_foo', ['stub.cc'])],

package_data={

'paddle.v2.master': ['libpaddle_master.so'],

'paddle.v2.fluid': ['core.so'],

'py_paddle':['*.py','_swig_paddle.so']

},

package_dir={

'': '${CMAKE_CURRENT_SOURCE_DIR}',

# The paddle.v2.fluid.proto will be generated while compiling.

# So that package points to other directory.

'paddle.v2.fluid.proto': '${PADDLE_BINARY_DIR}/paddle/framework',

'py_paddle': '${PADDLE_SOURCE_DIR}/paddle/py_paddle'

},

scripts=paddle_bins,

data_files=[(paddle_rt_lib_dir, paddle_rt_libs)]

)