lizard/lizard_ext/csvoutput.py at 2640089e8bc31afaf0acf96ab3e918357b2a71f2 · terryyin/lizard · 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
"""
This module extends the default output formatting to include CSV.

The output is intended to be the same in structure, with additional
tokens (to reduce the need to post-process), and a reduced verbosity
due to the nature of CSV outputs. The differences are:

 * No summary of the included files, only the output
   respective of each function.
 * No additional output for functions that break any CCN
   thresholds.
 * Each line has four additional, individual tokens:
     * File name
     * Function Name
     * Function line start
     * Function line end
"""


def csv_output(result, options):
    result = result.result

    extension_variables = []
    extension_captions = []
    for extension in options.extensions:
        if extension.__class__.__name__ == 'LizardExtension' and \
            hasattr(extension, 'FUNCTION_INFO'):
            for extension_variable in extension.FUNCTION_INFO:
                extension_variables.append(extension_variable)
                var_extension = extension.FUNCTION_INFO[extension_variable]
                extension_caption = var_extension['caption'] \
                    if 'caption' in var_extension else 'No_caption'
                extension_captions.append(extension_caption)

    if options.verbose:
        extension_caption = ""
        for caption in extension_captions:
            extension_caption = f"{extension_caption},{caption}"
        print(f"NLOC,CCN,token,PARAM,length,location,file,function," +
              f"long_name,start,end{extension_caption}")

    for source_file in result:
        if source_file:
            for source_function in source_file.function_list:
                if source_function:
                    extension_string = ''
                    for variable in extension_variables:
                        extension_string = f"{extension_string},{source_function.__getattribute__(variable)}"
                    print(f"{source_function.nloc},{source_function.cyclomatic_complexity},"
                          f"{source_function.token_count},{len(source_function.parameters)},"
                          f'{source_function.length},"{source_function.name.replace(chr(34), chr(39))}@'
                          f'{source_function.start_line}-{source_function.end_line}@{source_file.filename}",'
                          f'"{source_file.filename}","{source_function.name.replace(chr(34), chr(39))}",'
                          f'"{source_function.long_name.replace(chr(34), chr(39))}",'
                          f"{source_function.start_line},{source_function.end_line}{extension_string}")