检查发送多少短信的Python方法

Question

我的check_mk安装中有一个python脚本，用于在紧急情况下为虚拟主机和服务发送电子邮件和SMS。 我想确保在一小时内发送不超过15条SMS / E-Mail，以避免在物理服务器无法访问或宕机时发送垃圾邮件，从而不会为每个VM和服务发送通知。 没有计数器文件，有没有办法做到这一点？ 也许正在运行的服务器？

我会做这样的事情：

import urllib
import urllib2
from os import environ


# parameters
hosts = ['192.168.1.10', '192.168.1.11']
login = 'user'
password = 'secret'
number = 'xxxxxxxxxxxx'

def message():
    if environ['NOTIFY_WHAT'] == 'HOST':
        msg = "%s %s: %s (%s)" % (
            environ['NOTIFY_NOTIFICATIONTYPE'],
            environ['NOTIFY_HOSTNAME'],
            environ['NOTIFY_HOSTOUTPUT'],
            environ['NOTIFY_SHORTDATETIME'])
    elif environ['NOTIFY_WHAT'] == 'SERVICE':
        msg = "%s %s: %s %s (%s)" % (
            environ['NOTIFY_NOTIFICATIONTYPE'],
            environ['NOTIFY_HOSTNAME'],
            environ['NOTIFY_SERVICEDESC'],
            environ['NOTIFY_SERVICEOUTPUT'],
            environ['NOTIFY_SHORTDATETIME'])
    else:
        msg = "Unknown: " + environ['NOTIFY_WHAT']
    return msg

def sendSMS(receiver, message):
    args = {'login': login, 'pass': password, 'to': receiver, 'message': message}
    enc_args = urllib.urlencode(args)
    sent = False
    error = ''

    for host in hosts:
        url = 'http://%s/index.php/http_api/send_sms?%s' % (host, enc_args)
        if not sent:
            try:
                result = urllib2.urlopen(url).read()
            except urllib2.HTTPError, e:
                print('sending sms via host %s failed: %s' % (host, str(e.code)))
            except urllib2.URLError, e:
                print('sending sms via host %s failed: %s' % (host, str(e.args)))
            else:
                if result.startswith('OK;'):
                    sent = True
                    print('SMS successfully sent via %s to %s.' % (host, receiver))
                    break
                else:
                    sent = False
                    print 'Sending SMS via host %s failed: %s' % (host, result.rstrip()))
    return

    message = message()
    sendSMS(number, message)

该脚本由check_mk读取并执行，它是通过WATO控制的。 所以我没有执行权。

非常感谢

Answer 1

您将需要能够统计给定时间内已发送多少短信。 您可以将sendSMS函数移动到一个简单的Web服务中，脚本将其发布到该Web服务中。 这样，它可以跟踪sendSMS被调用的速率并采取相应的措施。

另一种方法是将redis用作存储计数而不是文件的位置。