107 lines
2.4 KiB
Python
107 lines
2.4 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',
|
|
'community',
|
|
'nickname',
|
|
'hardware',
|
|
'firmware',
|
|
'mac',
|
|
'netif',
|
|
'netmon_id',
|
|
'hostname',
|
|
'contact',
|
|
)
|
|
|
|
def query2where(query):
|
|
s = ""
|
|
t = []
|
|
i = 0
|
|
for k, v in query.items():
|
|
if not k in allowed_filters:
|
|
# prevent SQL injection
|
|
continue
|
|
if i==0:
|
|
prefix = " WHERE "
|
|
else:
|
|
prefix = " AND "
|
|
i += 1
|
|
s += prefix + k + " = %s"
|
|
t.append(v)
|
|
return (s,tuple(t))
|
|
|
|
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
|
|
query = {}
|
|
for key, value in query_usr.items():
|
|
#if value == "EXISTS":
|
|
# query[key] = {"$exists": True}
|
|
#elif value == "EXISTS_NOT":
|
|
# query[key] = {"$exists": False}
|
|
if key == 'mac':
|
|
query[key] = value.lower()
|
|
#elif key == 'netif':
|
|
# query[key] = {"$regex": value.replace('.', '\.'), "$options": 'i'}
|
|
elif key == 'hostname':
|
|
query[key] = value.replace('\\', '')
|
|
elif key == 'hardware':
|
|
query[key] = value.replace('\\', '').replace('_', ' ')
|
|
#elif key == 'netmon_id':
|
|
# query[key] = int(value)
|
|
elif key == 'contact':
|
|
query[key] = value.replace('\\', '')
|
|
#elif key == 'contact':
|
|
# if not '\.' in value:
|
|
# value = re.escape(value)
|
|
# query[key] = {"$regex": value, "$options": 'i'}
|
|
#elif value.startswith('!'):
|
|
# query[key] = {"$ne": value.replace('!', '', 1)}
|
|
else:
|
|
query[key] = value
|
|
return (query, 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
|