frodovdr/rspamd-influxdb
1
0
Fork 0
mirror of https://github.com/thannaske/rspamd-influxdb.git synced 2023-10-10 13:36:59 +02:00
Code Issues Releases Wiki Activity

Initial commit

Browse Source
This commit is contained in:
Tobias 2018-01-29 22:27:11 +01:00
commit 06c1179bbd
1 changed files with 89 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

89
rspamd-influxdb.py Normal file
View File

@ -0,0 +1,89 @@
#!/usr/bin/env python
import argparse
import json
import os
import urllib.request
parser = argparse.ArgumentParser(description="rspamd web interface statistic fetcher for InfluxDB usage")
parser.add_argument("url", action="store", help="URL to rspamd web interface installation")
parser.add_argument("password", action="store", help="Password for API authentication (same as for graphical login)")
args = parser.parse_args()
# Make sure we got the trailing slash at the URL
if str.endswith(args.url, "/"):
fetch_url = args.url + "stat?password=" + urllib.parse.quote_plus(args.password)
else:
fetch_url = args.url + "/stat?password=" + urllib.parse.quote_plus(args.password)
try:
resp = urllib.request.urlopen(fetch_url)
except Exception:
print("Could not send GET request to given URL. Check url parameter!")
os.exit(1)
# Authorization failed
if resp.code == 403:
print("Authorization with rspamd web interface failed. Check password parameter!")
os.exit(1)
elif resp.code == 404:
print("HTTP server returned HTTP status code 404. Check url parameter!")
os.exit(1)
elif resp.code == 200:
# Successful call
json = json.loads(resp.read())
action_reject = str(json["actions"]["reject"]) + "i"
action_soft_reject = str(json["actions"]["soft reject"]) + "i"
action_rewrite = str(json["actions"]["rewrite subject"]) + "i"
action_add_header = str(json["actions"]["add header"]) + "i"
action_greylist = str(json["actions"]["greylist"]) + "i"
action_no_action = str(json["actions"]["no action"]) + "i"
# Build InfluxDB Line protocol compatible output
print(
"rspamd_actions reject={0},soft_reject={1},rewrite_subject={2},add_header={3},greylist={4},no_action={5}".format(
action_reject, action_soft_reject, action_rewrite, action_add_header, action_greylist, action_no_action))
stat_scanned = str(json["scanned"]) + "i"
stat_learned = str(json["learned"]) + "i"
stat_spam_count = str(json["spam_count"]) + "i"
stat_ham_count = str(json["ham_count"]) + "i"
stat_connections = str(json["connections"]) + "i"
stat_control_connections = str(json["control_connections"]) + "i"
stat_pools_allocated = str(json["pools_allocated"]) + "i"
stat_pools_freed = str(json["pools_freed"]) + "i"
stat_bytes_allocated = str(json["bytes_allocated"]) + "i"
stat_chunks_allocated = str(json["chunks_allocated"]) + "i"
stat_chunks_freed = str(json["chunks_freed"]) + "i"
stat_chunks_oversized = str(json["chunks_oversized"]) + "i"
stat_fragmented = str(json["fragmented"]) + "i"
stat_total_learns = str(json["total_learns"]) + "i"
stat_fuzzy_rspamd = str(json["fuzzy_hashes"]["rspamd.com"]) + "i"
print("rspamd_stats scanned={0},learned={1},spam_count={2},ham_count={3},connections={4},control_connections={5},"
"pools_allocated={6},pools_freed={7},bytes_allocated={8},chunks_allocated={9},chunks_freed={10},"
"chunks_oversized={11},fragmented={12},total_learns={13},fuzzy={14}".format(stat_scanned, stat_learned,
stat_spam_count,
stat_ham_count,
stat_connections,
stat_control_connections,
stat_pools_allocated,
stat_pools_freed,
stat_bytes_allocated,
stat_chunks_allocated,
stat_chunks_freed,
stat_chunks_oversized,
stat_fragmented,
stat_total_learns,
stat_fuzzy_rspamd))
os.exit(0)
else:
print("Web request returned unhandled HTTP status code " + str(resp.code) + ". Please open an issue at GitHub "
"with further details.")
os.exit(1)