summaryrefslogtreecommitdiff
path: root/scripts/maint/practracker/util.py
blob: 695668f5618cc3f1352e9ac8c84b31a00aef54c3 (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
import os

# We don't want to run metrics for unittests, automatically-generated C files,
# external libraries or git leftovers.
EXCLUDE_SOURCE_DIRS = {"src/test/", "src/trunnel/", "src/rust/",
                       "src/ext/", ".git/"}

EXCLUDE_FILES = {"orconfig.h"}

def _norm(p):
    return os.path.normcase(os.path.normpath(p))

def get_tor_c_files(tor_topdir):
    """
    Return a list with the .c and .h filenames we want to get metrics of.
    """
    files_list = []
    exclude_dirs = { _norm(os.path.join(tor_topdir, p)) for p in EXCLUDE_SOURCE_DIRS }


    for root, directories, filenames in os.walk(tor_topdir):
        # Remove all the directories that are excluded.
        directories[:] = [ d for d in directories
                           if _norm(os.path.join(root,d)) not in exclude_dirs ]
        directories.sort()
        filenames.sort()
        for filename in filenames:
            # We only care about .c and .h files
            if not (filename.endswith(".c") or filename.endswith(".h")):
                continue
            if filename in EXCLUDE_FILES:
                continue

            full_path = os.path.join(root,filename)

            files_list.append(full_path)

    return files_list