allow all pushsrvs, fix boot msg timing

This commit is contained in:
2020-02-04 18:21:59 -05:00
parent de283cb1a9
commit 328ce83058
+18 -14
View File
@@ -199,12 +199,12 @@ def email(s, msg):
return ret
def pushmsg(msg):
if pushsrv == "pushover":
return pushover(msg)
elif pushsrv == "mattermost":
if pushsrv in ["all", "pushover"]:
pushover(msg)
if pushsrv in ["all","mattermost"]:
return pushmattermost(msg)
else:
print(msg)
if pushsrv in ["all"]:
print("notice:", msg)
def pushover(msg):
@@ -225,7 +225,7 @@ CHANNEL = "Monitoring"
TOKEN = "rxz6b3886iygxnhbzpmgbsrocy"
HOST = "192.168.10.101"
ICON = "https://in-transit.ca/HeartBeat.png"
USERNAME = "nagios"
USERNAME = "admin"
def pushmattermost(msg):
@@ -445,9 +445,17 @@ def readsock(sock):
message = msg.get('msg', None)
boot = msg.get('boot', 0)
conn, res = host.conndata(cid, addr[0], rtt, now)
if res:
log(name, res)
if name in watchhosts:
email("address change", "%s %s" % (host.name, res))
pushmsg("%s %s" % (host.name, res))
if boot:
log(name, "booted")
if name in watchhosts:
m = "%s booted" % (host.name)
email("booted", m)
pushmsg(m)
if message:
@@ -456,12 +464,6 @@ def readsock(sock):
email("msg", message)
pushmsg(message)
conn, res = host.conndata(cid, addr[0], rtt, now)
if res:
log(name, res)
if name in watchhosts:
email("address change", "%s %s" % (host.name, res))
pushmsg("%s %s" % (host.name, res))
if conn.getstate() != hbdclass.Connection.up: # XXX and interval > 0:
lasts = conn.state
@@ -799,7 +801,7 @@ def pickleit():
#
# Main
#
PUSHSRVS = [ "pushover", "mattermost" ]
PUSHSRVS = ["all", "pushover", "mattermost" ]
helpflag = False
forground = False
pushsrv = "pushover" # mattermost
@@ -837,7 +839,7 @@ for o, a in optlist:
DEBUG += 1
cmdargs += [o]
if pushsrv == "mattermost":
if pushsrv in ["all", "mattermost"]:
try:
from mattermostdriver import Driver
except:
@@ -933,6 +935,8 @@ if f:
drophosts = a
elif o == 'nsupdate_bin':
nsupdate_bin = a
elif o == 'pushsrv':
pushsrv = a
f.close()
if len(args) != 0: