aboutsummaryrefslogtreecommitdiffstats
path: root/longbow/src/python/site-packages/longbow/LongBow.py
blob: d1f0e77a3d3b684405240b4f74d656c027792d74 (plain)
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
#! /usr/bin/env python
# Copyright (c) 2017 Cisco and/or its affiliates.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

#
import sys

ansiRed = "\x1b[31m";
ansiGreen = "\x1b[32m";
ansiYellow = "\x1b[33m";
ansiMagenta = "\x1b[35m";
ansiReset = "\x1b[0m";


def buildRed(string):
    if sys.stdout.isatty():
        return ansiRed + string + ansiReset
    else:
        return string


def buildGreen(string):
    if sys.stdout.isatty():
        return ansiGreen + string + ansiReset
    else:
        return string


def buildYellow(string):
    if sys.stdout.isatty():
        return ansiYellow + string + ansiReset
    else:
        return string


def score(distribution, score):
    result = "red"

    if (score > distribution[0]):
        result = "green"
    elif (score > distribution[1]):
        result = "yellow"

    return result


def scoreBuilder(distribution, score, string):
    '''
    scores is a list of 2 decreasing values.
    The first is the minimum score for green, the second is the minimum score for yellow.
    The rest art red
    '''
    if (score > distribution[0]):
        return buildGreen(string)
    elif (score > distribution[1]):
        return buildYellow(string)
    else:
        return buildRed(string)


def scorePrinter(distribution, score, string):
    print scoreBuilder(distribution, score, string)


def countLines(fileName):
    i = 0
    with open(fileName) as f:
        for i, l in enumerate(f):
            pass
    return i + 1


def CFileNameToFunctionPrefix(fileName):
    '''
    Given the name of a C source file or header file,
    return the canonical name prefix for functions within that file.
    For example, the input name "parc_Buffer.c" results in "parcBuffer_"
    '''
    fileName = os.path.basename(fileName);
    fileNameSpace = os.path.splitext(fileName)[0]
    parts = fileNameSpace.partition("_")
    result = None
    if len(parts) == 3:
        result = parts[0] + parts[2] + "_"
    return result