From 3e92aab2e787f13185dc99f6972f5c72e8029df6 Mon Sep 17 00:00:00 2001
From: Filip Strajnar <filip.strajnar@gmail.com>
Date: Sat, 26 Oct 2024 19:02:33 +0200
Subject: [PATCH] Created PciDevice class.

---
 src/pci_passthrough_assist/pci_device.py | 10 ++++++++++
 1 file changed, 10 insertions(+)
 create mode 100644 src/pci_passthrough_assist/pci_device.py

diff --git a/src/pci_passthrough_assist/pci_device.py b/src/pci_passthrough_assist/pci_device.py
new file mode 100644
index 0000000..2ea16d4
--- /dev/null
+++ b/src/pci_passthrough_assist/pci_device.py
@@ -0,0 +1,10 @@
+from pci_passthrough_assist.pci import driver_of_pci_device
+
+
+class PciDevice:
+
+    def __init__(self, device_id: str):
+        self.device_id = device_id
+
+    def driver_name(self) -> str:
+        return driver_of_pci_device(self.device_id)