A small python script for updating a small oled display with some stats (heavily influenced from Adafruits samples)

stats.py

#!/usr/bin/python
# Copyright (c) 2017 Adafruit Industries
# Author: Tony DiCola & James DeVito
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
import time

import Adafruit_GPIO.SPI as SPI
import Adafruit_SSD1306

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

import subprocess
import signal
import sys
import getopt
import collections
import itertools

def signal_handler(sig, frame):
        print('You pressed Ctrl+C!')
        # Clear display.
        disp.clear()
        disp.display()
        disp.command(Adafruit_SSD1306.SSD1306_DISPLAYOFF)
        sys.exit(0)

def main(script,argv):        
      
    signal.signal(signal.SIGINT, signal_handler)

    millis=500.0
    steps=1
    
    try:
        opts, args = getopt.getopt(argv,"hm:s:",["millis=","stepsize="])
    except getopt.GetoptError:
        print script+' -m <millis> -s <stepsize>'
        sys.exit(2)
    for opt, arg in opts:
        if opt == '-h':
            print script+' -m <millis> -s <stepsize>'
            sys.exit()
        elif opt in ("-m", "--millis"):
            millis = float(arg)
        elif opt in ("-s", "--stepsize"):
            steps = int(arg)
            
    # Initialize library.
    disp.begin()

    # Clear display.
    disp.clear()
    disp.display()

    # Create blank image for drawing.
    # Make sure to create image with mode '1' for 1-bit color.
    width = disp.width
    height = disp.height
    image = Image.new('1', (width, height))

    # Get drawing object to draw on image.
    draw = ImageDraw.Draw(image)

    # Draw a black filled box to clear the image.
    draw.rectangle((0,0,width,height), outline=0, fill=0)

    # Draw some shapes.
    # First define some constants to allow easy resizing of shapes.
    padding = -2
    top = padding-8
    bottom = height-padding
    # Move left to right keeping track of the current x position for drawing shapes.
    x = 0

    line=0

    # Load default font.
    font = ImageFont.load_default()

    # Alternatively load a TTF font.  Make sure the .ttf font file is in the same directory as the python script!
    # Some other nice fonts to try: http://www.dafont.com/bitmap.php
    # font = ImageFont.truetype('Minecraftia.ttf', 8)

    while True:

        top=padding;
        # Draw a black filled box to clear the image.
        draw.rectangle((0,0,width,height), outline=0, fill=0)

        # Shell scripts for system monitoring from here : https://unix.stackexchange.com/questions/119126/command-to-display-memory-usage-disk-usage-and-cpu-load
        cmd = "hostname -I | cut -d\' \' -f1"
        IP = subprocess.check_output(cmd, shell = True )
        cmd = "top -bn1 | grep load | awk '{printf \"CPU Load: %.2f\", $(NF-2)}'"
        CPU = subprocess.check_output(cmd, shell = True )
        cmd = "free -m | awk 'NR==2{printf \"Mem: %s/%sMB %.2f%%\", $3,$2,$3*100/$2 }'"
        MemUsage = subprocess.check_output(cmd, shell = True )
        cmd = "df -h | awk '$NF==\"/\"{printf \"Disk: %d/%dGB %s\", $3,$2,$5}'"
        Disk = subprocess.check_output(cmd, shell = True )
        cmd="ntpq -p |grep \"*\"|grep P406|wc -l"
        ppsused=subprocess.check_output(cmd, shell = True )
        cmd="uptime|cut -d \" \" -f 1,2"
        uptime=subprocess.check_output(cmd, shell = True )
        cmd="/opt/vc/bin/vcgencmd measure_temp|cut -d \"=\" -f 2"
        temp=subprocess.check_output(cmd, shell = True )
    
        # Write text.
        ppsstatus="fail"
        if(str.strip(ppsused)=="1"):
            ppsstatus="OK"
    
        stats=[]
        stats.append("IP: " + str(IP).rstrip())    
        stats.append(str(CPU).rstrip())    
        stats.append(str(MemUsage).rstrip())    
        stats.append(str(Disk).rstrip())    
        #stats.append("GPS+PPS "+ppsstatus)    
        stats.append("Up: " + str(uptime).rstrip())    
        stats.append("Temp : " + str(temp).rstrip())    
        #stats.append()    
        
        ypos=top
        
        print "GPS+PPS "+ppsstatus
        draw.text((x, ypos),"GPS+PPS "+ppsstatus ,  font=font, fill=255)
        ypos=ypos+8;
        
        for i in [0,1,2][:]:
            print(stats[(i+line*steps)%len(stats)])
            draw.text((x, ypos),stats[(i+line*steps)%len(stats)] ,  font=font, fill=255)
            ypos=ypos+8;
        print("--- "+str(line))

        # Display image.
        disp.image(image)
        #disp.set_contrast(1)
        disp.display()
        time.sleep(millis/1000)
        line=line+1
        if(line*steps>=len(stats)):
            line=0
        
        #disp.command(0xAE)

if __name__ == "__main__":
    # Raspberry Pi pin configuration:
    RST = None     # on the PiOLED this pin isnt used
    # Note the following are only used with SPI:
    DC = 23
    SPI_PORT = 0
    SPI_DEVICE = 0

    # Beaglebone Black pin configuration:
    # RST = 'P9_12'
    # Note the following are only used with SPI:
    # DC = 'P9_15'
    # SPI_PORT = 1
    # SPI_DEVICE = 0
    
    # 128x32 display with hardware I2C:
    disp = Adafruit_SSD1306.SSD1306_128_32(rst=RST)

    # 128x64 display with hardware I2C:
    # disp = Adafruit_SSD1306.SSD1306_128_64(rst=RST)

    # Note you can change the I2C address by passing an i2c_address parameter like:
    # disp = Adafruit_SSD1306.SSD1306_128_64(rst=RST, i2c_address=0x3C)

    # Alternatively you can specify an explicit I2C bus number, for example
    # with the 128x32 display you would use:
    # disp = Adafruit_SSD1306.SSD1306_128_32(rst=RST, i2c_bus=2)

    # 128x32 display with hardware SPI:
    # disp = Adafruit_SSD1306.SSD1306_128_32(rst=RST, dc=DC, spi=SPI.SpiDev(SPI_PORT, SPI_DEVICE, max_speed_hz=8000000))

    # 128x64 display with hardware SPI:
    # disp = Adafruit_SSD1306.SSD1306_128_64(rst=RST, dc=DC, spi=SPI.SpiDev(SPI_PORT, SPI_DEVICE, max_speed_hz=8000000))

    # Alternatively you can specify a software SPI implementation by providing
    # digital GPIO pin numbers for all the required display pins.  For example
    # on a Raspberry Pi with the 128x32 display you might use:
    # disp = Adafruit_SSD1306.SSD1306_128_32(rst=RST, dc=DC, sclk=18, din=25, cs=22)

    main(sys.argv[0],sys.argv[1:])