From 548527c302d9c4096d6e0dd233774904ca997b73 Mon Sep 17 00:00:00 2001
From: Filip Strajnar <filip.strajnar@gmail.com>
Date: Sat, 26 Oct 2024 18:43:56 +0200
Subject: [PATCH] Added a function that gets name of PCI device's driver.

---
 src/pci_passthrough_assist/pci.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/pci_passthrough_assist/pci.py b/src/pci_passthrough_assist/pci.py
index 68b8780..259f8e0 100644
--- a/src/pci_passthrough_assist/pci.py
+++ b/src/pci_passthrough_assist/pci.py
@@ -1,4 +1,5 @@
 from os import listdir
+from os.path import realpath, basename
 
 
 def all_pci_device_ids() -> list[str]:
@@ -7,3 +8,9 @@ def all_pci_device_ids() -> list[str]:
 
 def all_pci_driver_ids() -> list[str]:
     return listdir("/sys/bus/pci/drivers")
+
+
+def driver_of_pci_device(pci_device_id: str) -> str:
+    driver_directory: str = realpath(
+        f"/sys/bus/pci/devices/{pci_device_id}/driver")
+    return basename(driver_directory)