419 lines
14 KiB
Python
419 lines
14 KiB
Python
|
"""
|
||
|
|
* Author: "PeppermintOS Team(peppermintosteam@proton.me)
|
||
|
|
*
|
||
|
|
* License: SPDX-License-Identifier: GPL-3.0-or-later
|
||
|
|
*
|
||
|
|
* This is a neo / screen fetch type tool written for python
|
||
|
|
"""
|
||
|
|
|
||
|
|
import platform
|
||
|
|
import os
|
||
|
|
import socket
|
||
|
|
import subprocess
|
||
|
|
import psutil
|
||
|
|
|
||
|
|
|
||
|
|
def get_mem_info():
|
||
|
|
""" Get the Memory information """
|
||
|
|
try:
|
||
|
|
free_output = subprocess.check_output(['free', '-m'],
|
||
|
|
universal_newlines=True)
|
||
|
|
lines = free_output.split('\n')
|
||
|
|
total_memory = int(lines[1].split()[1])
|
||
|
|
used_memory = int(lines[1].split()[2])
|
||
|
|
memory_info = f"{used_memory}MiB / {total_memory}MiB"
|
||
|
|
return memory_info
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
memory_info = 'Memory information not available'
|
||
|
|
return memory_info
|
||
|
|
|
||
|
|
|
||
|
|
def get_cpu_info():
|
||
|
|
""" Get the CPU INformation """
|
||
|
|
try:
|
||
|
|
lscpu_output= subprocess.check_output(['lscpu'],
|
||
|
|
universal_newlines=True)
|
||
|
|
cpu_info = ([line.split(":")[1].strip()
|
||
|
|
for line in lscpu_output.split("\n")
|
||
|
|
if "Model name" in line][0]
|
||
|
|
)
|
||
|
|
return cpu_info
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
cpu_info = 'CPU information not available'
|
||
|
|
return cpu_info
|
||
|
|
|
||
|
|
|
||
|
|
def get_uptime_info():
|
||
|
|
""" This will get the Uptime of the system """
|
||
|
|
try:
|
||
|
|
with open('/proc/uptime', 'r', encoding='utf-8') as _:
|
||
|
|
uptime_seconds = float(_.readline().split()[0])
|
||
|
|
uptime_days = int(uptime_seconds // (24 * 3600))
|
||
|
|
uptime_seconds %= (24 * 3600)
|
||
|
|
uptime_hours = int(uptime_seconds // 3600)
|
||
|
|
uptime_seconds %= 3600
|
||
|
|
uptime_minutes = int(uptime_seconds // 60)
|
||
|
|
uptime_str = (
|
||
|
|
f"{uptime_days} days, "
|
||
|
|
f"{uptime_hours} hours, "
|
||
|
|
f"{uptime_minutes} mins"
|
||
|
|
)
|
||
|
|
return uptime_str
|
||
|
|
except FileNotFoundError:
|
||
|
|
uptime_str = 'Uptime information not available'
|
||
|
|
return uptime_str
|
||
|
|
|
||
|
|
|
||
|
|
def get_os_info():
|
||
|
|
""" Get the OS Information """
|
||
|
|
if os_name == 'Linux':
|
||
|
|
try:
|
||
|
|
with open('/etc/os-release', encoding='utf-8') as file:
|
||
|
|
for line in file:
|
||
|
|
if line.startswith('PRETTY_NAME'):
|
||
|
|
os_pretty_name = (
|
||
|
|
line.strip().split('=')[1].strip('"')
|
||
|
|
)
|
||
|
|
return os_pretty_name
|
||
|
|
return 'Unknown Linux Distribution'
|
||
|
|
except FileNotFoundError:
|
||
|
|
return 'Unknown Linux Distribution'
|
||
|
|
else:
|
||
|
|
return os_name
|
||
|
|
|
||
|
|
|
||
|
|
def get_package_info():
|
||
|
|
""" Get the package Information """
|
||
|
|
try:
|
||
|
|
package_count = subprocess.check_output(['dpkg', '--list'],
|
||
|
|
universal_newlines=True).count('\n')
|
||
|
|
package_count -= 5 # Adjust for header lines in the dpkg output
|
||
|
|
package_info = f"{package_count} packages installed"
|
||
|
|
return package_info
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
return 'Failed to get package information (dpkg command failed)'
|
||
|
|
except FileNotFoundError:
|
||
|
|
return 'dpkg command not found'
|
||
|
|
except OSError as _:
|
||
|
|
return f'OS error: {_}'
|
||
|
|
|
||
|
|
|
||
|
|
def get_shell_info():
|
||
|
|
""" Get the Shell Information """
|
||
|
|
try:
|
||
|
|
version_output = subprocess.check_output(['bash', '--version'],
|
||
|
|
universal_newlines=True)
|
||
|
|
lines = version_output.split('\n')
|
||
|
|
for line in lines:
|
||
|
|
if line.startswith('GNU bash'):
|
||
|
|
version_parts = line.split()
|
||
|
|
if len(version_parts) >= 4:
|
||
|
|
shell_version = version_parts[3]
|
||
|
|
return shell_version
|
||
|
|
return 'Version not found'
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
return 'Failed to get shell information (bash command failed)'
|
||
|
|
except FileNotFoundError:
|
||
|
|
return 'bash command not found'
|
||
|
|
except OSError as _:
|
||
|
|
return f'OS error: {_}'
|
||
|
|
|
||
|
|
|
||
|
|
def get_display_info():
|
||
|
|
""" Get the Display Resolution """
|
||
|
|
try:
|
||
|
|
xdpyinfo_output = subprocess.check_output(['xdpyinfo'],
|
||
|
|
universal_newlines=True)
|
||
|
|
resolutions = []
|
||
|
|
for line in xdpyinfo_output.split('\n'):
|
||
|
|
if 'dimensions:' in line:
|
||
|
|
parts = line.split()
|
||
|
|
if len(parts) >= 2:
|
||
|
|
resolution = parts[1]
|
||
|
|
resolutions.append(resolution)
|
||
|
|
resolution_info = ', '.join(resolutions)
|
||
|
|
return resolution_info
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
return 'Failed to get display information (xdpyinfo command failed)'
|
||
|
|
except FileNotFoundError:
|
||
|
|
return 'xdpyinfo command not found'
|
||
|
|
except OSError as _:
|
||
|
|
return f'OS error: {_}'
|
||
|
|
|
||
|
|
|
||
|
|
def get_terminal_emulator():
|
||
|
|
""" Find the terminal being used """
|
||
|
|
current_process = psutil.Process(os.getpid())
|
||
|
|
while current_process:
|
||
|
|
parent = current_process.parent()
|
||
|
|
if parent is None:
|
||
|
|
break
|
||
|
|
current_process = parent
|
||
|
|
name = current_process.name().lower()
|
||
|
|
if any(term in name for term in ["gnome-terminal", "konsole",
|
||
|
|
"xterm", "xfce4-terminal", "lxterminal", "mate-terminal",
|
||
|
|
"tilix", "terminator", "alacritty", "kitty"]):
|
||
|
|
return current_process.name()
|
||
|
|
return "Terminal emulator not found"
|
||
|
|
|
||
|
|
|
||
|
|
def get_gpu_info():
|
||
|
|
""" Get the GPU Information """
|
||
|
|
try:
|
||
|
|
output = subprocess.check_output(['lspci', '-nnk'],
|
||
|
|
universal_newlines=True)
|
||
|
|
gpu_lines = ([line.strip()
|
||
|
|
for line in output.split('\n') if 'VGA controller' in line])
|
||
|
|
if gpu_lines:
|
||
|
|
gpu_info = gpu_lines[0].split(': ', 1)[1]
|
||
|
|
return gpu_info
|
||
|
|
except (subprocess.CalledProcessError, FileNotFoundError) as _:
|
||
|
|
print(f"Error: {_}")
|
||
|
|
return "Unknown"
|
||
|
|
|
||
|
|
|
||
|
|
def get_desktop_environment():
|
||
|
|
""" Get the Desktop being used """
|
||
|
|
desktop_session = os.environ.get("DESKTOP_SESSION")
|
||
|
|
if desktop_session:
|
||
|
|
desktop_session = desktop_session.lower()
|
||
|
|
if desktop_session in ["gnome", "unity", "cinnamon", "mate",
|
||
|
|
"xfce", "lxde", "lxqt", "kde-plasma", "deepin", "pantheon",
|
||
|
|
"i3"]:
|
||
|
|
return desktop_session.capitalize()
|
||
|
|
if "GNOME_DESKTOP_SESSION_ID" in os.environ:
|
||
|
|
return "Gnome"
|
||
|
|
if "MATE_DESKTOP_SESSION_ID" in os.environ:
|
||
|
|
return "Mate"
|
||
|
|
if "XDG_CURRENT_DESKTOP" in os.environ:
|
||
|
|
return os.environ["XDG_CURRENT_DESKTOP"].capitalize()
|
||
|
|
return "Unknown"
|
||
|
|
|
||
|
|
|
||
|
|
def get_version(d_e):
|
||
|
|
""" Get the Desktop versions """
|
||
|
|
commands = {
|
||
|
|
"Gnome": ["gnome-shell", "--version"],
|
||
|
|
"Kde-plasma": ["plasmashell", "--version"],
|
||
|
|
"Xfce": ["xfce4-session", "--version"],
|
||
|
|
"Lxde": ["lxpanel", "--version"],
|
||
|
|
"Lxqt": ["lxqt-session", "--version"],
|
||
|
|
"Cinnamon": ["cinnamon", "--version"],
|
||
|
|
"Mate": ["mate-session", "--version"],
|
||
|
|
"Unity": ["unity", "--version"],
|
||
|
|
"Deepin": ["deepin", "--version"],
|
||
|
|
"Pantheon": ["pantheon-session", "--version"],
|
||
|
|
"I3": ["i3", "--version"]
|
||
|
|
}
|
||
|
|
try:
|
||
|
|
output = subprocess.check_output(commands[d_e],
|
||
|
|
universal_newlines=True).strip().split('\n', maxsplit=1)[0]
|
||
|
|
version = output.split()[-1]
|
||
|
|
if d_e == "Xfce":
|
||
|
|
version = get_xfce_version()
|
||
|
|
elif d_e == "I3":
|
||
|
|
version = output.split()[-2]
|
||
|
|
return version
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
return 'Failed to get display information (xdpyinfo command failed)'
|
||
|
|
except FileNotFoundError:
|
||
|
|
return 'xdpyinfo command not found'
|
||
|
|
except OSError as _:
|
||
|
|
return f'OS error: {_}'
|
||
|
|
return "unknown version"
|
||
|
|
|
||
|
|
|
||
|
|
def get_xfce_version():
|
||
|
|
""" Get the xfce information """
|
||
|
|
components = ["xfce4-session", "xfwm4", "xfce4-panel",
|
||
|
|
"xfce4-settings"]
|
||
|
|
for component in components:
|
||
|
|
try:
|
||
|
|
output = subprocess.check_output([component, "--version"],
|
||
|
|
universal_newlines=True)
|
||
|
|
lines = output.strip().split("\n")
|
||
|
|
if lines:
|
||
|
|
version = lines[0].split()[-1]
|
||
|
|
version = version.strip('()')
|
||
|
|
return version
|
||
|
|
except (subprocess.CalledProcessError, FileNotFoundError):
|
||
|
|
continue
|
||
|
|
return "unknown version"
|
||
|
|
|
||
|
|
|
||
|
|
def get_window_manager():
|
||
|
|
""" Get the windows manager information """
|
||
|
|
try:
|
||
|
|
output = subprocess.check_output(['xprop', '-root',
|
||
|
|
'_NET_SUPPORTING_WM_CHECK'], universal_newlines=True)
|
||
|
|
if '_NET_SUPPORTING_WM_CHECK(WINDOW):' in output:
|
||
|
|
window_id = output.split()[-1]
|
||
|
|
wm_name_output = subprocess.check_output(['xprop', '-id',
|
||
|
|
window_id, '_NET_WM_NAME'], universal_newlines=True)
|
||
|
|
if '_NET_WM_NAME(UTF8_STRING)' in wm_name_output:
|
||
|
|
wm_name = wm_name_output.split('"')[1]
|
||
|
|
return wm_name
|
||
|
|
except (subprocess.CalledProcessError, FileNotFoundError):
|
||
|
|
pass
|
||
|
|
return "Unknown"
|
||
|
|
|
||
|
|
|
||
|
|
def get_xfwm4_theme():
|
||
|
|
""" Get the session theme """
|
||
|
|
try:
|
||
|
|
config_path = os.path.expanduser(
|
||
|
|
"~/.config/xfce4/xfconf/xfce-perchannel-xml/xfwm4.xml"
|
||
|
|
)
|
||
|
|
if os.path.exists(config_path):
|
||
|
|
with open(config_path, 'r', encoding='utf-8') as file:
|
||
|
|
for line in file:
|
||
|
|
if 'name="theme"' in line:
|
||
|
|
theme = line.split('value="')[1].split('"')[0]
|
||
|
|
return theme
|
||
|
|
except FileNotFoundError:
|
||
|
|
pass
|
||
|
|
except OSError as _:
|
||
|
|
print(f"OS error: {_}")
|
||
|
|
return "Default"
|
||
|
|
|
||
|
|
|
||
|
|
def get_gnome_desktop_theme():
|
||
|
|
""" Get the gnome themse """
|
||
|
|
try:
|
||
|
|
output = subprocess.check_output(['gsettings', 'get',
|
||
|
|
'org.gnome.desktop.interface', 'gtk-theme'],
|
||
|
|
universal_newlines=True)
|
||
|
|
theme = output.strip().strip("'")
|
||
|
|
return theme
|
||
|
|
except (subprocess.CalledProcessError, FileNotFoundError):
|
||
|
|
pass
|
||
|
|
return "Unknown"
|
||
|
|
|
||
|
|
|
||
|
|
def get_gnome_icon_theme():
|
||
|
|
""" Get the icon theme """
|
||
|
|
try:
|
||
|
|
output = subprocess.check_output(['gsettings', 'get',
|
||
|
|
'org.gnome.desktop.interface', 'icon-theme'],
|
||
|
|
universal_newlines=True)
|
||
|
|
theme = output.strip().strip("'")
|
||
|
|
return theme
|
||
|
|
except (subprocess.CalledProcessError, FileNotFoundError):
|
||
|
|
pass
|
||
|
|
return "Unknown"
|
||
|
|
|
||
|
|
|
||
|
|
def get_xfce_icon_theme():
|
||
|
|
""" Get the icon theme """
|
||
|
|
try:
|
||
|
|
output = subprocess.check_output(['xfconf-query', '-c',
|
||
|
|
'xfce4-desktop', '-p', '/icons/icon-theme'],
|
||
|
|
universal_newlines=True)
|
||
|
|
theme = output.strip()
|
||
|
|
return theme
|
||
|
|
except (subprocess.CalledProcessError, FileNotFoundError):
|
||
|
|
pass
|
||
|
|
return "Unknown"
|
||
|
|
|
||
|
|
|
||
|
|
def get_kde_icon_theme():
|
||
|
|
""" Get the icon theme """
|
||
|
|
try:
|
||
|
|
output = subprocess.check_output(['kreadconfig5', '--get',
|
||
|
|
'kdeglobals', 'Icons'], universal_newlines=True)
|
||
|
|
theme = output.strip()
|
||
|
|
return theme
|
||
|
|
except (subprocess.CalledProcessError, FileNotFoundError):
|
||
|
|
pass
|
||
|
|
return "Unknown"
|
||
|
|
|
||
|
|
|
||
|
|
def print_user_and_hostname():
|
||
|
|
""" Get and print the host name """
|
||
|
|
username = os.getenv('USER')
|
||
|
|
# Get the hostname
|
||
|
|
hostname = os.uname().nodename
|
||
|
|
|
||
|
|
# Print the user and hostname
|
||
|
|
print(" ###-No system fetch application installed!-###")
|
||
|
|
print(" Peppermint's System Informtion Tool Will Be Used")
|
||
|
|
print("-" * len(f"{username}@{hostname}"))
|
||
|
|
print(f"{username}@{hostname}")
|
||
|
|
print("-" * len(f"{username}@{hostname}"))
|
||
|
|
|
||
|
|
|
||
|
|
# These are the variables that run the functions
|
||
|
|
os_name = platform.system()
|
||
|
|
architecture = platform.machine()
|
||
|
|
hostname_information = socket.gethostname()
|
||
|
|
kernel_version = platform.release()
|
||
|
|
kernel_information = f"Kernel: {kernel_version}"
|
||
|
|
terminal_emulator = get_terminal_emulator()
|
||
|
|
desktop_environment = get_desktop_environment()
|
||
|
|
version_informtaion = get_version(desktop_environment)
|
||
|
|
window_manager_information = get_window_manager()
|
||
|
|
wm_theme = get_xfwm4_theme()
|
||
|
|
desktop_theme = get_gnome_desktop_theme()
|
||
|
|
desktop_environment = get_desktop_environment()
|
||
|
|
gpu_information = get_gpu_info()
|
||
|
|
mem_information = get_mem_info()
|
||
|
|
cpu_information = get_cpu_info()
|
||
|
|
uptime_string = get_uptime_info()
|
||
|
|
os_pretty = get_os_info()
|
||
|
|
os_full_information = f"{os_pretty} ({architecture})"
|
||
|
|
package_information = get_package_info()
|
||
|
|
shell_information = get_shell_info()
|
||
|
|
RESOLUTION_INFORMATION = get_display_info()
|
||
|
|
|
||
|
|
# Determmine what Desktop evironment is being used
|
||
|
|
if desktop_environment == "gnome":
|
||
|
|
ICON_THEME = get_gnome_icon_theme()
|
||
|
|
elif desktop_environment == "xfce":
|
||
|
|
ICON_THEME = get_xfce_icon_theme()
|
||
|
|
elif desktop_environment == "kde-plasma":
|
||
|
|
ICON_THEME = get_kde_icon_theme()
|
||
|
|
else:
|
||
|
|
ICON_THEME = "Unknown"
|
||
|
|
#return ICON_THEME
|
||
|
|
|
||
|
|
|
||
|
|
# Setup the visual formats.
|
||
|
|
BOLD_START = "\033[1m"
|
||
|
|
BOLD_END = "\033[0m"
|
||
|
|
bold_os_text = f"{BOLD_START}OS{BOLD_END}:"
|
||
|
|
bold_host_text = f"{BOLD_START}Host:{BOLD_END}"
|
||
|
|
bold_kernel_text = f"{BOLD_START}Kernel:{BOLD_END}"
|
||
|
|
bold_uptime_text = f"{BOLD_START}Uptime:{BOLD_END}"
|
||
|
|
bold_packages_text = f"{BOLD_START}Packages:{BOLD_END}"
|
||
|
|
bold_shell_text = f"{BOLD_START}Shell:{BOLD_END}"
|
||
|
|
bold_resolution_text = f"{BOLD_START}Resolution:{BOLD_END}"
|
||
|
|
bold_cpu_text = f"{BOLD_START}CPU:{BOLD_END}"
|
||
|
|
bold_mem_text = f"{BOLD_START}Memory:{BOLD_END}"
|
||
|
|
bold_term_text = f"{BOLD_START}Terminal:{BOLD_END}"
|
||
|
|
bold_de_text = f"{BOLD_START}DE:{BOLD_END}"
|
||
|
|
bold_wm_text = f"{BOLD_START}WM:{BOLD_END}"
|
||
|
|
bold_wt_text = f"{BOLD_START}WM Theme:{BOLD_END}"
|
||
|
|
bold_dt_text = f"{BOLD_START}Theme:{BOLD_END}"
|
||
|
|
bold_ic_text = f"{BOLD_START}Icons:{BOLD_END}"
|
||
|
|
bold_gpu_text = f"{BOLD_START}GPU:{BOLD_END}"
|
||
|
|
|
||
|
|
|
||
|
|
# Print the OS name and architecture
|
||
|
|
print_user_and_hostname()
|
||
|
|
print(f"{bold_os_text} {os_full_information}")
|
||
|
|
print(f"{bold_host_text} {hostname_information}")
|
||
|
|
print(f"{bold_kernel_text} {kernel_information}")
|
||
|
|
print(f"{bold_uptime_text} {uptime_string}")
|
||
|
|
print(f"{bold_packages_text} {package_information}")
|
||
|
|
print(f"{bold_shell_text} {shell_information}")
|
||
|
|
print(f"{bold_resolution_text} {RESOLUTION_INFORMATION}")
|
||
|
|
print(f"{bold_de_text} {desktop_environment} {version_informtaion}")
|
||
|
|
print(f"{bold_wm_text} {window_manager_information}")
|
||
|
|
print(f"{bold_wt_text} {wm_theme}")
|
||
|
|
print(f"{bold_dt_text} {desktop_theme}")
|
||
|
|
print(f"{bold_ic_text} {ICON_THEME}")
|
||
|
|
print(f"{bold_term_text} {terminal_emulator}")
|
||
|
|
print(f"{bold_cpu_text} {cpu_information}")
|
||
|
|
print(f"{bold_gpu_text} {gpu_information}")
|
||
|
|
print(f"{bold_mem_text} {mem_information}")
|