File: parse_logs.py

package info (click to toggle)
pint-xarray 0.6.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,328 kB
  • sloc: python: 5,047; makefile: 4
file content (102 lines) | stat: -rw-r--r-- 2,446 bytes parent folder | download | duplicates (2)
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
# type: ignore
import argparse
import functools
import json
import pathlib
import textwrap
from dataclasses import dataclass

from pytest import CollectReport, TestReport


@dataclass
class SessionStart:
    pytest_version: str
    outcome: str = "status"

    @classmethod
    def _from_json(cls, json):
        json_ = json.copy()
        json_.pop("$report_type")
        return cls(**json_)


@dataclass
class SessionFinish:
    exitstatus: str
    outcome: str = "status"

    @classmethod
    def _from_json(cls, json):
        json_ = json.copy()
        json_.pop("$report_type")
        return cls(**json_)


def parse_record(record):
    report_types = {
        "TestReport": TestReport,
        "CollectReport": CollectReport,
        "SessionStart": SessionStart,
        "SessionFinish": SessionFinish,
    }
    cls = report_types.get(record["$report_type"])
    if cls is None:
        raise ValueError(f"unknown report type: {record['$report_type']}")

    return cls._from_json(record)


@functools.singledispatch
def format_summary(report):
    return f"{report.nodeid}: {report}"


@format_summary.register
def _(report: TestReport):
    message = report.longrepr.chain[0][1].message
    return f"{report.nodeid}: {message}"


@format_summary.register
def _(report: CollectReport):
    message = report.longrepr.split("\n")[-1].removeprefix("E").lstrip()
    return f"{report.nodeid}: {message}"


def format_report(reports, py_version):
    newline = "\n"
    summaries = newline.join(format_summary(r) for r in reports)
    message = textwrap.dedent(
        """\
        <details><summary>Python {py_version} Test Summary</summary>

        ```
        {summaries}
        ```

        </details>
        """
    ).format(summaries=summaries, py_version=py_version)
    return message


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("filepath", type=pathlib.Path)
    args = parser.parse_args()

    py_version = args.filepath.stem.split("-")[1]

    print("Parsing logs ...")

    lines = args.filepath.read_text().splitlines()
    reports = [parse_record(json.loads(line)) for line in lines]

    failed = [report for report in reports if report.outcome == "failed"]

    message = format_report(failed, py_version=py_version)

    output_file = pathlib.Path("pytest-logs.txt")
    print(f"Writing output file to: {output_file.absolute()}")
    output_file.write_text(message)