mirror of
https://github.com/optim-enterprises-bv/patroni.git
synced 2025-11-03 20:07:55 +00:00
137 lines
4.2 KiB
Python
137 lines
4.2 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
Setup file for patroni
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
import inspect
|
|
|
|
import setuptools
|
|
from setuptools.command.test import test as TestCommand
|
|
from setuptools import setup
|
|
|
|
if sys.version_info < (2, 7, 0):
|
|
sys.stderr.write('FATAL: patroni needs to be run with Python 2.7+\n')
|
|
sys.exit(1)
|
|
|
|
__location__ = os.path.join(os.getcwd(), os.path.dirname(inspect.getfile(inspect.currentframe())))
|
|
|
|
|
|
NAME = 'patroni'
|
|
MAIN_PACKAGE = 'patroni.py'
|
|
HELPERS = 'helpers'
|
|
VERSION = '0.1'
|
|
DESCRIPTION = 'A Template for PostgreSQL HA with etcd'
|
|
LICENSE = 'The MIT License'
|
|
|
|
COVERAGE_XML = True
|
|
COVERAGE_HTML = False
|
|
JUNIT_XML = True
|
|
|
|
# Add here all kinds of additional classifiers as defined under
|
|
# https://pypi.python.org/pypi?%3Aaction=list_classifiers
|
|
CLASSIFIERS = [
|
|
'Development Status :: 4 - Beta',
|
|
'Environment :: Console',
|
|
'Intended Audience :: Developers',
|
|
'Intended Audience :: System Administrators',
|
|
'License :: OSI Approved :: The MIT License',
|
|
'Operating System :: POSIX :: Linux',
|
|
'Programming Language :: Python',
|
|
'Programming Language :: Python :: 2.7',
|
|
'Programming Language :: Python :: 3.3',
|
|
'Programming Language :: Python :: 3.4',
|
|
'Programming Language :: Python :: Implementation :: CPython',
|
|
]
|
|
|
|
|
|
class PyTest(TestCommand):
|
|
|
|
user_options = [('cov=', None, 'Run coverage'), ('cov-xml=', None, 'Generate junit xml report'), ('cov-html=',
|
|
None, 'Generate junit html report'), ('junitxml=', None, 'Generate xml of test results')]
|
|
|
|
def initialize_options(self):
|
|
TestCommand.initialize_options(self)
|
|
self.cov_xml = False
|
|
self.cov_html = False
|
|
self.junitxml = None
|
|
|
|
def finalize_options(self):
|
|
TestCommand.finalize_options(self)
|
|
if self.cov_xml or self.cov_html:
|
|
self.cov = ['--cov', MAIN_PACKAGE, '--cov', HELPERS, '--cov-report', 'term-missing']
|
|
if self.cov_xml:
|
|
self.cov.extend(['--cov-report', 'xml'])
|
|
if self.cov_html:
|
|
self.cov.extend(['--cov-report', 'html'])
|
|
if self.junitxml is not None:
|
|
self.junitxml = ['--junitxml', self.junitxml]
|
|
|
|
def run_tests(self):
|
|
try:
|
|
import pytest
|
|
except:
|
|
raise RuntimeError('py.test is not installed, run: pip install pytest')
|
|
params = {'args': self.test_args}
|
|
if self.cov:
|
|
params['args'] += self.cov
|
|
params['plugins'] = ['cov']
|
|
if self.junitxml:
|
|
params['args'] += self.junitxml
|
|
params['args'] += ['--doctest-modules', HELPERS, '-s']
|
|
errno = pytest.main(**params)
|
|
sys.exit(errno)
|
|
|
|
|
|
def get_install_requirements(path):
|
|
content = open(os.path.join(__location__, path)).read()
|
|
return [req for req in content.split('\n') if req != '']
|
|
|
|
|
|
def read(fname):
|
|
return open(os.path.join(__location__, fname)).read()
|
|
|
|
|
|
def setup_package():
|
|
# Assemble additional setup commands
|
|
cmdclass = {}
|
|
cmdclass['test'] = PyTest
|
|
|
|
# Some helper variables
|
|
version = os.getenv('GO_PIPELINE_LABEL', VERSION)
|
|
|
|
requirements = 'requirements-py2.txt' if sys.version_info[0] == 2 else 'requirements-py3.txt'
|
|
install_reqs = get_install_requirements(requirements)
|
|
|
|
command_options = {'test': {'test_suite': ('setup.py', 'tests')}}
|
|
if JUNIT_XML:
|
|
command_options['test']['junitxml'] = 'setup.py', 'junit.xml'
|
|
if COVERAGE_XML:
|
|
command_options['test']['cov_xml'] = 'setup.py', True
|
|
if COVERAGE_HTML:
|
|
command_options['test']['cov_html'] = 'setup.py', True
|
|
|
|
setup(
|
|
name=NAME,
|
|
version=version,
|
|
description=DESCRIPTION,
|
|
license=LICENSE,
|
|
keywords='etcd governor patroni postgresql postgres ha zookeeper',
|
|
long_description=read('README.md'),
|
|
classifiers=CLASSIFIERS,
|
|
test_suite='tests',
|
|
packages=setuptools.find_packages(exclude=['tests', 'tests.*']),
|
|
package_data={MAIN_PACKAGE: ["*.json"]},
|
|
install_requires=install_reqs,
|
|
setup_requires=['six', 'flake8'],
|
|
cmdclass=cmdclass,
|
|
tests_require=['pytest-cov', 'pytest'],
|
|
command_options=command_options,
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
setup_package()
|