Public/fgAI
SHA256
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix restore config

Browse Source
This commit is contained in:
larssand
2026-07-06 09:46:22 +02:00
parent bddc84395c
commit 830ca79dfa
2 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/fgai/config.py
View File

@@ -82,10 +82,13 @@ class ConfigStore:
elif key == "threat_intel_provider" and value in {"auto", "abuseipdb", "virustotal"}: elif key == "threat_intel_provider" and value in {"auto", "abuseipdb", "virustotal"}:
current[key] = value current[key] = value
elif key == "graylog_streams" and isinstance(value, list): elif key == "graylog_streams" and isinstance(value, list):
current[key] = [ streams = [
{"id": str(item.get("id", "")), "title": str(item.get("title", "")), "enabled": bool(item.get("enabled"))} {"id": str(item.get("id", "")), "title": str(item.get("title", "")), "enabled": bool(item.get("enabled"))}
for item in value if isinstance(item, dict) and item.get("id") for item in value if isinstance(item, dict) and item.get("id")
] ]
if not streams and current.get("graylog_streams"):
continue
current[key] = streams
elif key == "graylog_stream_profiles" and isinstance(value, list): elif key == "graylog_stream_profiles" and isinstance(value, list):
current[key] = [item for item in value if isinstance(item, dict) and item.get("stream_id")] current[key] = [item for item in value if isinstance(item, dict) and item.get("stream_id")]
elif isinstance(value, str): elif isinstance(value, str):

7
tests/test_config.py
View File

@@ -54,6 +54,13 @@ class ConfigTests(unittest.TestCase):
public = store.update({"graylog_tls_verify": False}) public = store.update({"graylog_tls_verify": False})
self.assertFalse(public["graylog_tls_verify"]) self.assertFalse(public["graylog_tls_verify"])
def test_empty_stream_update_does_not_clear_existing_streams(self):
with tempfile.TemporaryDirectory() as directory:
store = ConfigStore(str(Path(directory) / "config.json"))
store.update({"graylog_streams": [{"id": "stream-1", "title": "Firewall", "enabled": True}]})
public = store.update({"graylog_streams": []})
self.assertEqual(public["graylog_streams"], [{"id": "stream-1", "title": "Firewall", "enabled": True}])
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()