Argon40-ArgonOne-Script/source/scripts/argonpowerbutton-libgpiod.py

# For Libreelec/Lakka, note that we need to add system paths
# import sys
# sys.path.append('/storage/.kodi/addons/virtual.rpi-tools/lib')
import gpiod
import os
import time

# Debug Logger
def argonpowerbutton_debuglog(typestr, logstr):
	try:
		DEBUGFILE="/dev/shm/argononegpiodebuglog.txt"
		tmpstrpadding = "                      "

		with open(DEBUGFILE, "a") as txt_file:
			txt_file.write("["+time.asctime(time.localtime(time.time()))+"] "+typestr.upper()+" "+logstr.strip().replace("\n","\n"+tmpstrpadding)+"\n")
	except:
		pass


# This function is the thread that monitors activity in our shutdown pin
# The pulse width is measured, and the corresponding shell command will be issued

def argonpowerbutton_getconfigval(keyname, datatype="int"):
	keyname = keyname.lower()
	fname = "/etc/argononeupd.conf"
	try:
		with open(fname, "r") as fp:
			for curline in fp:
				if not curline:
					continue
				tmpline = curline.replace(" ", "").replace("\t", "")
				if not tmpline:
					continue
				if tmpline[0] == "#":
					continue
				tmppair = tmpline.split("=")
				if len(tmppair) != 2:
					continue

				tmpvar = tmppair[0].lower()
				if tmpvar != keyname:
					continue

				try:
					if datatype == "int":
						return int(tmppair[1])
					elif datatype == "float":
						return float(tmppair[1])
					return tmppair[1]
				except:
					continue
	except:
		pass
	if datatype == "int":
		return -1
	elif datatype == "float":
		return -1
	return ""

def argonpowerbutton_monitorlid(writeq):
	try:
		argonpowerbutton_debuglog("lid-monitor", "Starting")
		monitormode = True
		# 0 - Lid is closed, 1 - Lid is open
		# Pin Assignments
		LINE_LIDMONITOR=27
		try:
			# Pi5 mapping
			chip = gpiod.Chip('4')
		except Exception as gpioerr:
			# Old mapping
			chip = gpiod.Chip('0')

		lineobj = chip.get_line(LINE_LIDMONITOR)
		lineobj.request(consumer="argon", type=gpiod.LINE_REQ_EV_BOTH_EDGES, flags=gpiod.LINE_REQ_FLAG_BIAS_PULL_UP)
		while monitormode == True:
			hasevent = lineobj.event_wait(10)
			if hasevent:
				eventdata = lineobj.event_read()
				if eventdata.type == gpiod.LineEvent.FALLING_EDGE:
					targetsecs = argonpowerbutton_getconfigval("lidshutdownsecs")
					if targetsecs > 0:
						argonpowerbutton_debuglog("lid-monitor", "Close Detect; Wait for :"+str(targetsecs))
					else:
						argonpowerbutton_debuglog("lid-monitor", "Close Detected; Do nothing")
					# Time pulse data
					time.sleep(1)
					pulsetimesec = 1
					while lineobj.get_value() == 0:
						if targetsecs > 0:
							if pulsetimesec >= targetsecs:
								argonpowerbutton_debuglog("lid-monitor", "Target Reached, shutting down")
								monitormode = False
								os.system("shutdown now -h")
								break

						time.sleep(1)
						pulsetimesec += 1
					argonpowerbutton_debuglog("lid-monitor", "Open Detected")

		lineobj.release()
		chip.close()
	except Exception as liderror:
		try:
			argonpowerbutton_debuglog("lid-monitor-error", str(liderror))
		except:
			argonpowerbutton_debuglog("lid-monitor-error", "Error aborting")
		#pass

def argonpowerbutton_monitor(writeq):

	try:
		# Reference https://github.com/brgl/libgpiod/blob/master/bindings/python/examples/gpiomon.py

		# Pin Assignments
		LINE_SHUTDOWN=4
		try:
			# Pi5 mapping
			chip = gpiod.Chip('4')
		except Exception as gpioerr:
			# Old mapping
			chip = gpiod.Chip('0')

		lineobj = chip.get_line(LINE_SHUTDOWN)
		lineobj.request(consumer="argon", type=gpiod.LINE_REQ_EV_BOTH_EDGES)
		while True:
			hasevent = lineobj.event_wait(10)
			if hasevent:
				pulsetime = 0
				eventdata = lineobj.event_read()
				if eventdata.type == gpiod.LineEvent.RISING_EDGE:
					# Time pulse data
					while lineobj.get_value() == 1:
						time.sleep(0.01)
						pulsetime += 1

					if pulsetime >=2 and pulsetime <=3:
						# Testing
						#writeq.put("OLEDSWITCH")
						writeq.put("OLEDSTOP")
						os.system("reboot")
						break
					elif pulsetime >=4 and pulsetime <=5:
						writeq.put("OLEDSTOP")
						os.system("shutdown now -h")
						break
					elif pulsetime >=6 and pulsetime <=7:
						writeq.put("OLEDSWITCH")
		lineobj.release()
		chip.close()
	except Exception:
		writeq.put("ERROR")


def argonpowerbutton_monitorswitch(writeq):

	try:
		# Reference https://github.com/brgl/libgpiod/blob/master/bindings/python/examples/gpiomon.py

		# Pin Assignments
		LINE_SHUTDOWN=4
		try:
			# Pi5 mapping
			chip = gpiod.Chip('4')
		except Exception as gpioerr:
			# Old mapping
			chip = gpiod.Chip('0')

		lineobj = chip.get_line(LINE_SHUTDOWN)
		lineobj.request(consumer="argon", type=gpiod.LINE_REQ_EV_BOTH_EDGES)
		while True:
			hasevent = lineobj.event_wait(10)
			if hasevent:
				pulsetime = 0
				eventdata = lineobj.event_read()
				if eventdata.type == gpiod.LineEvent.RISING_EDGE:
					# Time pulse data
					while lineobj.get_value() == 1:
						time.sleep(0.01)
						pulsetime += 1

					if pulsetime >= 10:
						writeq.put("OLEDSWITCH")
		lineobj.release()
		chip.close()
	except Exception:
		writeq.put("ERROR")