aboutsummaryrefslogtreecommitdiffstats
path: root/.github/scripts/get_build_diff.py
blob: 30ba3f46c409c346418c6bc6c0657deaf03ec8ff (plain) (blame)
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
#!/usr/bin/env python3

from decimal import Decimal, ROUND_HALF_UP
import get_current_build_size
import get_main_build_size
import humanize
import os


# Форматирование числа
def format_number(num):
    return humanize.intcomma(num).replace(",", " ")


def bytes_to_human_iec(num):
    return humanize.naturalsize(num, binary=True)


def main():

    yellow_treshold = int(os.environ.get("yellow_treshold"))
    red_treshold = int(os.environ.get("red_treshold"))

    branch = os.environ.get("branch_to_compare")
    current_pr_commit_sha = os.environ.get("commit_git_sha")

    current_sizes_result = get_current_build_size.get_build_size()
    main_sizes_result = get_main_build_size.get_build_size()

    if main_sizes_result and current_sizes_result:
        main_github_sha = main_sizes_result["github_sha"]
        main_size_bytes = int(main_sizes_result["size_bytes"])
        main_size_stripped_bytes = int(main_sizes_result["size_stripped_bytes"])

        current_size_bytes = int(current_sizes_result["size_bytes"])
        current_size_stripped_bytes = int(current_sizes_result["size_stripped_bytes"])

        bytes_diff = current_size_bytes - main_size_bytes
        diff_perc = Decimal(bytes_diff * 100 / main_size_bytes).quantize(
            Decimal(".001"), rounding=ROUND_HALF_UP
        )

        human_readable_size = bytes_to_human_iec(current_size_bytes)
        human_readable_size_diff = bytes_to_human_iec(bytes_diff)

        if bytes_diff >= 0:
            sign = "+"
            if bytes_diff >= red_treshold:
                color = "red"
            elif bytes_diff >= yellow_treshold:
                color = "yellow"
            else:
                color = "green"
        else:
            sign = ""
            color = "green"

        comment = (
            f"merge: {current_pr_commit_sha} ydbd size {human_readable_size} **{sign}{human_readable_size_diff} {sign}{diff_perc}%%** vs build {branch}: {main_github_sha}\n\n"
            "<details><summary>Build size details</summary><p>\n\n"
            f"{branch}: {main_github_sha} ydbd build size:\n"
            f"  - binary size **{format_number(main_size_bytes)}** Bytes \n"
            f"  - stripped binary size **{format_number(main_size_stripped_bytes)}** Bytes\n\n"
            f"merge: {current_pr_commit_sha} ydbd build size:\n"
            f"  - binary size **{format_number(current_size_bytes)}** Bytes \n"
            f"  - stripped binary size **{format_number(current_size_stripped_bytes)}** Bytes\n\n"
            "[ydbd size dashboard](https://datalens.yandex/cu6hzmpaki700)\n\n"
            "</p></details>"
        )
        print(f"{color};;;{comment}")
    else:
        print(f'Error: Cant get build data: {branch}_sizes_result = {main_sizes_result}, current_sizes_result = {current_sizes_result}')   
    

if __name__ == "__main__":
    main()