setup.py

#!/usr/bin/env python
"""Defines the setup instructions for domaintools"""
import glob
import os
import sys
from os import path

from setuptools import Extension, find_packages, setup
from setuptools.command.test import test as TestCommand


MYDIR = path.abspath(os.path.dirname(__file__))
JYTHON = "java" in sys.platform
PYPY = bool(getattr(sys, "pypy_version_info", False))
CYTHON = False
if not PYPY and not JYTHON:
 try:
 from Cython.Distutils import build_ext

 CYTHON = True
 except ImportError:
 pass

cmdclass = {}
ext_modules = []
if CYTHON:

 def list_modules(dirname):
 filenames = glob.glob(path.join(dirname, "*.py"))

 module_names = []
 for name in filenames:
 module, ext = path.splitext(path.basename(name))
 if module != "__init__":
 module_names.append(module)

 return module_names

 ext_modules = [
 Extension("domaintools." + ext, [path.join("domaintools", ext + ".py")])
 for ext in list_modules(path.join(MYDIR, "domaintools"))
 ]
 cmdclass["build_ext"] = build_ext


class PyTest(TestCommand):
 extra_kwargs = {"tests_require": ["pytest", "mock"]}

 def finalize_options(self):
 TestCommand.finalize_options(self)
 self.test_args = []
 self.test_suite = True

 def run_tests(self):
 import pytest

 sys.exit(pytest.main(self.test_args))


cmdclass["test"] = PyTest

setup(
 name="domaintools_api",
 cmdclass=cmdclass,
 ext_modules=ext_modules,
)