canup/commit_info.py at master · UBCFormulaElectric/canup · 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
"""
commit_info.py
-----------------
Fetches the commit information for the commits for each ECU on the CAN bus.

"""

import argparse

import can
from github import Github, Commit


def get_commit_info(sha: str) -> Commit:
    """
    Fetches the commit information for a given commit SHA.
    :param sha: The given commit SHA.
    :return: Commit Object from the GitHub API.
    """
    g = Github()
    repo = g.get_repo("UBCFormulaElectric/Consolidated-Firmware")
    c = repo.get_commit(sha=sha)
    return c


bus_mapping: dict[int, str] = {
    140: "BMS",
    506: "CRIT",
    819: "dimos",
    302: "FSM",
    419: "RSM",
    204: "VC",
    799: "RSM_boot",
    599: "FSM_boot",
    899: "CRIT_boot",
}

bus_found: dict[str, bool] = {
    "BMS": False,
    "CRIT": False,
    "dimos": False,
    "FSM": False,
    "RSM": False,
    "VC": False,
    "VC_boot": False,
    "BMS_boot": False,
    "RSM_boot": False,
    "FSM_boot": False,
    "CRIT_boot": False,
}

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--bus", type=str, default="pcan", help="python-can bus type")
    parser.add_argument(
        "--channel", type=str, default="PCAN_USBBUS1", help="python-can channel"
    )
    parser.add_argument("--bit_rate", type=int, default=500000, help="CAN bus bit rate")
    args = parser.parse_args()

    with can.interface.Bus(
        interface=args.bus, channel=args.channel, bitrate=args.bit_rate
    ) as bus:
        try:
            while not all(bus_found.values()):
                rx_message = bus.recv()
                if (
                    rx_message.arbitration_id in bus_mapping.keys()
                    and not bus_found[bus_mapping[rx_message.arbitration_id]]
                ):
                    message_board = bus_mapping[rx_message.arbitration_id]
                    commit_hex = "".join(
                        "{:02x}".format(x) for x in reversed(rx_message.data[0:4])
                    ).capitalize()[1:]
                    clean = (
                        "{:02x}".format(rx_message.data[-1]) == "01"
                        and "Clean"
                        or "Dirty"
                    )
                    bus_found[message_board] = True

                    commit = get_commit_info(commit_hex)

                    print()
                    print(
                        f'{message_board}: "{commit.commit.message}" by {commit.author.name}'
                    )
                    print(f"Date: {commit.commit.author.date}")
                    print(f"SHA: {commit.sha} ({clean})")
        except KeyboardInterrupt:
            print("Exiting...")
        bus.shutdown()