#!/usr/bin/python3                 

# Version 2.0.1
# Original script by Stewart C. Russell via https://github.com/scruss/shutdown_button
# Modified by Steve Magnuson, AG7GN to control LED on different GPIO during Button
# press.
                                                                                                                                                                                
# -*- coding: utf-8 -*-
# example gpiozero code that could be used to have a reboot
#  and a shutdown function on one GPIO button
# scruss - 2017-10

use_button=26     # Button on use_button (BCM GPIO 26 by default)

from gpiozero import Button
from gpiozero import LED
from signal import pause
from subprocess import check_call

held_for=0.0
led=LED(24)			# LED on BCM GPIO 24

def rls():
        global held_for
        if (held_for > 5.0):
                check_call(['/sbin/poweroff'])
        elif (held_for > 2.0):
                check_call(['/sbin/reboot'])
        else:
        	held_for = 0.0

def hld():
        # callback for when button is held
        #  is called every hold_time seconds
        global held_for
        # need to use max() as held_time resets to zero on last callback
        held_for = max(held_for, button.held_time + button.hold_time)
        if (held_for > 5.0):
            led.off()
        elif (held_for > 2.0):
            led.on()
        else:
            led.off()

button=Button(use_button, hold_time=1.0, hold_repeat=True)
button.when_held = hld
button.when_released = rls

pause() # wait forever