simple script to display either the arguments or the first lines of whatever is being piped into it to an SSD1306 driven oled screen

display.py

#!/usr/bin/python3
import sys
import Adafruit_GPIO.SPI as SPI
import Adafruit_SSD1306

from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont

disp = Adafruit_SSD1306.SSD1306_128_32(rst=None)
disp.begin()
#disp.clear()

if len(sys.argv)>1 :
    lines = [ ' '.join(sys.argv[1:]) ]
elif not sys.stdin.isatty() :
    lines = sys.stdin.readlines()
    if len(lines)>3 : lines = lines[:3]
else:
    lines = None

if lines:
    width = disp.width
    height = disp.height
    image = Image.new('1', (width, height))
    draw = ImageDraw.Draw(image)
    font = ImageFont.load_default()
    for y, line in enumerate(lines):
        draw.text(
            (0, y*8+2),
            line,
            font=font, fill=255
        )
    disp.image(image)

disp.display()