From 6c8f248e3edf538ba1d9298c89919ceb08aa18f3 Mon Sep 17 00:00:00 2001
From: Filip Strajnar <filip.strajnar@gmail.com>
Date: Sat, 26 Oct 2024 21:37:10 +0200
Subject: [PATCH] Added set_driver_override method.

---
 src/pci_passthrough_assist/pci_device.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/pci_passthrough_assist/pci_device.py b/src/pci_passthrough_assist/pci_device.py
index e5539a2..0d4f414 100644
--- a/src/pci_passthrough_assist/pci_device.py
+++ b/src/pci_passthrough_assist/pci_device.py
@@ -29,6 +29,11 @@ class PciDevice:
                   "w") as device_driver:
             device_driver.write(self.device_id)
 
+    def set_driver_override(self, reserved_for_driver: str):
+        with open(f"/sys/bus/pci/devices/{self.device_id}/driver_override",
+                  "w") as driver_override:
+            driver_override.write(reserved_for_driver)
+
     def bind_to_driver(self, driver_to_bind: str, unbind_first: bool = True):
         if unbind_first:
             self.unbind_driver()