i3blocks/.local/bin/i3blocks-gpu


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

#!/usr/bin/env python3

import os
import glob

ICON = "\U000f0379"  # 󰍹


def main():
    files = glob.glob("/sys/class/drm/card*/device/gpu_busy_percent")
    if not files:
        return

    with open(files[0]) as f:
        load = int(f.readline())

    print(f" {ICON} {load}% \n")
    if load > 90:
        print(f"#{os.environ.get("BASE16_COLOR_00_HEX")}")
        print(f"#{os.environ.get("BASE16_COLOR_08_HEX")}")
    elif load > 60:
        print(f"#{os.environ.get("BASE16_COLOR_00_HEX")}")
        print(f"#{os.environ.get("BASE16_COLOR_0A_HEX")}")
    elif load > 30:
        print(f"#{os.environ.get("BASE16_COLOR_00_HEX")}")
        print(f"#{os.environ.get("BASE16_COLOR_0C_HEX")}")


if __name__ == "__main__":
    main()