mirror of https://github.com/kubeflow/examples.git
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
from __future__ import print_function
|
|
import subprocess
|
|
from distutils.command.build import build as distutils_build #pylint: disable=no-name-in-module
|
|
from setuptools import setup, find_packages, Command as SetupToolsCommand
|
|
|
|
|
|
with open('requirements.txt', 'r') as f:
|
|
install_requires = f.readlines()
|
|
|
|
with open('README.md') as f:
|
|
README = f.read()
|
|
|
|
VERSION = '0.1.0'
|
|
CUSTOM_COMMANDS = [
|
|
['python', '-m', 'spacy', 'download', 'en']
|
|
]
|
|
|
|
|
|
class Build(distutils_build):
|
|
sub_commands = distutils_build.sub_commands + [('CustomCommands', None)]
|
|
|
|
|
|
class CustomCommands(SetupToolsCommand):
|
|
def initialize_options(self):
|
|
pass
|
|
|
|
def finalize_options(self):
|
|
pass
|
|
|
|
@staticmethod
|
|
def run_custom_command(command_list):
|
|
print('Running command: %s' % command_list)
|
|
p = subprocess.Popen(command_list, stdin=subprocess.PIPE,
|
|
stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
|
stdout_data, _ = p.communicate()
|
|
print('Command output: %s' % stdout_data)
|
|
if p.returncode != 0:
|
|
raise RuntimeError('Command %s failed: exit code: %s' % (command_list, p.returncode))
|
|
|
|
def run(self):
|
|
for command in CUSTOM_COMMANDS:
|
|
self.run_custom_command(command)
|
|
|
|
|
|
setup(name='kubeflow-code-search',
|
|
description='Kubeflow Code Search Demo',
|
|
long_description=README,
|
|
long_description_content_type='text/markdown',
|
|
url='https://www.github.com/kubeflow/examples',
|
|
author='Sanyam Kapoor',
|
|
author_email='sanyamkapoor@google.com',
|
|
version=VERSION,
|
|
license='MIT',
|
|
packages=find_packages(),
|
|
install_requires=install_requires,
|
|
extras_require={},
|
|
cmdclass={
|
|
'build': Build,
|
|
'CustomCommands': CustomCommands,
|
|
})
|