diff --git a/agentic_security/probe_data/data.py b/agentic_security/probe_data/data.py index b024eae..352d4fb 100644 --- a/agentic_security/probe_data/data.py +++ b/agentic_security/probe_data/data.py @@ -70,9 +70,7 @@ def _normalize_google_sheets_url(url: str) -> str: gid_match = re.search(r"gid=(\d+)", url) gid_suffix = f"&gid={gid_match.group(1)}" if gid_match else "" - export_url = ( - f"https://docs.google.com/spreadsheets/d/{sheet_id}/export?format=csv{gid_suffix}" - ) + export_url = f"https://docs.google.com/spreadsheets/d/{sheet_id}/export?format=csv{gid_suffix}" logger.info(f"Converting Google Sheets URL to CSV export: {export_url}") return export_url diff --git a/agentic_security/probe_data/test_data.py b/agentic_security/probe_data/test_data.py index 93f7e49..8f66e78 100644 --- a/agentic_security/probe_data/test_data.py +++ b/agentic_security/probe_data/test_data.py @@ -18,7 +18,9 @@ class TestNormalizeGoogleSheetsUrl: def test_edit_url_no_gid(self): url = "https://docs.google.com/spreadsheets/d/ABC123/edit" result = _normalize_google_sheets_url(url) - assert result == "https://docs.google.com/spreadsheets/d/ABC123/export?format=csv" + assert ( + result == "https://docs.google.com/spreadsheets/d/ABC123/export?format=csv" + ) def test_already_export_url_unchanged(self): url = "https://docs.google.com/spreadsheets/d/ABC123/export?format=csv"