parse verbose info from gpu name

pennybelle · Oct 4, 2024 · 4df038b · 4df038b
1 parent 802bbda
commit 4df038b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/pbfetch/parse/gpu.py b/src/pbfetch/parse/gpu.py
@@ -1,4 +1,5 @@
 from subprocess import Popen, PIPE
+from re import sub
 # import nvidia_smi
 
 
@@ -30,7 +31,7 @@ def parse_gpu():
         if gpu_name:
             gpu_name = gpu_name.split(": ")[1]
             gpu_name = gpu_name.strip()
-            return gpu_name
+            return sub(r"\(.+\)$", "", gpu_name.strip())
 
     except Exception:
         pass
@@ -45,7 +46,7 @@ def parse_gpu():
         gpu_name = gpu_name.replace("OpenGL compatibility profile renderer: ", "")
         # print(gpu_name)
 
-        return gpu_name
+        return sub(r"\(.+\)$", "", gpu_name.strip())
 
     except Exception:
         pass
@@ -63,7 +64,7 @@ def parse_gpu():
         gpu_name = gpu_name[2 : len(gpu_name) - 3]
         gpu_name = gpu_name.split("VGA compatible controller:")[1].strip()
 
-        return gpu_name
+        return sub(r"\(.+\)$", "", gpu_name.strip())
 
     except Exception as e:
         print(f"GPU Parse Error: {e}")