azalea/zshrc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[F] Fix path

Browse Source
This commit is contained in:
Azalea Gui
2024-10-07 16:48:31 -04:00
parent 5fe6d0ebc7
commit ae83089c12
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/bin/tmux-tool.py
+6 -2
View File
@@ -1,6 +1,8 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import os import os
from pathlib import Path
import re import re
from shutil import which
import subprocess import subprocess
import argparse import argparse
from datetime import datetime from datetime import datetime
@@ -58,18 +60,20 @@ def get_ip_address():
return None return None
def get_hardware_info(cpu=False, gpu=False): def get_hardware_info(cpu=False, gpu=False):
# Check if command exist
fastfetch = which('fastfetch') or Path(__file__).parent / 'fastfetch'
try: try:
cpu_str, gpu_str = '', '' cpu_str, gpu_str = '', ''
if cpu: if cpu:
# Fetch CPU info # Fetch CPU info
cpu_info = subprocess.run(['fastfetch', '--logo', 'none', '-s', 'CPU'], stdout=subprocess.PIPE) cpu_info = subprocess.run([fastfetch, '--logo', 'none', '-s', 'CPU'], stdout=subprocess.PIPE)
cpu_str = cpu_info.stdout.decode().strip() cpu_str = cpu_info.stdout.decode().strip()
# cpu_str = cpu_str.split(':', 1)[1].split('@', 1)[0].strip() # cpu_str = cpu_str.split(':', 1)[1].split('@', 1)[0].strip()
cpu_str = cpu_str.split(':', 1)[1].strip() cpu_str = cpu_str.split(':', 1)[1].strip()
if gpu: if gpu:
# Fetch GPU info # Fetch GPU info
gpu_info = subprocess.run(['fastfetch', '--logo', 'none', '-s', 'GPU'], stdout=subprocess.PIPE) gpu_info = subprocess.run([fastfetch, '--logo', 'none', '-s', 'GPU'], stdout=subprocess.PIPE)
gpu_str = gpu_info.stdout.decode().strip() gpu_str = gpu_info.stdout.decode().strip()
gpu_str = gpu_str.split(':', 1)[1].strip() gpu_str = gpu_str.split(':', 1)[1].strip()