拿来就用的脚本案例（三）

大家好，我是杰哥（鸽）。

之前给大家分享了《7 个非常实用的 Shell 拿来就用脚本实例！》和《超硬核！11 个非常实用的 Python 和 Shell 拿来就用脚本实例！》两篇文章。

得到了多位读者朋友的留言说：简直太实用了。

最近工作中，又碰到了几个实际场景并运用了脚本，大致包括：Zabbix 自动发现监控 URL、阿里云短信接口、每周删除 Elasticsearch X 个月前的索引，具体如下：

一、Zabbix自动发现监控URL

此脚本用于监控 URL，自动发现文件或其他存储的 URL。

1.1 URL自动发现脚本

from os.path import abspath, dirname, join
import json
import sys


reload(sys)
sys.setdefaultencoding('utf-8')

# URL存储文件
# 每一行格式为(可写多行)：应用名称 URL地址 注释
URL_FILE_PATH = join(dirname(abspath(__file__)), 'web_url_list')

# 输出结果，zabbix官方定义的格式
data = {'data': []}

with open(URL_FILE_PATH, 'r') as read_f:
    # 读取URL文件，并进行分割，组成官网定义数据
    for line in read_f:
        if line:
           name = line.split()[0]
           url = line.split()[1]
           description = line.split()[2]
           data['data'].append({"{#NAME}": name, "{#URL}": url, "{#DESCRIPTION}": description})

print json.dumps(data)

1.2 URL测试脚本

from sys import argv
import requests

# 获取URL状态
def get_url(url, timeout=5):
    # 多数情况下返回200即为访问成功，提前测试好自己的URL访问返回状态码
    try:
        request_obj = requests.get(url, timeout=timeout)
        status_code = request_obj.status_code
    except Exception as e:
        status_code = 99999

    print(status_code)


url = argv[1]
get_url(url)

1.3 zabbix 配置自定义 key

UserParameter=discovery_web_url, /bin/python /opt/zabbix_scripts/web_url/url_discovery.py 
UserParameter=url_get[*], /bin/python /opt/zabbix_scripts/web_url/url_get_status.py $1

1.4 zabbix WEB 端配置

• 配置自动发现

• 添加监控项、触发器原型

二、阿里云短信接口

根据官网样例配置短信发送

2.1 阿里云配置

• 签名配置：控制台-短信服务-国内消息-签名配置

• 模板配置：控制台-短信服务-国内消息-模板配置

自己定义好变量，如果告警信息过长（默认是35个字符），提交工单解除限制。

• AccessKey配置：头像-AccessKey管理

2.2 发送脚本

#coding=utf-8

   from aliyunsdkcore.client import AcsClient
   from aliyunsdkcore.request import CommonRequest
   from sys import argv


   LOG_FILE_PATH = '/tmp/zabbix-sms.log'


   def sendSms(phone_numbers, subject, alert_message):
       # AccessKet
       client = AcsClient('AccessKey ID', 'Secret')
       alert_message = subject + alert_message

       # 官方定义的格式，具体含义没有研究
       request = CommonRequest()
       request.set_accept_format('json')
       request.set_domain('dysmsapi.aliyuncs.com')
       request.set_method('POST')
       request.set_protocol_type('https') # https | http
       request.set_version('2017-05-25')
       request.set_action_name('SendSms')

       request.add_query_param('PhoneNumbers', phone_numbers)
       # 签名名称
       request.add_query_param('SignName', "XXXX")
       # 模板code
       request.add_query_param('TemplateCode', "XXXX")
       # 传递的参数
       request.add_query_param('TemplateParam', "{\"alert_message\":\"%s\"}" % alert_message)

       response = client.do_action(request)
       return response

   def writeLog(message, response, log_file_path):
       with open(log_file_path, 'a') as a_file:
           a_file.write(message + ' | ' + response)

   if __name__ == '__main__':
       phone_numbers = argv[1]
       subject = argv[2]
       alert_message = argv[3]
       # 手机号、主题、告警信息
       # 由zabbix端传递过来的
       res = sendSms('130XXXXXXXX', subject, alert_message)
       writeLog(subject + alert_message, res, LOG_FILE_PATH)

2.3 zabbix WEB 端配置

• 报警媒介

• 用户配置报警媒介

• 动作中应用即可

三、每周删除Elasticsearch X个月前的索引

索引格式示例：index-2021.11.21

#!/bin/bash
# Filename   :  delete_es_indices.sh
# Date       :  2021/11/21
# Author     :  xxx 
# Email      :  xxx
# Crontab    :  10 00 * * 6 /bin/bash $BASE_PATH/delete_es_indices.sh >/dev/null 2>&1
# Notes      :  将脚本加入crontab中，每周六0点10分执行
# Description:  每周删除Elasticsearch X个月前的索引，索引格式示例：index-2021.04.11


check_args(){
    if [ "$#" -ne 2 ];then
        echo "Usage: $0 [ES_URL] [DEL DAYS]"
        echo "ES_URL示例：http://1.1.1.1:9200，DEL DAYS：表示要删除多少天前的索引。"
        exit 1
    fi
}


# 删除索引函数
DeleteIndices(){
    indices="$1"
    time="$(echo $indices | awk -F'-' '{print $NF}')"
    delete_time="$(date -d "-${DELETE_DAYS_AGO} day" +'%Y-%m-%d')"

    # 过滤时间是不是 2021.04.11 格式的，如果不是就退出函数
    if ! echo "$time" | egrep -o "[0-9]{4}\.[0-9]{2}\.[0-9]{2}" &>/dev/null;then
        return   
    fi

    # 转换为unix好时间，date识别不了2021.11.21日志，所以转换为2021-11-21
    format_time=$(date -d "$(echo $time | tr '.' '-')" +'%s')
    format_delete_time=$(date -d "$delete_time" +'%s')

    # 如果索引时间小于要删除的时间并且索引名字不是monitor开头的，则删除
    if [[ "$format_time" -lt "$format_delete_time" && ! "$indices" =~ ^monitor.* ]];then
        curl -XDELETE "$ES_IP/$indices"
    fi
}

main(){
    check_args $@
    # ES URL
    ES_IP="$1"
    # 删除多少天前的索引
    DELETE_DAYS_AGO="$2"

    curl $ES_IP/_cat/indices | awk '{print $3}' | while read line
    do
        DeleteIndices $line
    done
}

main $@

以上，就是今天分享的全部内容了。

希望大家通过这些案例能够学以致用，结合自身的实际场景进行运用，从而提高自己的工作效率。

如果你有更多脚本实例，也欢迎大家留言分享或通过本文进行留言说说你具体的脚本实例需求，如果实例过多的话，下次杰哥在整一篇合集脚本文章实例来跟大家分享。