mirror of
https://github.com/jokob-sk/NetAlertX.git
synced 2025-12-07 09:36:05 -08:00
api layer v0.2.6 - /events work
This commit is contained in:
@@ -4,7 +4,7 @@ from flask_cors import CORS
|
|||||||
from .graphql_endpoint import devicesSchema
|
from .graphql_endpoint import devicesSchema
|
||||||
from .device_endpoint import get_device_data, set_device_data, delete_device, delete_device_events, reset_device_props, copy_device, update_device_column
|
from .device_endpoint import get_device_data, set_device_data, delete_device, delete_device_events, reset_device_props, copy_device, update_device_column
|
||||||
from .devices_endpoint import get_all_devices, delete_unknown_devices, delete_all_with_empty_macs, delete_devices, export_devices, import_csv, devices_totals, devices_by_status
|
from .devices_endpoint import get_all_devices, delete_unknown_devices, delete_all_with_empty_macs, delete_devices, export_devices, import_csv, devices_totals, devices_by_status
|
||||||
from .events_endpoint import delete_events, delete_events_30, get_events, create_event
|
from .events_endpoint import delete_events, delete_events_older_than, get_events, create_event
|
||||||
from .history_endpoint import delete_online_history
|
from .history_endpoint import delete_online_history
|
||||||
from .prometheus_endpoint import get_metric_stats
|
from .prometheus_endpoint import get_metric_stats
|
||||||
from .sessions_endpoint import get_sessions, delete_session, create_session, get_sessions_calendar
|
from .sessions_endpoint import get_sessions, delete_session, create_session, get_sessions_calendar
|
||||||
@@ -315,11 +315,16 @@ def api_get_events():
|
|||||||
mac = request.args.get("mac")
|
mac = request.args.get("mac")
|
||||||
return get_events(mac)
|
return get_events(mac)
|
||||||
|
|
||||||
@app.route("/events/30days", methods=["DELETE"])
|
@app.route("/events/<int:days>", methods=["DELETE"])
|
||||||
def api_delete_old_events():
|
def api_delete_old_events(days: int):
|
||||||
|
"""
|
||||||
|
Delete events older than <days> days.
|
||||||
|
Example: DELETE /events/30
|
||||||
|
"""
|
||||||
if not is_authorized():
|
if not is_authorized():
|
||||||
return jsonify({"error": "Forbidden"}), 403
|
return jsonify({"error": "Forbidden"}), 403
|
||||||
return delete_events_30()
|
|
||||||
|
return delete_events_older_than(days)
|
||||||
|
|
||||||
# --------------------------
|
# --------------------------
|
||||||
# Sessions
|
# Sessions
|
||||||
|
|||||||
@@ -75,18 +75,23 @@ def get_events(mac=None):
|
|||||||
conn.close()
|
conn.close()
|
||||||
return jsonify({"success": True, "events": events})
|
return jsonify({"success": True, "events": events})
|
||||||
|
|
||||||
def delete_events_30():
|
def delete_events_older_than(days):
|
||||||
"""Delete all events older than 30 days"""
|
"""Delete all events older than a specified number of days"""
|
||||||
|
|
||||||
conn = get_temp_db_connection()
|
conn = get_temp_db_connection()
|
||||||
cur = conn.cursor()
|
cur = conn.cursor()
|
||||||
|
|
||||||
sql = "DELETE FROM Events WHERE eve_DateTime <= date('now', '-30 days')"
|
# Use a parameterized query with sqlite date function
|
||||||
cur.execute(sql)
|
sql = "DELETE FROM Events WHERE eve_DateTime <= date('now', ?)"
|
||||||
|
cur.execute(sql, [f'-{days} days'])
|
||||||
|
|
||||||
conn.commit()
|
conn.commit()
|
||||||
conn.close()
|
conn.close()
|
||||||
|
|
||||||
return jsonify({"success": True, "message": "Deleted events older than 30 days"})
|
return jsonify({
|
||||||
|
"success": True,
|
||||||
|
"message": f"Deleted events older than {days} days"
|
||||||
|
})
|
||||||
|
|
||||||
def delete_events():
|
def delete_events():
|
||||||
"""Delete all events"""
|
"""Delete all events"""
|
||||||
|
|||||||
@@ -99,7 +99,7 @@ def test_delete_all_events(client, api_token, test_mac):
|
|||||||
assert len(resp.json.get("events", [])) == 0
|
assert len(resp.json.get("events", [])) == 0
|
||||||
|
|
||||||
|
|
||||||
def test_delete_events_30days(client, api_token, test_mac):
|
def test_delete_events_dynamic_days(client, api_token, test_mac):
|
||||||
# create old + new events
|
# create old + new events
|
||||||
create_event(client, api_token, test_mac, days_old=40) # should be deleted
|
create_event(client, api_token, test_mac, days_old=40) # should be deleted
|
||||||
create_event(client, api_token, test_mac, days_old=5) # should remain
|
create_event(client, api_token, test_mac, days_old=5) # should remain
|
||||||
@@ -108,9 +108,10 @@ def test_delete_events_30days(client, api_token, test_mac):
|
|||||||
assert len(resp.json) == 2
|
assert len(resp.json) == 2
|
||||||
|
|
||||||
# delete events older than 30 days
|
# delete events older than 30 days
|
||||||
resp = client.delete("/events/30days", headers=auth_headers(api_token))
|
resp = client.delete("/events/30", headers=auth_headers(api_token))
|
||||||
assert resp.status_code == 200
|
assert resp.status_code == 200
|
||||||
assert resp.json.get("success") is True
|
assert resp.json.get("success") is True
|
||||||
|
assert "Deleted events older than 30 days" in resp.json.get("message", "")
|
||||||
|
|
||||||
# confirm only recent remains
|
# confirm only recent remains
|
||||||
resp = list_events(client, api_token, test_mac)
|
resp = list_events(client, api_token, test_mac)
|
||||||
@@ -118,3 +119,4 @@ def test_delete_events_30days(client, api_token, test_mac):
|
|||||||
mac_events = [ev for ev in events if ev.get("eve_MAC") == test_mac]
|
mac_events = [ev for ev in events if ev.get("eve_MAC") == test_mac]
|
||||||
assert len(mac_events) == 1
|
assert len(mac_events) == 1
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -65,7 +65,6 @@ def test_graphql_post_devices(client, api_token):
|
|||||||
assert resp.status_code == 200
|
assert resp.status_code == 200
|
||||||
|
|
||||||
body = resp.get_json()
|
body = resp.get_json()
|
||||||
# print("FULL RESPONSE:", body)
|
|
||||||
|
|
||||||
# GraphQL spec: response always under "data"
|
# GraphQL spec: response always under "data"
|
||||||
assert "data" in body
|
assert "data" in body
|
||||||
|
|||||||
Reference in New Issue
Block a user