当前位置: 首页 > 教程 > 心得 > Python 3脚本实现断网关机

Python 3脚本实现断网关机

发布于:2021-4-23 心得 0条评论 8,547 views
本站提供Linux服务器运维,自动化脚本编写等服务,如有需要请联系博主微信:xiaozme

家里小主机和NAS都接了UPS备用电源,威联通NAS使用UPS USB通讯接口默认支持断电正常关机。但UPS USB通讯接口只有一个,NAS占用通讯接口后小主机就无法实现断电自动正常关机,于是换一个思路,改用断网关机方案。

实现原理

家里一旦断电后,路由器便停止工作,网络中断。由于小主机是接到UPS上的,断电后依然在运行,但USP撑不了太长时间。为了保障断网正常关机,于是使用Python脚本 + crontab定时任务定时检测是否断网,如果断网则执行关机命令。

脚本内容

将下面脚本保存为auto-shutdown.py

#!/usr/bin/python3
import requests
import os

#修改文件
def err_num(num):
    #打开一个文件只用于写入。如果该文件已存在则打开文件,并从开头开始编辑,即原有内容会被删除。如果该文件不存在,创建新文件。
    fo = open("/tmp/err_num","w")
    #转换为字符串
    num = str(num)
    #fo = open("D:/temp/123.txt","w")
    fo.write(num)
    #关闭文件
    fo.close()

#获取输入参数
try:
    #请求百度
    r = requests.get('https://www.baidu.com/',timeout=10)
    code = r.status_code
except:
    code = -1
    pass

if( code >= 200 ):
    err_num('0')
else:
    #读取当前错误次数
    fo = open("/tmp/err_num","r")
    num = fo.read()
    #重置次数为0
    if ( num == '' ):
        num = 0
    fo.close()
    #当前错误次数+1
    num = int(num) + 1
    #print(num)

    err_num(num)
    if( num >= 5 ):
        #执行关机操作
        os.system('/usr/bin/sync && /usr/sbin/shutdown -h now')
  • 访问百度:https://www.baidu.com/ 判断是否存在网络
  • 如果连续超过5次均失败,则执行关机操作

定时任务

添加crontab -e定时任务,2分钟检测一次:

*/2 * * * * /root/code/python/auto-shutdown.py >> /dev/null

结合上面的脚本,平均2分钟执行一次,连续5次失败进行关机,那么也就意味着断网10分钟就执行关机,可根据实际情况来修改crontab检测频率。其实利用shell脚本也能实现,只不过Python 3写起来要方便一些。


发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注