scikit-multilearn/utils.py at master · scikit-multilearn-ng/scikit-multilearn · 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
#!/usr/bin/env python

import os
import subprocess
import sys
from utils.files import files


class Config:
    root = "."
    meka = "meka"
    skmultilearn = "skmultilearn"
    tests = "tests"
    utils = "utils"


# #######################################################


def test():
    return subprocess.call("py.test", shell=True)


def clean():
    files.removeFilesRecursive(Config.meka, (lambda f: f.endswith(".pyc")))
    files.removeFilesRecursive(Config.skmultilearn, (lambda f: f.endswith(".pyc")))
    files.removeFilesRecursive(Config.tests, (lambda f: f.endswith(".pyc")))
    files.removeFilesRecursive(Config.utils, (lambda f: f.endswith(".pyc")))


def package():
    sourceDirs = [Config.meka, Config.skmultilearn, Config.tests, Config.utils]
    for sourceDir in sourceDirs:
        for root, dirs, files in os.walk(sourceDir):
            for dir in dirs:
                initFile = os.path.join(root, dir, "__init__.py")
                if not os.path.isfile(initFile):
                    print("creating : %s" % initFile)
                    open(initFile, "a").close()


def lint(full=False):
    from pylint import epylint

    sources = [Config.root, Config.meka, Config.skmultilearn]
    if full:
        fullReport = "y"
    else:
        fullReport = "n"

    config = '--rcfile ./utils/pylint.config --msg-template="{C}:{msg_id}:{line:3d},{column:2d}:{msg}({symbol})" -r %s %s'
    for dir in sources:
        print("lint %s" % dir)
        epylint.py_run(config % (fullReport, dir), script="pylint")


def lint_full():
    lint(True)


def default():
    clean()
    package()
    lint()
    # test()


def install():
    try:
        import build.install as installer
    except ImportError:
        subprocess.call("python ./build/get-pip.py", shell=True)
        import utils.install as installer

    installer.installRequirements("./build/requirements.txt")


########################################################


def step(msg):
    span = "=" * ((80 - len(msg)) / 2)
    print(" ".join([span, msg, span]))


if __name__ == "__main__":
    os.environ["PYTHONPATH"] = os.path.dirname(os.path.realpath(__file__))

    if len(sys.argv) > 1:
        for task in sys.argv[1:]:
            if task in locals():
                step(task)
                locals()[task]()
            else:
                print('Error: task "%s" not found' % task)
                sys.exit(1)
    else:
        default()