oled_test.py

#!/usr/bin/env python3
"""
Simple 128x64 I2C OLED test for Raspberry Pi.

Expected device:
  - SSD1309 OLED
  - I2C address: 0x3C
  - Resolution: 128x64

Install dependencies on the Pi if needed:
  python3 -m pip install luma.oled pillow
"""

import time

from luma.core.interface.serial import i2c
from luma.core.render import canvas
from luma.oled.device import sh1106, ssd1306, ssd1309


WIDTH = 128
HEIGHT = 64
I2C_ADDRESS = 0x3C


def main():
    print("Opening I2C bus...")
    serial = i2c(port=1, address=I2C_ADDRESS)

    drivers = (
        ("SSD1309", ssd1309),
        ("SSD1306", ssd1306),
        ("SH1106", sh1106),
    )

    try:
        for name, driver in drivers:
            print(f"Trying {name} at 0x{I2C_ADDRESS:02X}...")
            display = driver(serial, width=WIDTH, height=HEIGHT)

            print("Flashing display...")
            display.clear()
            time.sleep(0.5)

            with canvas(display) as draw:
                draw.rectangle((0, 0, WIDTH - 1, HEIGHT - 1), fill=255)

            time.sleep(1)

            print(f"Showing {name} test image for 5 seconds...")
            with canvas(display) as draw:
                draw.rectangle((0, 0, WIDTH - 1, HEIGHT - 1), outline=255)
                draw.text((4, 4), "OLED Test", fill=255)
                draw.text((4, 20), f"{name} 128x64", fill=255)
                draw.text((4, 36), "I2C: 0x3C", fill=255)
                draw.text((4, 52), "Hello Pi Zero 2", fill=255)

            time.sleep(5)
            display.clear()

        print("Done trying drivers. Press Ctrl+C to exit.")
        while True:
            time.sleep(1)
    except KeyboardInterrupt:
        print("\nExiting.")


if __name__ == "__main__":
    main()