+++ /dev/null
-#!/usr/bin/env python3
-# _*_ coding: utf-8 _*_
-
-"""
-A psutil-based command to display customizable system usage info in a single line, intended for Tint2 executors
-
-Author: Piotr Miller
-e-mail: nwg.piotr@gmail.com
-Website: http://nwg.pl
-Project: https://github.com/nwg-piotr/psuinfo
-License: GPL3
-
-Inspired by https://github.com/tknomanzr/scripts/blob/master/tint2/executors/cpu.py by William Bradley (@tknomanzr)
-"""
-
-import sys
-import psutil
-import time
-import os
-
-
-def main():
- fahrenheit = False
- names = False
- testing = False
- time_start = None
- components = "gStfM"
- separator = " "
- home = os.getenv("HOME")
- draw_icons = False
-
- pcpu, avg, speed, freqs, temp, fans, b_time, memory, swap, disks_usage, which, ul, dl, xfer_start, xfer_finish, \
- path_to_icon, c_name= None, None, None, None, None, None, None, None, None, None, None, None, None, None, \
- None, None, None
-
- for i in range(1, len(sys.argv)):
- if sys.argv[i] == "-h" or sys.argv[i] == "--help":
- print_help()
- exit(0)
-
- if sys.argv[i] == "-F":
- fahrenheit = True
-
- if sys.argv[i] == "-N":
- names = True
-
- if sys.argv[i] == "-T":
- testing = True
-
- if sys.argv[i].startswith("-C"):
- components = sys.argv[i][2::]
-
- if sys.argv[i].startswith("-S"):
- try:
- # if number given
- spacing = int(sys.argv[i][2::])
- separator = " " * spacing
- except ValueError:
- # string given
- separator = sys.argv[i][2::]
-
- if sys.argv[i].startswith("-W"):
- try:
- which = int(sys.argv[i][2::])
- except ValueError:
- pass
-
- if sys.argv[i].upper() == "-ALL":
- components = "gpaQStfMcWDUk"
- names = True
- testing = True
-
- if sys.argv[i].startswith("-I"):
- draw_icons = True
- # We can only have one icon per executor, so let's strip components to the first one
- components = sys.argv[i][2]
- # exception for UL/DL speed; to assign an icon to it we need to calculate speeds first
- if components != "k":
- path_to_icon = icon_path(home, components)
-
- if sys.argv[i].startswith("-M"):
- # We can only have a custom name for a single component
- components = components[0]
- names = True
- c_name = sys.argv[i][2::]
-
- if testing:
- time_start = int(round(time.time() * 1000))
-
- output = ""
-
- # Prepare ONLY requested data, ONLY once
- if "g" or "p" in components:
- try:
- pcpu = psutil.cpu_percent(interval=1, percpu=True)
- except:
- pass
-
- if "a" in components:
- try:
- avg = str(psutil.cpu_percent(interval=1))
- if len(avg) < 4:
- avg = " " + avg
- except:
- pass
-
- if "s" or "S" in components:
- try:
- speed = psutil.cpu_freq(False)
- except:
- pass
-
- if "q" or "Q" in components:
- try:
- freqs = psutil.cpu_freq(True)
- if len(freqs) == 0:
- freqs = None
- except:
- pass
-
- if "t" in components:
- try:
- temp = psutil.sensors_temperatures(fahrenheit)
- except:
- pass
-
- if "f" in components:
- try:
- fans = psutil.sensors_fans()
- except:
- pass
-
- if "m" or "M" or "z" or "Z" in components:
- try:
- memory = psutil.virtual_memory()
- except:
- pass
-
- if "w" or "W" or "x" in components:
- try:
- swap = psutil.swap_memory()
- except:
- pass
-
- if "k" in components:
- try:
- xfer_start = psutil.net_io_counters()
- time.sleep(1)
- xfer_finish = psutil.net_io_counters()
- ul = (xfer_finish[0] - xfer_start[0]) / 1024
- dl = (xfer_finish[1] - xfer_start[1]) / 1024
- # We've not selected an icon previously. Now we have enough data.
- if draw_icons:
- path_to_icon = net_icon(home, ul, dl)
- except:
- pass
-
- drives = []
- # Find drive names, mountpoints
- if "d" or "D" or "n" or "N" in components:
- try:
- d = psutil.disk_partitions()
- # This will store name, mountpoint
- for entry in d:
- n = entry[0].split("/")
- name = n[len(n) - 1]
- # name, mountpoint
- drive = name, entry[1]
- drives.append(drive)
- except:
- pass
-
- if "d" or "D" in components:
- try:
- disks_usage = []
- for drive in drives:
- # Search drives by path
- data = psutil.disk_usage(drive[1])
- # Store name, used, total, percent
- essential = drive[0].upper(), data[1], data[0], data[3]
- disks_usage.append(essential)
- except:
- pass
-
- if "n" in components or "N" in components:
- try:
- disks_usage = []
- for drive in drives:
- # Search drives by path
- data = psutil.disk_usage(drive[1])
- # Store mountpoint, used, total, percent
- essential = drive[1], data[1], data[0], data[3]
- disks_usage.append(essential)
- except:
- pass
-
- if "u" or "U" in components:
- try:
- b_time = psutil.boot_time()
- except:
- pass
-
- # Build output component after component
- output += separator
-
- for char in components:
- if char == "g" and pcpu is not None:
- if c_name:
- output += c_name
- output += graph_per_cpu(pcpu) + separator
-
- if char == "p" and pcpu is not None:
- if names:
- output += c_name if c_name else "CPU: "
- output += per_cpu(pcpu) + separator
-
- if char == "a" and avg is not None:
- if names:
- output += c_name if c_name else "avCPU: "
- output += avg + "%" + separator
-
- if char == "q" and freqs is not None:
- if names:
- output += c_name if c_name else "CPU: "
- output += freq_per_cpu(freqs)[0][:-1] + " GHz" + separator
-
- if char == "Q" and freqs is not None:
- if names:
- output += c_name if c_name else "CPU: "
- result = freq_per_cpu(freqs)
- output += result[0][:-1] + "/" + str(result[1]) + " GHz" + separator
-
- if char == "s" and speed is not None:
- if names:
- output += c_name if c_name else "SPD: "
- output += str(round(speed[0] / 1000, 1)) + " GHz" + separator
-
- if char == "S" and speed is not None:
- if names:
- output += c_name if c_name else "avSPD: "
- output += str(round(speed[0] / 1000, 1)) + "/" + str(round(speed[2] / 1000, 1)) + " GHz" + separator
-
- if char == "t" and temp is not None and len(temp) > 0:
- if names:
- output += c_name if c_name else "CORE: "
- if "k10temp" in temp.keys():
- # ryzen, multiple Die temperatures for threadripper/Epyc
- ryzen_die_temps = [sensor.current for sensor in temp["k10temp"] if sensor.label == 'Tdie']
- output += str(int(max(ryzen_die_temps)))
- if "coretemp" in temp.keys():
- # intel
- output += str(int(temp["coretemp"][0][1]))
- output += "℉" if fahrenheit else "℃"
- output += separator
-
- if char == "f" and fans is not None and len(fans) > 0:
- if names:
- output += c_name if c_name else "FAN: "
- fan0 = next(iter(fans.values()))
- output += str(fan0[0][1]) + "/m" + separator
-
- if char == 'm' and memory is not None:
- if names:
- output += c_name if c_name else "MEM: "
- output += str(round((memory[0] - memory[1]) / 1073741824, 1)) + " GB" + separator
-
- if char == 'M' and memory is not None:
- if names:
- output += c_name if c_name else "MEM: "
- output += str(round((memory[3]) / 1073741824, 1)) + "/" + str(
- round(memory[0] / 1073741824, 1)) + " GB" + separator
-
- if char == 'c' and memory is not None:
- if names:
- output += c_name if c_name else "MEM: "
- output += str(memory[2]) + "%" + separator
-
- if char == 'C' and memory is not None:
- if names:
- output += c_name if c_name else "MEM: "
- output += str(100 - memory[2]) + "%" + separator
-
- if char == 'u' and b_time is not None:
- up_time = int(time.time()) - b_time
- m, s = divmod(up_time, 60)
- h, m = divmod(m, 60)
- if names:
- output += c_name if c_name else "UP: "
- output += "%d:%02d" % (h, m) + separator
-
- if char == 'U' and b_time is not None:
- up_time = int(time.time()) - b_time
- m, s = divmod(up_time, 60)
- h, m = divmod(m, 60)
- if names:
- output += c_name if c_name else "UP: "
- output += "%d:%02d:%02d" % (h, m, s) + separator
-
- if char == "w" and swap is not None:
- if names:
- output += c_name if c_name else "SWAP: "
- output += str(round(swap[1] / 1073741824, 1)) + " GB" + separator
-
- if char == "W" and swap is not None:
- if names:
- output += c_name if c_name else "SWAP: "
- output += str(round(swap[1] / 1073741824, 1)) + "/"
- output += str(round(swap[0] / 1073741824, 1)) + " GB" + separator
-
- if char == "x" and swap is not None:
- if names:
- output += c_name if c_name else "SWAP: "
- output += str(swap[3]) + "%" + separator
-
- if char == "d" or char == "n" and disks_usage is not None:
- if which is not None:
- try:
- entry = disks_usage[which]
- output += entry[0] + ": "
- output += str(entry[3]) + "%" + separator
- except IndexError:
- pass
- else:
- for entry in disks_usage:
- output += entry[0] + ": "
- output += str(entry[3]) + "%" + separator
-
- if char == "D" or char == "N" and disks_usage is not None:
- if c_name:
- output += c_name
- if which is not None:
- try:
- entry = disks_usage[which]
- output += entry[0] + ": "
- output += str(round(entry[1] / 1073741824, 1)) + "/"
- output += str(round(entry[2] / 1073741824, 1)) + " GB" + separator
- except IndexError:
- pass
- else:
- for entry in disks_usage:
- output += entry[0] + ": "
- output += str(round(entry[1] / 1073741824, 1)) + "/"
- output += str(round(entry[2] / 1073741824, 1)) + " GB" + separator
-
- if char == "k":
- if names and xfer_start is not None and xfer_finish is not None:
- output += c_name if c_name else "Net: "
- output += '{:0.2f}'.format((xfer_finish[0] - xfer_start[0]) / 1024) + ' {:0.2f} kB/s'.format(
- (xfer_finish[1] - xfer_start[1]) / 1024) + separator
-
- if testing:
- output += "[" + str(int((round(time.time() * 1000)) - time_start) / 1000) + "s]" + separator
-
- # remove leading and trailing separator
- l = len(separator)
- if l > 0:
- output = output[l:-l]
-
- if draw_icons:
- print(path_to_icon)
-
- print(output)
-
-
-def per_cpu(result):
- string = ""
- for val in result:
- proc = str(int(round(val, 1)))
- if len(proc) < 2:
- proc = " " + proc
- string += proc + "% "
- return string
-
-
-def freq_per_cpu(result):
- string = ""
- max_freq = 0
- for val in result:
- freq = str(round(val[0] / 1000, 1))
- string += freq + "|"
- max_freq = str(round(val[2] / 1000, 1))
-
- return string, max_freq
-
-
-def graph_per_cpu(result):
- graph = "_▁▂▃▄▅▆▇███"
-
- string = ""
- for val in result:
- proc = int(round(val / 10, 0))
- string += graph[proc]
- return string
-
-
-def print_help():
-
- print("\npsuinfo [-C{components}] | [-I{component}] [-F] [-N] [-S<number>] | [-S<string>] [-T] [-W{number}] [-all] [-h] [--help]")
-
- print("\n-C defines multiple components. -I defines a single component. If none given, -CgStfM argument will be used by default.\n")
- print(" g - (g)raphical CPU load bar")
- print(" p - (p)ercentage for each core (text)")
- print(" a - (a)verage CPU load (text)")
- print(" q - fre(q)ency for each thread")
- print(" Q - fre(Q)ency for each thread/max frequency")
- print(" s - current CPU (s)peed")
- print(" S - current/max CPU (S)peed")
- print(" t - CPU (t)emperature")
- print(" f - (f)an speed")
- print(" m - (m)emory in use")
- print(" M - (M)emory in use/total")
- print(" c - used memory per(c)entage")
- print(" C - free memory per(C)entage")
- print(" w - s(w)ap memory in use")
- print(" W - s(W)ap memory in use/total")
- print(" x - swap usage in %")
- print(" d - (d)rives as names usage in %")
- print(" D - (D)rives as names used/total")
- print(" n - drives as mou(n)tpoints usage in %")
- print(" N - drives as mou(N)tpoints used/total")
- print(" u - (u)ptime HH:MM")
- print(" U - (U)ptime HH:MM:SS")
- print(" k - current networ(k) traffic as upload/download in kB/s")
-
- print("\n-F - use Fahrenheit instead of ℃")
- print("-N - display field names (except for (g)raphical CPU load bar)")
- print("-S<number> - number of spaces between components (-S2 if none given)")
- print("-S<string> for custom separator (use \' | \' to include spaces)")
- print("-M<string> for custom component name (\'My custom name: \')")
- print("-T - test execution time")
- print("-all - display all possible data (for use in terminal)\n")
-
- print("-I<component> - show an icon before text; 1 component per executor allowed")
- print("-W<number> - select 0 to n-th element from multiple output (drives, mountpoints)\n")
-
-
-def icon_path(home, component):
- icons = {'g': '',
- 'p': 'cpu.svg',
- 'a': 'cpu.svg',
- 'q': 'cpu.svg',
- 'Q': 'cpu.svg',
- 's': 'cpu.svg',
- 'S': 'cpu.svg',
- 't': 'temp.svg',
- 'f': 'fan.svg',
- 'm': 'network-card.svg',
- 'M': 'network-card.svg',
- 'c': 'network-card.svg',
- 'C': 'network-card.svg',
- 'w': 'swap.svg',
- 'W': 'swap.svg',
- 'x': 'swap.svg',
- 'd': 'drive-harddisk.svg',
- 'D': 'drive-harddisk.svg',
- 'n': 'drive-harddisk.svg',
- 'N': 'drive-harddisk.svg',
- 'u': 'system.svg',
- 'U': 'system.svg'}
- try:
- f_name = icons[component]
- except KeyError:
- return ""
-
- return icon_to_use(home, f_name)
-
-
-def net_icon(home, ul, dl):
- f_name = "knemo-monitor-transmit.svg"
- return icon_to_use(home, f_name)
-
-
-def icon_to_use(home, f_name):
- icon_custom = home + '/.local/share/psuinfo/' + f_name
- icon_default = "/usr/share/icons/MB-Mango-Suru-GLOW/devices/16/" + f_name
- if os.path.isfile(icon_custom):
- return icon_custom
- else:
- return icon_default
-
-
-if __name__ == "__main__":
- main()
