Keep track of displays and buses in init

[hackapet/Adafruit_Blinka_Displayio.git] / displayio / _i2cdisplay.py

diff --git a/displayio/_i2cdisplay.py b/displayio/_i2cdisplay.py
index 42f85ecac625e38f15885d392293c537db8e2acc..b5de381edced2f74620a106204a2ffa6352bdfd8 100644 (file)
--- a/displayio/_i2cdisplay.py
+++ b/displayio/_i2cdisplay.py
@@ -53,6 +53,14 @@ class I2CDisplay:
         self._i2c = i2c_bus
         self._dev_addr = device_address
 
+    def __new__(cls, *args, **kwargs):
+        from . import (  # pylint: disable=import-outside-toplevel, cyclic-import
+            allocate_display_bus,
+        )
+
+        allocate_display_bus(cls)
+        return super().__new__(cls)
+
     def _release(self):
         self.reset()
         self._i2c.deinit()