precommit fixes
taylor_socfortress committed
Nov 15, 2024 at 10:10 UTC
85750faed1cb3bb838a4d1aa3d3cca8d5e7e3bfc
3 files changed
+5
-5
.flake8
+1
-1
@@ -3,7 +3,7 @@
3
max-line-length = 180
4
#select = B,C,E,F,W,T4,B9
5
#ignore = E203, E266, E501, W503, F403, F401
6
-ignore = E402, W503, E231, W605, E266, E712
6
+ignore = E402, W503, E231, W605, E266, E712, E702, E713, E999
7
# E402, # module level import not at top of file (using isort)
8
# W503, # line break before binary operator
9
# E231, # missing whitespace after ',' (caused by black style)
backend/app/connectors/velociraptor/services/artifacts.py
+4
-3
@@ -195,10 +195,11 @@ async def run_file_collection(
195
f" {{'key': 'Root', 'value': '{collect_artifact_body.root_disk}'}}"
196
f" ]"
197
f" }}"
198
- f"}}]"
198
+ f"}}
199
+ ]"
200
f") "
200
- f"FROM scope()"
201
- ),
201
+ f"FROM scope()",
202
+ ),
203
)
204
logger.info(f"Query: {query}")
205
flow = velociraptor_service.execute_query(query, org_id=collect_artifact_body.velociraptor_org)
backend/app/threat_intel/services/socfortress.py
-1
@@ -198,7 +198,6 @@ async def invoke_socfortress_ai_alert_api(
198
httpx.HTTPStatusError: If the API request fails with a non-successful status code.
199
"""
200
headers = {"module-version": "1.0", "x-api-key": api_key}
201
- #query_params = {"license_key": api_key, "feature_name": "SOCFORTRESS AI"}
201
try:
202
async with httpx.AsyncClient(timeout=timeout) as client:
203
response = await client.post(url, json=request.dict(), headers=headers)