`refactor: Update endpoint in provision.py for huntress` (#329)
taylor_socfortress committed
Nov 4, 2024 at 17:27 UTC
810ef03ff4632df5351c07882b609edeff0f3fa6
3 files changed
+6
-6
backend/app/integrations/huntress/routes/provision.py
+2
-2
@@ -40,9 +40,9 @@ async def provision_huntress_route(
40
await provision_huntress(provision_huntress_request, session)
41
await add_scheduler_jobs(
42
CreateSchedulerRequest(
43
- function_name="invoke_huntress_integration_collection",
43
+ function_name="invoke_huntress_integration_collect",
44
time_interval=provision_huntress_request.time_interval,
45
- job_id="invoke_huntress_integration_collection",
45
+ job_id="invoke_huntress_integration_collect",
46
),
47
)
48
return ProvisionHuntressResponse(
backend/app/schedulers/scheduler.py
+1
-1
@@ -264,7 +264,7 @@ def get_function_by_name(function_name: str):
264
"invoke_sap_siem_integration_brute_force_failed_logins": invoke_sap_siem_integration_brute_force_failed_logins,
265
"invoke_sap_siem_integration_brute_force_failed_logins_same_ip": invoke_sap_siem_integration_brute_force_failed_logins_same_ip,
266
"invoke_sap_siem_integration_successful_login_after_multiple_failed_logins": invoke_sap_siem_integration_successful_login_after_multiple_failed_logins,
267
- "invoke_huntress_integration_collection": invoke_huntress_integration_collect,
267
+ "invoke_huntress_integration_collect": invoke_huntress_integration_collect,
268
"invoke_duo_integration_collect": invoke_duo_integration_collect,
269
"invoke_darktrace_integration_collect": invoke_darktrace_integration_collect,
270
"invoke_carbonblack_integration_collection": invoke_carbonblack_integration_collect,
backend/app/schedulers/services/invoke_huntress.py
+3
-3
@@ -34,7 +34,7 @@ async def invoke_huntress_integration_collect() -> InvokeHuntressResponse:
34
InvokeHuntressRequest(
35
customer_code=customer_code,
36
integration_name="Huntress",
37
- time_range=f"{(await get_scheduled_job_metadata('invoke_huntress_integration_collection')).time_interval}m",
37
+ time_range=f"{(await get_scheduled_job_metadata('invoke_huntress_integration_collect')).time_interval}m",
38
),
39
session,
40
)
@@ -42,13 +42,13 @@ async def invoke_huntress_integration_collect() -> InvokeHuntressResponse:
42
await session.close()
43
with get_sync_db_session() as session:
44
# Synchronous ORM operations
45
- job_metadata = session.query(JobMetadata).filter_by(job_id="invoke_huntress_integration_collection").one_or_none()
45
+ job_metadata = session.query(JobMetadata).filter_by(job_id="invoke_huntress_integration_collect").one_or_none()
46
if job_metadata:
47
job_metadata.last_success = datetime.utcnow()
48
session.add(job_metadata)
49
session.commit()
50
else:
51
# Handle the case where job_metadata does not exist
52
- print("JobMetadata for 'invoke_huntress_integration_collection' not found.")
52
+ print("JobMetadata for 'invoke_huntress_integration_collect' not found.")
53
54
return InvokeHuntressResponse(success=True, message="Huntress integration invoked.")