import RPi.GPIO as GPIO
import time
import sys
import os
import signal

LED_PIN = 12
BUTTON_PIN = 26
button_flag = False

def main():
    GPIO.setwarnings(False)
    GPIO.setmode(GPIO.BCM)
    GPIO.setup(LED_PIN, GPIO.OUT)
    GPIO.setup(BUTTON_PIN, GPIO.IN)
    GPIO.output(LED_PIN, GPIO.LOW)
    # 起動したサインとしてLEDを３回点滅させる
    blink_led(3, 0.1)
    # 入力待ちのサインとしてLEDを点灯させる
    GPIO.output(LED_PIN, GPIO.HIGH)
    GPIO.add_event_detect(BUTTON_PIN, GPIO.FALLING, callback=button_pushed, bouncetime=300) 
    # killされた場合も終了処理（finally節）を行う
    signal.signal(signal.SIGTERM, sigterm_handler)
    # wait_for_edgeではkillしてもイベントが発生するまでフリーズするので、
    # ループで待機するようにする
    global button_flag
    try:
        while True:
            time.sleep(0.1)
            if button_flag:
                choose_action()
                button_flag = False
    except KeyboardInterrupt:
        pass
    finally:
        # 終了のサインとしてLEDを消灯させる
        GPIO.output(LED_PIN, GPIO.LOW)
        GPIO.remove_event_detect(BUTTON_PIN)
        GPIO.cleanup()   

# LEDを指定の回数、指定の間隔で点滅させる
def blink_led(times, wait):
    for i in range(times):
        GPIO.output(LED_PIN, GPIO.HIGH)
        time.sleep(wait)
        GPIO.output(LED_PIN, GPIO.LOW)
        time.sleep(wait)

# コールバック関数で時間のかかる処理をすると２回実行されることがあるので、
# グローバル変数のフラグの変更のみ行う。
def button_pushed(gpio_pin):
    global button_flag
    button_flag = True

# トグルボタンを押した長さでreboot、shutdownの処理を選ぶ
def choose_action():
    GPIO.output(LED_PIN, GPIO.LOW)
    time.sleep(0.2)
    count = 0
    while True:
        blink_led(1, 0.5)
        count += 1
        if GPIO.input(BUTTON_PIN) == GPIO.HIGH:
            # ３回点滅した場合は再起動
            if count == 3:
                blink_led(3, 0.1)
                os.system("sudo reboot")
                sys.exit()
            # ５回点滅した場合はシャットダウン
            if count == 5:
                blink_led(5, 0.1)
                os.system("sudo shutdown -h now")
                sys.exit()
            # その他の場合は５回すばやくLEDを点滅させる
            blink_led(5, 0.1)
            GPIO.output(LED_PIN, GPIO.HIGH)
            break

# killされた場合も終了処理（finally節）を行う
def sigterm_handler(signal_number, stack_frame):
    sys.exit()

if __name__ == "__main__":
    main()

[Unit]
Description=for Reboot and Shutdown board

[Service]
ExecStart=/usr/bin/python3 /home/pi/reboot_shutdown.py

[Install]
WantedBy = multi-user.target