setup.py
import os
from glob import glob
from setuptools import setup, find_packages
import subprocess
import imp
DISTNAME = 'dynamicgem'
MAINTAINER = 'Palash Goyal, Sujit Rokka Chhetri'
MAINTAINER_EMAIL = 'palashgo@usc.edu, schhetri@uci.edu'
DESCRIPTION = 'dynamicGEM: A Python module for Dynamic Graph Embedding Methods'
LONG_DESCRIPTION = open('README.md').read()
URL = 'https://github.com/palash1992/DynamicGEM'
DOWNLOAD_URL = 'https://github.com/palash1992/DynamicGEM'
KEYWORDS = ['dynamic graph embedding', 'graph embedding', 'network analysis',
'network embedding', 'data mining', 'machine learning']
LICENSE = 'BSD'
VERSION = '1.0.0'
ISRELEASED = True
INSTALL_REQUIRES = (
'Cython>=0.29',
'decorator>=4.3.0',
'dill>=0.2.8.2',
'h5py>=2.8.0',
'joblib>=0.12.5',
'Keras>=2.2.4',
'Keras-Applications>=1.0.6',
'Keras-Preprocessing>=1.0.5',
'matplotlib>=3.0.1',
'networkx>=1.11',
'numpy>=1.15.3',
'pandas>=0.23.4',
'protobuf>=3.6.1',
'scikit-learn>=0.20.0',
'scipy>=1.1.0',
'seaborn>=0.9.0',
'six>=1.11.0',
'sklearn>=0.0',
'Theano>=1.0.3'
)
def get_package_data(topdir, excluded=set()):
retval = []
for dirname, subdirs, files in os.walk(os.path.join(DISTNAME, topdir)):
for x in excluded:
if x in subdirs:
subdirs.remove(x)
retval.append(os.path.join(dirname[len(DISTNAME)+1:], '*.*'))
return retval
def get_data_files(dest, source):
retval = []
for dirname, subdirs, files in os.walk(source):
retval.append(
(os.path.join(dest, dirname[len(source)+1:]), glob(os.pathjoin(dirname, '*.*')))
)
return retval
# Return the git revision as a string
def git_version():
"""Return the git revision as a string.
Copied from numpy setup.py
"""
def _minimal_ext_cmd(cmd):
# construct minimal environment
env = {}
for k in ['SYSTEMROOT', 'PATH']:
v = os.environ.get(k)
if v is not None:
env[k] = v
# LANGUAGE is used on win32
env['LANGUAGE'] = 'C'
env['LANG'] = 'C'
env['LC_ALL'] = 'C'
out = subprocess.Popen(cmd, stdout = subprocess.PIPE, env=env).communicate()[0]
return out
try:
out = _minimal_ext_cmd(['git', 'rev-parse', 'HEAD'])
GIT_REVISION = out.strip().decode('ascii')
except OSError:
GIT_REVISION = "Unknown"
return GIT_REVISION
def write_version_py(filename='dynamicgem/version.py'):
# Copied from numpy setup.py
cnt = """
# THIS FILE IS GENERATED FROM DynamicGEM SETUP.PY
short_version = '%(version)s'
version = '%(version)s'
full_version = '%(full_version)s'
git_revision = '%(git_revision)s'
release = %(isrelease)s
if not release:
version = full_version
short_version += ".dev"
"""
FULLVERSION = VERSION
if os.path.exists('.git'):
GIT_REVISION = git_version()
elif os.path.exists('dynamicgem/version.py'):
# must be a source distribution, use existing version file
version = imp.load_source('DynamicGEM.version', 'dynamicgem/version.py')
GIT_REVISION = version.git_revision
else:
GIT_REVISION = 'Unknown'
if not ISRELEASED:
FULLVERSION += '.dev0+' + GIT_REVISION[:7]
a = open(filename, 'w')
try:
a.write(cnt % {'version': VERSION,
'full_version': FULLVERSION,
'git_revision': GIT_REVISION,
'isrelease': str(ISRELEASED)})
finally:
a.close()
def setup_package():
write_version_py()
setup(
name=DISTNAME,
version=VERSION,
author=MAINTAINER,
author_email=MAINTAINER_EMAIL,
maintainer=MAINTAINER,
maintainer_email=MAINTAINER_EMAIL,
description=DESCRIPTION,
url=URL,
download_url=DOWNLOAD_URL,
keywords=KEYWORDS,
install_requires=INSTALL_REQUIRES,
packages=find_packages(),
package_dir={DISTNAME: 'dynamicgem'},
package_data={DISTNAME: get_package_data('datasets')},
license=LICENSE,
long_description=LONG_DESCRIPTION,
classifiers=['Intended Audience :: Science/Research',
'Intended Audience :: Developers',
'License :: OSI Approved',
'Programming Language :: Python',
'Topic :: Software Development',
'Topic :: Scientific/Engineering',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
'Topic :: Scientific/Engineering :: Graph Analysis',
'Programming Language :: Python :: 3',],
)
if __name__ == "__main__":
setup_package()