树莓派 SAKS 扩展板实用应用 之 树莓派关机键

本文所介绍的内容基于树莓派瑞士军刀扩展板（SAKS），由于 SAKS 基于树莓派通用的 GPIO 设计，文中的原理和代码也适用于其他情况。

本文由树莓派实验室创作，遵循CC协议（署名-非商业性使用-禁止演绎-相同方式共享），欢迎树莓派爱好者们遵循协议内容转载传播。谢绝不遵守协议的转载和抄袭。

树莓派是精简版的微型电脑，连电源开关都精简掉了。要完成正常的关机和重启的操作，需要通过终端输入关机和重启命令才行。当树莓派没有配备显示器时，登录 SSH 关机更显得颇为麻烦。现在既然有了 SAKS 扩展板，为什么不用上面的轻触开关来实现一个关机和重启的按键功能呢？

本例程中涉及到的 SAKS 扩展板引脚编号如下：

我们将要实现的功能，是将 SAKS 扩展板的 K1 轻触开关改造成树莓派的关机和重启键，同时配以两个 LED 来指示操作状态。

轻触开关1次，进入重启倒计时，黄色 LED 闪烁，倒计时为10秒，倒计时结束后立刻重启。
在重启倒计时过程中再次轻触开关1次，由重启倒计时切换到关机倒计时，红色 LED 闪烁，倒计时结束后立刻关机。
在关机倒计时过程中，再次轻触开关1次，取消倒计时，LED 停止闪烁，系统恢复正常运行。

这样一个按键配上两个 LED 就能实现关机、重启、取消指令的操作了。实现代码如下：

#!/usr/bin/env python

# coding=utf-8

import RPi.GPIO as GPIO

import time

import os,sys

import signal



#定义关机键和关机状态指示灯的GPIO引脚

GPIO.setmode(GPIO.BCM)

pin_btn = 23

pin_led_reboot = 7

pin_led_halt = 8



#初始化SAKS上相应按键和LED的状态，按键内部上拉、LED不亮

GPIO.setup(pin_btn, GPIO.IN, pull_up_down = GPIO.PUD_UP)

GPIO.setup(pin_led_reboot, GPIO.OUT, initial = GPIO.HIGH)

GPIO.setup(pin_led_halt, GPIO.OUT, initial = GPIO.HIGH)



#初始化按下关机键的次数

press_times = 0

#按下关机键后等待并倒数10次

count_down = 10

led_on_reboot = 0

led_on_halt = 0



def onPress(channel):

    global press_times, count_down

    print('pressed')

    press_times += 1

    if press_times > 3:

        press_times = 1

    #重启模式

    if press_times == 1:

        GPIO.output(pin_led_reboot, 0)

        GPIO.output(pin_led_halt, 1)

        print('system will restart in %s' % (count_down))

    #关机模式

    elif press_times == 2:

        GPIO.output(pin_led_reboot, 1)

        GPIO.output(pin_led_halt, 0)

        print('system will halt in %s' % (count_down))

    #模式取消

    elif press_times == 3:

        GPIO.output(pin_led_reboot, 1)

        GPIO.output(pin_led_halt, 1)

        print 'cancel'

        count_down = 10



#设置按键检测，检测到按下时调用 onPress 函数

GPIO.add_event_detect(pin_btn, GPIO.FALLING, callback = onPress, bouncetime = 500)



try:

    while True:

    	#重启模式

        if press_times == 1:

            if count_down == 0:

                print "start restart"

                os.system("shutdown -r -t 5 now")

                sys.exit()

            led_on_reboot = not led_on_reboot

            #黄色 LED 闪烁

            GPIO.output(pin_led_reboot, led_on_reboot)

        #关机模式

        if press_times == 2:

            if count_down == 0:

	            print "start shutdown"

	            os.system("shutdown  -t 5 now")

	            sys.exit()

            led_on_halt = not led_on_halt

            #红色 LED 闪烁

            GPIO.output(pin_led_halt, led_on_halt)

 

        if press_times == 1 or press_times == 2:

            count_down -= 1

            print "%s second" % (count_down)

        time.sleep(1)

except KeyboardInterrupt:

    print('User press Ctrl+c, exit;')

finally:

    GPIO.cleanup()

接下来运行。

sudo python powerbutton.py

将这个脚本设为开机启动，编辑 rc.local：

vim /etc/rc.local

在 exit 0 所在行之前插入一行，文件路径需根据实际自行修改：

python /home/pi/powerbutton.py &

后面我们后面将开始介绍如何基于 SAKS 扩展板 DIY 各种有趣、实用的东西，欢迎持续关注！

本文属于《树莓派瑞士军刀扩展板（SAKS）DIY 教程》系列文章，查看系列文章目录，请访问：../../../swiss-army-knife-shield-for-raspberry-pi-diy-tutorials