Urgh

esp32-lora/screen.py

@@ -4,26 +4,18 @@ import time
 
 class Screen:
 
-    sda = None
-    scl = None
-    rst = None
-    oled = None
-
     def __init__(self):
+        # OLED pins on Heltec V3
         self.sda = Pin(17)
         self.scl = Pin(18)
         self.rst = Pin(21, Pin.OUT)
 
-        self.__reset_oled()  # reset first
+        self.__reset_oled()
 
-        self.oled = SSD1306_I2C(
-            128,
-            64,
-            SoftI2C(scl=self.scl, sda=self.sda, freq=100000),
-            addr=0x3C
-        )
+        self.i2c = SoftI2C(scl=self.scl, sda=self.sda, freq=100000)
+        self.oled = SSD1306_I2C(128, 64, self.i2c, addr=0x3C)
 
-        self.reset()  # clear screen
+        self.clear()
 
     def __reset_oled(self):
         self.rst.value(0)
@@ -31,10 +23,11 @@ class Screen:
         self.rst.value(1)
         time.sleep_ms(150)
 
-    def reset(self):
+    def clear(self):
         self.oled.fill(0)
         self.oled.show()
 
-    def display(self, message, col, row):
+    def display(self, message, col=0, row=0):
+        self.clear()
         self.oled.text(message, col, row)
         self.oled.show()