|
1 | 1 | import json |
| 2 | +import os |
2 | 3 | import sqlite3 |
3 | 4 |
|
| 5 | +def _get_f2b_data(db_path): |
| 6 | + if not os.path.isfile(db_path): |
| 7 | + return [] |
| 8 | + |
| 9 | + con = sqlite3.connect(db_path) |
| 10 | + cur = con.cursor() |
| 11 | + results = cur.execute(""" |
| 12 | + SELECT jails.name, |
| 13 | + COUNT(bans.ip) AS bans, |
| 14 | + (SELECT DISTINCT bans.ip from bans where jails.name = bans.jail ORDER BY timeofban DESC) as last_ban, |
| 15 | + (SELECT DISTINCT bans.data from bans where jails.name = bans.jail ORDER BY timeofban DESC) as data |
| 16 | + FROM jails |
| 17 | + LEFT JOIN bans ON jails.name=bans.jail |
| 18 | + GROUP BY jails.name |
| 19 | + """).fetchall() |
| 20 | + con.close() |
| 21 | + return [{ |
| 22 | + "name": name, |
| 23 | + "bans": bans, |
| 24 | + "last_ban": last_ban, |
| 25 | + "data": json.dumps(json.loads(data), indent=4, sort_keys=True) if data else None |
| 26 | + } for (name, bans, last_ban, data) in results] |
4 | 27 |
|
5 | | -con = sqlite3.connect("/config/fail2ban/fail2ban.sqlite3") |
6 | | -cur = con.cursor() |
7 | | -results = cur.execute(""" |
8 | | - SELECT jails.name, |
9 | | - COUNT(bans.ip) AS bans, |
10 | | - (SELECT DISTINCT bans.ip from bans where jails.name = bans.jail ORDER BY timeofban DESC) as last_ban, |
11 | | - (SELECT DISTINCT bans.data from bans where jails.name = bans.jail ORDER BY timeofban DESC) as data |
12 | | - FROM jails |
13 | | - LEFT JOIN bans ON jails.name=bans.jail |
14 | | - GROUP BY jails.name |
15 | | - """).fetchall() |
16 | | -con.close() |
17 | | -formatted_results = [{ |
18 | | - "name": name, |
19 | | - "bans": bans, |
20 | | - "last_ban": last_ban, |
21 | | - "data": json.dumps(json.loads(data), indent=4, sort_keys=True) if data else None |
22 | | -} for (name, bans, last_ban, data) in results] |
| 28 | +swag_f2b = _get_f2b_data("/config/fail2ban/fail2ban.sqlite3") |
| 29 | +host_f2b = _get_f2b_data("/dashboard/fail2ban/fail2ban.sqlite3") |
23 | 30 |
|
24 | | -output = json.dumps(formatted_results, sort_keys=True) |
| 31 | +output = json.dumps(swag_f2b + host_f2b, sort_keys=True) |
25 | 32 | print(output) |
0 commit comments