diff --git a/CONTRIBUTING.rst b/CONTRIBUTING.rst index f5bbb73..ca71a77 100644 --- a/CONTRIBUTING.rst +++ b/CONTRIBUTING.rst @@ -19,3 +19,4 @@ Contributors * Michael Svanström (@squeezeday) * Tomislav Kopić (@Tkopic001) * Maciej Sokolowski (@matemaciek) +* Sangho Kim (@rlaace423) diff --git a/README.rst b/README.rst index 57cb5cb..c557af4 100644 --- a/README.rst +++ b/README.rst @@ -78,6 +78,7 @@ sprite_animation.py Using sprite maps for animation effects starfield.py 3D starfield simulation sys_histogram.py Display system information including a rolling histogram sys_info.py Display basic system information +sys_info_extended.py Display detailed system information in graph format terminal.py Simple println capabilities tv_snow.py Example image-blitting tweet_scroll.py Using Twitter's Streaming API to display scrolling notifications diff --git a/examples/fonts/DejaVuSansMono.ttf b/examples/fonts/DejaVuSansMono.ttf new file mode 100644 index 0000000..041cffc Binary files /dev/null and b/examples/fonts/DejaVuSansMono.ttf differ diff --git a/examples/sys_info_extended.py b/examples/sys_info_extended.py new file mode 100644 index 0000000..b4c31e4 --- /dev/null +++ b/examples/sys_info_extended.py @@ -0,0 +1,166 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +# Copyright (c) 2023 Richard Hull and contributors +# See LICENSE.rst for details. + +""" +Display detailed system information in graph format. +It provides a display of CPU, memory, disk utilization, temperature, IP address and system Uptime. + +Needs psutil (+ dependencies) installed:: + + $ sudo apt-get install python-dev + $ sudo -H pip install psutil +""" + +import time +from pathlib import Path +from datetime import datetime +from demo_opts import get_device +from luma.core.render import canvas +from PIL import ImageFont +import psutil +import subprocess as sp +import socket +from collections import OrderedDict + + +def get_temp(): + temp = float(sp.getoutput("vcgencmd measure_temp").split("=")[1].split("'")[0]) + return temp + + +def get_cpu(): + return psutil.cpu_percent() + + +def get_mem(): + return psutil.virtual_memory().percent + + +def get_disk_usage(): + usage = psutil.disk_usage("/") + return usage.used / usage.total * 100 + + +def get_uptime(): + uptime = ("%s" % (datetime.now() - datetime.fromtimestamp(psutil.boot_time()))).split(".")[0] + return "UpTime: %s" % (uptime) + + +def find_single_ipv4_address(addrs): + for addr in addrs: + if addr.family == socket.AddressFamily.AF_INET: # IPv4 + return addr.address + + +def get_ipv4_address(interface_name=None): + if_addrs = psutil.net_if_addrs() + + if isinstance(interface_name, str) and interface_name in if_addrs: + addrs = if_addrs.get(interface_name) + address = find_single_ipv4_address(addrs) + return address if isinstance(address, str) else "" + else: + if_stats = psutil.net_if_stats() + # remove loopback + if_stats_filtered = {key: if_stats[key] for key, stat in if_stats.items() if "loopback" not in stat.flags} + # sort interfaces by + # 1. Up/Down + # 2. Duplex mode (full: 2, half: 1, unknown: 0) + if_names_sorted = [stat[0] for stat in sorted(if_stats_filtered.items(), key=lambda x: (x[1].isup, x[1].duplex), reverse=True)] + if_addrs_sorted = OrderedDict((key, if_addrs[key]) for key in if_names_sorted if key in if_addrs) + + for _, addrs in if_addrs_sorted.items(): + address = find_single_ipv4_address(addrs) + if isinstance(address, str): + return address + + return "" + + +def get_ip(network_interface_name): + return "IP: %s" % (get_ipv4_address(network_interface_name)) + + +def format_percent(percent): + return "%5.1f" % (percent) + + +def draw_text(draw, margin_x, line_num, text): + draw.text((margin_x, margin_y_line[line_num]), text, font=font_default, fill="white") + + +def draw_bar(draw, line_num, percent): + top_left_y = margin_y_line[line_num] + bar_margin_top + draw.rectangle((margin_x_bar, top_left_y, margin_x_bar + bar_width, top_left_y + bar_height), outline="white") + draw.rectangle((margin_x_bar, top_left_y, margin_x_bar + bar_width * percent / 100, top_left_y + bar_height), fill="white") + + +def draw_bar_full(draw, line_num): + top_left_y = margin_y_line[line_num] + bar_margin_top + draw.rectangle((margin_x_bar, top_left_y, margin_x_bar + bar_width_full, top_left_y + bar_height), fill="white") + draw.text((65, top_left_y - 2), "100 %", font=font_full, fill="black") + + +def stats(device): + with canvas(device) as draw: + temp = get_temp() + draw_text(draw, 0, 0, "Temp") + draw_text(draw, margin_x_figure, 0, "%s'C" % (format_percent(temp))) + + cpu = get_cpu() + draw_text(draw, 0, 1, "CPU") + if cpu < 100: + draw_text(draw, margin_x_figure, 1, "%s %%" % (format_percent(cpu))) + draw_bar(draw, 1, cpu) + else: + draw_bar_full(draw, 1) + + mem = get_mem() + draw_text(draw, 0, 2, "Mem") + if mem < 100: + draw_text(draw, margin_x_figure, 2, "%s %%" % (format_percent(mem))) + draw_bar(draw, 2, mem) + else: + draw_bar_full(draw, 2) + + disk = get_disk_usage() + draw_text(draw, 0, 3, "Disk") + if disk < 100: + draw_text(draw, margin_x_figure, 3, "%s %%" % (format_percent(disk))) + draw_bar(draw, 3, disk) + else: + draw_bar_full(draw, 3) + + if datetime.now().second % (toggle_interval_seconds * 2) < toggle_interval_seconds: + draw_text(draw, 0, 4, get_uptime()) + else: + draw_text(draw, 0, 4, get_ip(network_interface_name)) + + +font_size = 12 +font_size_full = 10 +margin_y_line = [0, 13, 25, 38, 51] +margin_x_figure = 78 +margin_x_bar = 31 +bar_width = 52 +bar_width_full = 95 +bar_height = 8 +bar_margin_top = 3 +toggle_interval_seconds = 4 + + +# None : find suitable IPv4 address among all network interfaces +# or specify the desired interface name as string. +network_interface_name = None + + +device = get_device() +font_default = ImageFont.truetype(str(Path(__file__).resolve().parent.joinpath("fonts", "DejaVuSansMono.ttf")), font_size) +font_full = ImageFont.truetype(str(Path(__file__).resolve().parent.joinpath("fonts", "DejaVuSansMono.ttf")), font_size_full) + + +while True: + stats(device) + time.sleep(0.5)