Raspi - OLED Infodisplay mit Multifunktionsswitch

• sudo python3 /home/pi/sdisp.py &

• sudo kill <oben notierte Zahl>
  

• sudo mkdir /var/local/lib
• sudo mkdir /var/local/lib/startup

• sudo mv /home/pi/sdisp.py /var/local/lib/startup

• sudo nano /etc/rc.local

import time
import subprocess
from board import SCL, SDA
import busio
import RPi.GPIO as GPIO
from PIL import Image, ImageDraw, ImageFont
import adafruit_ssd1306
import psutil

GPIO.setmode(GPIO.BCM)
GPIO.setup(17, GPIO.OUT)
GPIO.setup(27, GPIO.OUT)
GPIO.setup(22, GPIO.OUT)
GPIO.setup(10, GPIO.IN)

# Raspberry Pi pin configuration:

INFO_BTN = 10
LEDR = 17
LEDB = 22
LEDG = 27

# Usercount

usercnt = 0

# Timer for Display timeout

disp_timer = 0
DISP_TIMEOUT = 15

# Menu Variables

menu_state = 0 # 0 = Info; 1 = Reboot; 2 = Restart
menu_timer = 0
REBOOT_TIMEOUT = 10
SHUTDOWN_TIMEOUT = 15
do_reboot = 0
do_shutdown = 0

# Create the I2C interface.

i2c = busio.I2C(SCL, SDA)

# Create the SSD1306 OLED class.
# The first two parameters are the pixel width and pixel height.  Change these
# to the right size for your display!

disp = adafruit_ssd1306.SSD1306_I2C(128, 32, i2c)

# Clear display.

disp.rotation = 0
disp.fill(0)
disp.show()

# 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
bottom = height - padding

# Move left to right keeping track of the current x position for drawing shapes.

x = 0
te = 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('/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf', 9)

GPIO.output(LEDR, GPIO.LOW)
GPIO.output(LEDG, GPIO.HIGH)
GPIO.output(LEDB, GPIO.LOW)

# Startup Info

draw.rectangle((0,0,width,height), outline=0, fill=0)
draw.text((x, top),    "--------------------", font=font, fill=255)
draw.text((x, top+12), "  Raspberry booted  ", font=font, fill=255)
draw.text((x, top+24), "--------------------", font=font, fill=255)
disp.image(image)
disp.show()

time.sleep(5)

draw.rectangle((0,0,width,height), outline=0, fill=0)
disp.image(image)
disp.show()

while True:

  cmd = "who | cut  -f1 -d' ' | sort -u | wc -l"
  usercnt = subprocess.check_output(cmd, shell=True).decode("utf-8")
  usercnt = int(usercnt)
  cmd = "ps -eaf | grep nginx | wc -l"
  procok = subprocess.check_output(cmd, shell=True).decode("utf-8")
  procok = int(procok)

  if usercnt == 0 and procok > 2:
    GPIO.output(LEDR, GPIO.LOW)
    GPIO.output(LEDG, GPIO.HIGH)
    GPIO.output(LEDB, GPIO.LOW)

  elif usercnt > 0 and procok > 2:
    GPIO.output(LEDR, GPIO.LOW)
    GPIO.output(LEDG, GPIO.LOW)
    GPIO.output(LEDB, GPIO.HIGH)

  elif procok < 3:
    GPIO.output(LEDR, GPIO.HIGH)
    GPIO.output(LEDG, GPIO.LOW)
    GPIO.output(LEDB, GPIO.LOW)

  draw.rectangle((0, 0, width, height), outline=0, fill=0)

  # Info Button pressed?

  if GPIO.input(INFO_BTN) == 0:

     if menu_timer >= REBOOT_TIMEOUT:
       menu_state = 1

     if menu_timer >= SHUTDOWN_TIMEOUT:
       menu_state = 2

     disp_timer = DISP_TIMEOUT
     menu_timer = menu_timer+1

  elif disp_timer == 0:
     disp.image(image)
     disp.show()

  if disp_timer > 0:

    if menu_state == 0:
      cmd = "hostname"
      HOSTNAME =  subprocess.check_output(cmd, shell = True)
      cmd = "hostname -I | cut -d\' \' -f1"
      IP = subprocess.check_output(cmd, shell = True )
      CPU = "{:3.0f}".format(psutil.cpu_percent())
      svmem = psutil.virtual_memory()
      MemUsage = "{:2.0f}".format(svmem.percent)

      draw.text((x, top),       "NAME: " + HOSTNAME.decode('UTF-8'), font=font, fill=255)
      draw.text((x, top+12),    "IP  : " + IP.decode('UTF-8'),  font=font, fill=255)
      draw.text((x, top+24),    "CPU : " + CPU + "% | MEM: " + MemUsage + "%", font=font, fill=255)
      disp_timer =  disp_timer-1

      if GPIO.input(INFO_BTN) == 1:
        menu_timer = 0

    if menu_state == 1:

      if GPIO.input(INFO_BTN) == 1:
        do_reboot = 1

        draw.text((x, top+12), "Performing Reboot...", font=font, fill=255)
        disp.image(image)
        disp.show()

        time.sleep(3)

        draw.rectangle((0,0,width,height), outline=0, fill=0)

      else:

        draw.text((x, top),    ".......Reboot......."     , font=font, fill=255)
        draw.text((x, top+12), "   Release Button   "   , font=font, fill=255)
        draw.text((x, top+24), "      To Reboot     "    , font=font, fill=255)

    if menu_state == 2:

        if GPIO.input(INFO_BTN) == 1:
          do_shutdown = 1

          draw.text((x, top+12), "Shutting down.......", font=font, fill=255)
          disp.image(image)
          disp.show()

          time.sleep(3)

          draw.rectangle((0,0,width,height), outline=0, fill=0)

        else:

          draw.text((x, top),    "......Shutdown......"     , font=font, fill=255)
          draw.text((x, top+12), "   Release Button   "   , font=font, fill=255)
          draw.text((x, top+24), "    To Shutdown     "      , font=font, fill=255)

    disp.image(image)
    disp.show()

    if do_reboot == 1:
      cmd = "sudo reboot now"
      subprocess.Popen(cmd, shell = True)

    if do_shutdown == 1:
      cmd = "sudo shutdown now"
      subprocess.Popen(cmd, shell = True)

    time.sleep(1)