157 lines
4.2 KiB
Python
157 lines
4.2 KiB
Python
#!/usr/bin/python3
|
|
|
|
import bson
|
|
import re
|
|
import smtplib
|
|
from email.mime.text import MIMEText
|
|
|
|
def format_query(query_usr):
|
|
query_list = []
|
|
for key, value in query_usr.items():
|
|
if key == "hostname":
|
|
qtag = ""
|
|
else:
|
|
qtag = "%s:" % key
|
|
query_list.append("%s%s" % (qtag, value))
|
|
return " ".join(query_list)
|
|
|
|
allowed_filters = (
|
|
'status',
|
|
'hood',
|
|
'nickname',
|
|
'hardware',
|
|
'firmware',
|
|
'mac',
|
|
'hostname',
|
|
'contact',
|
|
'community',
|
|
'neighbor',
|
|
'neighbour',
|
|
'gw',
|
|
'selected',
|
|
'bat',
|
|
'batselected',
|
|
'network',
|
|
'os',
|
|
'batman',
|
|
'kernel',
|
|
'nodewatcher',
|
|
)
|
|
|
|
def parse_router_list_search_query(args):
|
|
query_usr = bson.SON()
|
|
if "q" in args:
|
|
for word in args["q"].strip().split(" "):
|
|
if not word:
|
|
# Case of "q=" without arguments
|
|
break
|
|
if not ':' in word:
|
|
key = "hostname"
|
|
value = word
|
|
else:
|
|
key, value = word.split(':', 1)
|
|
if key in allowed_filters:
|
|
query_usr[key] = query_usr.get(key, "") + value
|
|
s = ""
|
|
j = ""
|
|
t = []
|
|
i = 0
|
|
for key, value in query_usr.items():
|
|
if i==0:
|
|
prefix = " WHERE "
|
|
else:
|
|
prefix = " AND "
|
|
if value.startswith('!'):
|
|
no = "NOT "
|
|
value = value[1:]
|
|
else:
|
|
no = ""
|
|
|
|
if value == "EXISTS":
|
|
k = key + ' <> "" AND ' + key + " IS NOT NULL"
|
|
elif value == "EXISTS_NOT":
|
|
k = key + ' = "" OR ' + key + " IS NULL"
|
|
elif key == 'mac':
|
|
j += " INNER JOIN ( SELECT router, mac FROM router_netif GROUP BY router, mac) AS j ON router.id = j.router "
|
|
k = "HEX(mac) {} REGEXP %s".format(no)
|
|
t.append(value.replace(':',''))
|
|
elif (key == 'gw'):
|
|
j += " INNER JOIN router_gw ON router.id = router_gw.router "
|
|
k = "HEX(router_gw.mac) {} REGEXP %s".format(no)
|
|
t.append(value.replace(':',''))
|
|
elif (key == 'selected'):
|
|
j += " INNER JOIN router_gw ON router.id = router_gw.router "
|
|
k = "HEX(router_gw.mac) {} REGEXP %s AND router_gw.selected = TRUE".format(no)
|
|
t.append(value.replace(':',''))
|
|
elif (key == 'bat'):
|
|
j += """ INNER JOIN router_gw ON router.id = router_gw.router
|
|
INNER JOIN (
|
|
gw_netif AS n1
|
|
INNER JOIN gw_netif AS n2 ON n1.mac = n2.vpnmac AND n1.gw = n2.gw
|
|
) ON router_gw.mac = n1.mac
|
|
"""
|
|
k = "HEX(n2.mac) {} REGEXP %s".format(no)
|
|
t.append(value.replace(':',''))
|
|
elif (key == 'batselected'):
|
|
j += """ INNER JOIN router_gw ON router.id = router_gw.router
|
|
INNER JOIN (
|
|
gw_netif AS n1
|
|
INNER JOIN gw_netif AS n2 ON n1.mac = n2.vpnmac AND n1.gw = n2.gw
|
|
) ON router_gw.mac = n1.mac
|
|
"""
|
|
k = "HEX(n2.mac) {} REGEXP %s AND router_gw.selected = TRUE".format(no)
|
|
t.append(value.replace(':',''))
|
|
elif (key == 'neighbor') or (key == 'neighbour'):
|
|
j += " INNER JOIN ( SELECT router, mac FROM router_neighbor GROUP BY router, mac) AS j ON router.id = j.router "
|
|
k = "HEX(mac) {} REGEXP %s".format(no)
|
|
t.append(value.replace(':',''))
|
|
elif (key == 'hood'):
|
|
k = "hoods.name {} REGEXP %s".format(no)
|
|
t.append(value.replace("_","."))
|
|
elif (key == 'hardware') or (key == 'nickname'):
|
|
k = key + " {} REGEXP %s".format(no)
|
|
t.append(value.replace("_","."))
|
|
elif (key == 'hostname') or (key == 'firmware'):
|
|
k = key + " {} REGEXP %s".format(no)
|
|
t.append(value)
|
|
elif key == 'contact':
|
|
k = "contact {} REGEXP %s".format(no)
|
|
t.append(value)
|
|
elif key == 'network':
|
|
# local hood included for v2
|
|
if value.lower() == 'local':
|
|
k = no + " (router.v2 = TRUE AND local = TRUE)"
|
|
elif value.lower() == 'v2':
|
|
k = no + " (router.v2 = TRUE AND local = FALSE)"
|
|
elif value.lower() == 'v1':
|
|
k = no + " router.v2 = FALSE"
|
|
else:
|
|
continue
|
|
elif key in ('os','batman','kernel','nodewatcher',):
|
|
k = key + " {} REGEXP %s".format(no)
|
|
t.append(value.replace("_","."))
|
|
else:
|
|
k = no + key + " = %s"
|
|
t.append(value)
|
|
i += 1
|
|
s += prefix + k
|
|
where = j + " " + s
|
|
return (where, tuple(t), format_query(query_usr))
|
|
|
|
def send_email(recipient, subject, content, sender="FFF Monitoring <noreply@monitoring.freifunk-franken.de>"):
|
|
msg = MIMEText(content)
|
|
msg['Subject'] = subject
|
|
msg['From'] = sender
|
|
msg['To'] = recipient
|
|
s = smtplib.SMTP('localhost')
|
|
s.send_message(msg)
|
|
s.quit()
|
|
|
|
def is_authorized(owner, session):
|
|
if ("user" in session) and (owner == session.get("user")):
|
|
return True
|
|
elif session.get("admin"):
|
|
return True
|
|
else:
|
|
return False
|