contatester/setup.py at master · CNRGH/contatester · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
import shutil
import subprocess
from distutils.command.clean import clean
from glob import glob
from os import path, walk, remove

from typing import List

from setuptools import setup, Command
from setuptools.config import read_configuration


def get_files(directory: str, recursive: bool = False) -> List[str]:
    return [item for item in glob(path.join(directory, '*'), recursive=recursive) if path.isfile(item)]


class ExtendedClean(clean):
    """
    This class extend clean command in order to remove all __pycache__ directories.
    Indeed these directories contains pre-computed file with module path write in
    unix or windows style. Depends of latest os which build the file.
    In multi-OS environment this could lead to strange behavior.
    """

    @staticmethod
    def _find_all_directories(dir_name: str, path_to_scan: str) -> List[str]:
        result = []
        for root, dirs, files in walk(path_to_scan):
            if dir_name in dirs:
                result.append(path.join(root, dir_name))
        return result

    def run(self) -> None:
        clean.run(self)
        c = clean(self.distribution)
        c.finalize_options()
        c.run()
        # sources directories
        for directory in self.distribution.package_dir.values():
            for file_path in self._find_all_directories('__pycache__', directory):
                shutil.rmtree(file_path)
            for file_path in self._find_all_directories(self.distribution.metadata.name+'.egg-info', directory):
                shutil.rmtree(file_path)
        # tests directory
        if path.exists('tests'):
            for file_path in self._find_all_directories('__pycache__', 'tests'):
                shutil.rmtree(file_path)
        if path.exists('.coverage'):
            remove('.coverage')
        if path.exists('htmlcov'):
            shutil.rmtree('htmlcov')
        if path.exists('.eggs'):
            shutil.rmtree('.eggs')
        if path.exists('.pytest_cache'):
            shutil.rmtree('.pytest_cache')


class Coverage(Command):
    description = 'generate report'

    user_options = [
        ('source=', 's', 'source directory')
    ]

    _sources = ''

    def initialize_options(self):
        self._sources = None

    def finalize_options(self):
        if self._sources is None:
            self._sources = ['--source={dir}'.format(dir=directory) for directory in
                             self.distribution.package_dir.values()]

    def run(self) -> None:
        subprocess.call(['coverage', 'run'] + self._sources + ['setup.py', 'test'])
        subprocess.call(['coverage', 'report'])
        subprocess.call(['coverage', 'html'])


if __name__ == '__main__':
    conf = read_configuration('setup.cfg')
    setup(
        name=conf['metadata']['name'],
        version=conf['metadata']['version'],
        description=conf['metadata']['description'],
        author=conf['metadata']['author'],
        license=conf['metadata']['license'],
        classifiers=conf['metadata']['classifiers'],
        keywords=conf['metadata']['keywords'],
        cmdclass={'clean': ExtendedClean, 'coverage': Coverage},
        packages=['fr.cea.cnrgh.lbi.contatester'],
        package_dir={'': 'src'},
        include_package_data=True,

        data_files=[('share/{}/'.format(conf['metadata']['name']), get_files('data'))],
        scripts=get_files('scripts'),
        install_requires=['wheel >= 0.31.0'],
        setup_requires=['pytest-runner', 'setuptools >= 40.0.0 '],
        tests_require=['pytest  >= 3.4.0',
                       'pytest-dependency >= 0.3.0',
                       'pytest-mock >= 1.10.0',
                       'coverage >= 4.5.1'],
        entry_points={
            'console_scripts': [
                'contatester = fr.cea.cnrgh.lbi.contatester.__main__:main'
            ]
        },
        extras_require={}
    )