scons/setup.py at master · softagram/scons · GitHub

46 lines (34 loc) · 1.18 KB

#!python3

import fnmatch

from setuptools import setup

from setuptools.command.build_py import build_py as build_py_orig

import codecs

import os.path

def read(rel_path):

here = os.path.abspath(os.path.dirname(__file__))

with codecs.open(os.path.join(here, rel_path), 'r') as fp:

return fp.read()

def get_version(rel_path):

for line in read(rel_path).splitlines():

if line.startswith('__version__'):

delim = '"' if '"' in line else "'"

return line.split(delim)[1]

else:

raise RuntimeError("Unable to find version string.")

exclude = ['*Tests']

class build_py(build_py_orig):

def find_package_modules(self, package, package_dir):

"""

Custom module to find package modules.

It will strip out any modules which match the glob patters in exclude above

"""

modules = super().find_package_modules(package, package_dir)

return [(pkg, mod, file, ) for (pkg, mod, file, ) in modules

if not any(fnmatch.fnmatchcase(mod, pat=pattern)

for pattern in exclude)]

setup(

cmdclass={

'build_py': build_py,

version=get_version('SCons/__init__.py'),

)

Expand file tree