[welcome] split into multiple modules
This commit is contained in:
Dmytro Meleshko
parent
190c04cb4b
commit
82b6f6b5fd
|
|
@ -0,0 +1 @@
|
|||
*.pyc
|
||||
|
|
@ -0,0 +1,21 @@
|
|||
from colorama import Fore, Style, ansi
|
||||
|
||||
COLORS = [ansi.code_to_chars(30 + color_index) for color_index in range(0, 8)]
|
||||
|
||||
|
||||
def colored(string, *colors):
|
||||
return "".join(colors + (string, Style.RESET_ALL))
|
||||
|
||||
|
||||
def bright_colored(string, *colors):
|
||||
return "".join(colors + (Style.BRIGHT, string, Style.RESET_ALL))
|
||||
|
||||
|
||||
def colorize_percent(percent, warning, critical, inverse=False):
|
||||
COLORS = [Fore.GREEN, Fore.YELLOW, Fore.RED]
|
||||
|
||||
color_index = 0 if percent < warning else 1 if percent < critical else 2
|
||||
if inverse:
|
||||
color_index = 2 - color_index
|
||||
|
||||
return colored("%.2f%%" % percent, COLORS[color_index])
|
||||
|
|
@ -0,0 +1,33 @@
|
|||
def humanize_timedelta(timedelta):
|
||||
result = []
|
||||
|
||||
days = timedelta.days
|
||||
mm, ss = divmod(timedelta.seconds, 60)
|
||||
hh, mm = divmod(mm, 60)
|
||||
|
||||
def plural(n):
|
||||
return n, "s" if abs(n) != 1 else ""
|
||||
|
||||
if days > 0:
|
||||
result.append("%d day%s" % plural(days))
|
||||
if hh > 0 or result:
|
||||
result.append("%d hour%s" % plural(hh))
|
||||
if mm > 0 or result:
|
||||
result.append("%d min%s" % plural(mm))
|
||||
if len(result) <= 1:
|
||||
result.append("%d sec%s" % plural(ss))
|
||||
|
||||
return ", ".join(result)
|
||||
|
||||
|
||||
def humanize_bytes(bytes):
|
||||
units = ["B", "kB", "MB", "GB"]
|
||||
|
||||
factor = 1
|
||||
for _unit in units:
|
||||
next_factor = factor << 10
|
||||
if bytes < next_factor:
|
||||
break
|
||||
factor = next_factor
|
||||
|
||||
return "%.2f %s" % (float(bytes) / factor, _unit)
|
||||
243
welcome/main.py
243
welcome/main.py
|
|
@ -1,246 +1,7 @@
|
|||
import os
|
||||
import platform
|
||||
import re
|
||||
import socket
|
||||
from datetime import datetime, timedelta
|
||||
from getpass import getuser
|
||||
|
||||
import psutil
|
||||
from colorama import Fore, Style, ansi
|
||||
|
||||
COLORS = [ansi.code_to_chars(30 + color_index) for color_index in range(0, 8)]
|
||||
|
||||
|
||||
def colored(string, *colors):
|
||||
return "".join(colors + (string, Style.RESET_ALL))
|
||||
|
||||
|
||||
def bright_colored(string, *colors):
|
||||
return "".join(colors + (Style.BRIGHT, string, Style.RESET_ALL))
|
||||
|
||||
|
||||
def format_timedelta(timedelta):
|
||||
result = []
|
||||
|
||||
days = timedelta.days
|
||||
mm, ss = divmod(timedelta.seconds, 60)
|
||||
hh, mm = divmod(mm, 60)
|
||||
|
||||
def plural(n):
|
||||
return n, "s" if abs(n) != 1 else ""
|
||||
|
||||
if days > 0:
|
||||
result.append("%d day%s" % plural(days))
|
||||
if hh > 0 or result:
|
||||
result.append("%d hour%s" % plural(hh))
|
||||
if mm > 0 or result:
|
||||
result.append("%d min%s" % plural(mm))
|
||||
if len(result) <= 1:
|
||||
result.append("%d sec%s" % plural(ss))
|
||||
|
||||
return ", ".join(result)
|
||||
|
||||
|
||||
def humanize_bytes(bytes):
|
||||
units = ["B", "kB", "MB", "GB"]
|
||||
|
||||
factor = 1
|
||||
for _unit in units:
|
||||
next_factor = factor << 10
|
||||
if bytes < next_factor:
|
||||
break
|
||||
factor = next_factor
|
||||
|
||||
return "%.2f %s" % (float(bytes) / factor, _unit)
|
||||
|
||||
|
||||
def colorize_percent(percent, warning, critical, inverse=False):
|
||||
COLORS = [Fore.GREEN, Fore.YELLOW, Fore.RED]
|
||||
|
||||
color_index = 0 if percent < warning else 1 if percent < critical else 2
|
||||
if inverse:
|
||||
color_index = 2 - color_index
|
||||
|
||||
return colored("%.2f%%" % percent, COLORS[color_index])
|
||||
|
||||
|
||||
def get_hostname():
|
||||
hostname = socket.gethostname()
|
||||
local_ip = socket.gethostbyname(hostname)
|
||||
return hostname, local_ip
|
||||
|
||||
|
||||
def uptime():
|
||||
return datetime.now() - datetime.fromtimestamp(psutil.boot_time())
|
||||
|
||||
|
||||
def users():
|
||||
users = {}
|
||||
|
||||
for user in psutil.users():
|
||||
name = user.name
|
||||
terminal = user.terminal
|
||||
if name in users:
|
||||
users[name].append(terminal)
|
||||
else:
|
||||
users[name] = [terminal]
|
||||
|
||||
result = []
|
||||
|
||||
for name in users:
|
||||
terminals = users[name]
|
||||
|
||||
colored_name = bright_colored(name, Fore.BLUE)
|
||||
colored_terminals = [
|
||||
colored(term, Style.BRIGHT, Fore.BLACK) for term in terminals
|
||||
]
|
||||
|
||||
terminals_str = ", ".join(colored_terminals)
|
||||
if len(colored_terminals) > 1:
|
||||
terminals_str = "(%s)" % terminals_str
|
||||
result.append(colored_name + "@" + terminals_str)
|
||||
|
||||
return ", ".join(result)
|
||||
|
||||
|
||||
def shell():
|
||||
return os.environ["SHELL"]
|
||||
|
||||
|
||||
def cpu_usage():
|
||||
percent = psutil.cpu_percent()
|
||||
|
||||
return colorize_percent(percent, warning=60, critical=80)
|
||||
|
||||
|
||||
def memory():
|
||||
memory = psutil.virtual_memory()
|
||||
return (
|
||||
humanize_bytes(memory.used),
|
||||
humanize_bytes(memory.total),
|
||||
colorize_percent(memory.percent, warning=60, critical=80),
|
||||
)
|
||||
|
||||
|
||||
def disks():
|
||||
result = []
|
||||
for disk in psutil.disk_partitions(all=False):
|
||||
if psutil.WINDOWS and ("cdrom" in disk.opts or disk.fstype == ""):
|
||||
# skip cd-rom drives with no disk in it on Windows; they may raise
|
||||
# ENOENT, pop-up a Windows GUI error for a non-ready partition or
|
||||
# just hang
|
||||
continue
|
||||
|
||||
usage = psutil.disk_usage(disk.mountpoint)
|
||||
result.append(
|
||||
(
|
||||
disk.mountpoint,
|
||||
humanize_bytes(usage.used),
|
||||
humanize_bytes(usage.total),
|
||||
colorize_percent(usage.percent, warning=70, critical=85),
|
||||
)
|
||||
)
|
||||
return result
|
||||
|
||||
|
||||
def battery():
|
||||
if not hasattr(psutil, "sensors_battery"):
|
||||
return None
|
||||
|
||||
try:
|
||||
battery = psutil.sensors_battery()
|
||||
except Exception as e:
|
||||
print(e)
|
||||
return None
|
||||
|
||||
if battery is None:
|
||||
return None
|
||||
|
||||
percent = battery.percent
|
||||
|
||||
if battery.power_plugged:
|
||||
status = "charging" if percent < 100 else "fully charged"
|
||||
else:
|
||||
status = "%s left" % format_timedelta(timedelta(seconds=battery.secsleft))
|
||||
|
||||
return colorize_percent(percent, critical=10, warning=20, inverse=True), status
|
||||
|
||||
|
||||
def is_android():
|
||||
return os.path.isdir("/system/app") and os.path.isdir("/system/priv-app")
|
||||
|
||||
|
||||
def get_distro_info():
|
||||
if psutil.WINDOWS:
|
||||
return "windows", platform.system(), platform.release(), ""
|
||||
elif psutil.OSX:
|
||||
from plistlib import readPlist
|
||||
|
||||
sw_vers = readPlist("/System/Library/CoreServices/SystemVersion.plist")
|
||||
return "mac", sw_vers["ProductName"], sw_vers["ProductVersion"], ""
|
||||
elif is_android():
|
||||
from subprocess import check_output
|
||||
|
||||
android_version = check_output(["getprop", "ro.build.version.release"])
|
||||
return "android", "Android", android_version.decode().strip(), ""
|
||||
elif psutil.LINUX:
|
||||
import distro
|
||||
|
||||
return distro.id(), distro.name(), distro.version(), distro.codename()
|
||||
|
||||
raise NotImplementedError("unsupported OS")
|
||||
|
||||
|
||||
def get_system_info():
|
||||
info_lines = []
|
||||
|
||||
def info(name, value, *format_args):
|
||||
line = colored(name + ":", Style.BRIGHT, Fore.YELLOW) + " " + value
|
||||
if format_args:
|
||||
line = line % format_args
|
||||
info_lines.append(line)
|
||||
|
||||
username = getuser()
|
||||
hostname, local_ip = get_hostname()
|
||||
|
||||
info_lines.append(
|
||||
bright_colored(username, Fore.BLUE) + "@" + bright_colored(hostname, Fore.RED)
|
||||
)
|
||||
info_lines.append("")
|
||||
|
||||
distro_id, distro_name, distro_version, distro_codename = get_distro_info()
|
||||
info("OS", " ".join([distro_name, distro_version, distro_codename]))
|
||||
|
||||
logo_path = os.path.join(os.path.dirname(__file__), "logos", distro_id)
|
||||
with open(logo_path) as logo_file:
|
||||
logo_lines = logo_file.read().splitlines()
|
||||
|
||||
info("Kernel", "%s %s", platform.system(), platform.release())
|
||||
|
||||
info("Uptime", format_timedelta(uptime()))
|
||||
|
||||
users_info = users()
|
||||
if users_info:
|
||||
info("Users", users_info)
|
||||
|
||||
info("Shell", shell())
|
||||
|
||||
info("IP address", local_ip)
|
||||
|
||||
info_lines.append("")
|
||||
|
||||
info("CPU Usage", "%s", cpu_usage())
|
||||
info("Memory", "%s / %s (%s)", *memory())
|
||||
|
||||
for disk_info in disks():
|
||||
info("Disk (%s)", "%s / %s (%s)", *disk_info)
|
||||
|
||||
battery_info = battery()
|
||||
if battery_info is not None:
|
||||
info("Battery", "%s (%s)", *battery_info)
|
||||
|
||||
return logo_lines, info_lines
|
||||
|
||||
from colors import Style, COLORS
|
||||
from system_info import get_system_info
|
||||
|
||||
print("")
|
||||
|
||||
|
|
|
|||
|
|
@ -0,0 +1,186 @@
|
|||
import os
|
||||
import platform
|
||||
import socket
|
||||
from datetime import datetime, timedelta
|
||||
from getpass import getuser
|
||||
|
||||
import psutil
|
||||
|
||||
from colors import Fore, Style, bright_colored, colored, colorize_percent
|
||||
from humanize import humanize_bytes, humanize_timedelta
|
||||
|
||||
|
||||
def get_system_info():
|
||||
info_lines = []
|
||||
|
||||
def info(name, value, *format_args):
|
||||
line = colored(name + ":", Style.BRIGHT, Fore.YELLOW) + " " + value
|
||||
if format_args:
|
||||
line = line % format_args
|
||||
info_lines.append(line)
|
||||
|
||||
username = getuser()
|
||||
hostname, local_ip = _get_hostname()
|
||||
|
||||
info_lines.append(
|
||||
bright_colored(username, Fore.BLUE) + "@" + bright_colored(hostname, Fore.RED)
|
||||
)
|
||||
info_lines.append("")
|
||||
|
||||
distro_id, distro_name, distro_version, distro_codename = _get_distro_info()
|
||||
info("OS", " ".join([distro_name, distro_version, distro_codename]))
|
||||
|
||||
logo_path = os.path.join(os.path.dirname(__file__), "logos", distro_id)
|
||||
with open(logo_path) as logo_file:
|
||||
logo_lines = logo_file.read().splitlines()
|
||||
|
||||
info("Kernel", "%s %s", platform.system(), platform.release())
|
||||
|
||||
info("Uptime", humanize_timedelta(_get_uptime()))
|
||||
|
||||
users_info = _get_users()
|
||||
if users_info:
|
||||
info("Users", users_info)
|
||||
|
||||
info("Shell", _get_shell())
|
||||
|
||||
info("IP address", local_ip)
|
||||
|
||||
info_lines.append("")
|
||||
|
||||
info("CPU Usage", "%s", _get_cpu_usage())
|
||||
info("Memory", "%s / %s (%s)", *_get_memory())
|
||||
|
||||
for disk_info in _get_disks():
|
||||
info("Disk (%s)", "%s / %s (%s)", *disk_info)
|
||||
|
||||
battery_info = _get_battery()
|
||||
if battery_info is not None:
|
||||
info("Battery", "%s (%s)", *battery_info)
|
||||
|
||||
return logo_lines, info_lines
|
||||
|
||||
|
||||
def _get_hostname():
|
||||
hostname = socket.gethostname()
|
||||
local_ip = socket.gethostbyname(hostname)
|
||||
return hostname, local_ip
|
||||
|
||||
|
||||
def _get_uptime():
|
||||
return datetime.now() - datetime.fromtimestamp(psutil.boot_time())
|
||||
|
||||
|
||||
def _get_users():
|
||||
users = {}
|
||||
|
||||
for user in psutil.users():
|
||||
name = user.name
|
||||
terminal = user.terminal
|
||||
if name in users:
|
||||
users[name].append(terminal)
|
||||
else:
|
||||
users[name] = [terminal]
|
||||
|
||||
result = []
|
||||
|
||||
for name in users:
|
||||
terminals = users[name]
|
||||
|
||||
colored_name = bright_colored(name, Fore.BLUE)
|
||||
colored_terminals = [
|
||||
colored(term, Style.BRIGHT, Fore.BLACK) for term in terminals
|
||||
]
|
||||
|
||||
terminals_str = ", ".join(colored_terminals)
|
||||
if len(colored_terminals) > 1:
|
||||
terminals_str = "(%s)" % terminals_str
|
||||
result.append(colored_name + "@" + terminals_str)
|
||||
|
||||
return ", ".join(result)
|
||||
|
||||
|
||||
def _get_shell():
|
||||
return os.environ["SHELL"]
|
||||
|
||||
|
||||
def _get_cpu_usage():
|
||||
percent = psutil.cpu_percent()
|
||||
|
||||
return colorize_percent(percent, warning=60, critical=80)
|
||||
|
||||
|
||||
def _get_memory():
|
||||
memory = psutil.virtual_memory()
|
||||
return (
|
||||
humanize_bytes(memory.used),
|
||||
humanize_bytes(memory.total),
|
||||
colorize_percent(memory.percent, warning=60, critical=80),
|
||||
)
|
||||
|
||||
|
||||
def _get_disks():
|
||||
result = []
|
||||
for disk in psutil.disk_partitions(all=False):
|
||||
if psutil.WINDOWS and ("cdrom" in disk.opts or disk.fstype == ""):
|
||||
# skip cd-rom drives with no disk in it on Windows; they may raise
|
||||
# ENOENT, pop-up a Windows GUI error for a non-ready partition or
|
||||
# just hang
|
||||
continue
|
||||
|
||||
usage = psutil.disk_usage(disk.mountpoint)
|
||||
result.append(
|
||||
(
|
||||
disk.mountpoint,
|
||||
humanize_bytes(usage.used),
|
||||
humanize_bytes(usage.total),
|
||||
colorize_percent(usage.percent, warning=70, critical=85),
|
||||
)
|
||||
)
|
||||
return result
|
||||
|
||||
|
||||
def _get_battery():
|
||||
if not hasattr(psutil, "sensors_battery"):
|
||||
return None
|
||||
|
||||
try:
|
||||
battery = psutil.sensors_battery()
|
||||
except Exception as e:
|
||||
print(e)
|
||||
return None
|
||||
|
||||
if battery is None:
|
||||
return None
|
||||
|
||||
percent = battery.percent
|
||||
if battery.power_plugged:
|
||||
status = "charging" if percent < 100 else "fully charged"
|
||||
else:
|
||||
status = "%s left" % humanize_timedelta(timedelta(seconds=battery.secsleft))
|
||||
return colorize_percent(percent, critical=10, warning=20, inverse=True), status
|
||||
|
||||
|
||||
def _get_distro_info():
|
||||
if psutil.WINDOWS:
|
||||
return "windows", platform.system(), platform.release(), ""
|
||||
elif psutil.OSX:
|
||||
from plistlib import readPlist
|
||||
|
||||
sw_vers = readPlist("/System/Library/CoreServices/SystemVersion.plist")
|
||||
return "mac", sw_vers["ProductName"], sw_vers["ProductVersion"], ""
|
||||
elif _is_android():
|
||||
from subprocess import check_output
|
||||
|
||||
android_version = check_output(["getprop", "ro.build.version.release"])
|
||||
return "android", "Android", android_version.decode().strip(), ""
|
||||
elif psutil.LINUX:
|
||||
import distro
|
||||
|
||||
return distro.id(), distro.name(), distro.version(), distro.codename()
|
||||
|
||||
raise NotImplementedError("unsupported OS")
|
||||
|
||||
|
||||
def _is_android():
|
||||
return os.path.isdir("/system/app") and os.path.isdir("/system/priv-app")
|
||||
Loading…
Reference in New Issue