mirror of
https://github.com/yogeshojha/rengine.git
synced 2026-10-01 06:04:54 +02:00
79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
# include all the celery tasks to be used in the API, do not put in tasks.py
|
|
import requests
|
|
|
|
from reNgine.common_func import create_inappnotification, get_hackerone_key_username
|
|
from reNgine.definitions import PROJECT_LEVEL_NOTIFICATION
|
|
from reNgine.celery import app
|
|
|
|
@app.task(name='import_hackerone_programs_task', bind=False, queue='api_queue')
|
|
def import_hackerone_programs_task(handles, project_slug):
|
|
"""
|
|
Runs in the background to import programs from HackerOne
|
|
|
|
Args:
|
|
handles (list): List of handles to import
|
|
project_slug (str): Slug of the project
|
|
Returns:
|
|
None
|
|
rather creates inapp notifications
|
|
"""
|
|
def fetch_program_details_from_hackerone(program_handle):
|
|
url = f'https://api.hackerone.com/v1/hackers/programs/{program_handle}'
|
|
headers = {'Accept': 'application/json'}
|
|
username, api_key = get_hackerone_key_username()
|
|
|
|
response = requests.get(
|
|
url,
|
|
headers=headers,
|
|
auth=(username, api_key)
|
|
)
|
|
|
|
if response.status_code == 200:
|
|
return response.json()
|
|
else:
|
|
return None
|
|
|
|
for handle in handles:
|
|
program_details = fetch_program_details_from_hackerone(handle)
|
|
if program_details:
|
|
# TODO: Implement actual import logic to reNgine here
|
|
|
|
program_name = program_details['attributes']['name']
|
|
|
|
try:
|
|
create_inappnotification(
|
|
title=f"HackerOne Program Imported: {handle}",
|
|
description=f"The program '{program_name}' from hackerone has been successfully imported.",
|
|
notification_type=PROJECT_LEVEL_NOTIFICATION,
|
|
project_slug=project_slug,
|
|
icon="mdi-check-circle",
|
|
status='success'
|
|
)
|
|
|
|
except Exception as e:
|
|
create_inappnotification(
|
|
title=f"HackerOne Program Import Failed: {handle}",
|
|
description=f"Failed to import program from hackerone with handle '{handle}'. {str(e)}",
|
|
notification_type=PROJECT_LEVEL_NOTIFICATION,
|
|
project_slug=project_slug,
|
|
icon="mdi-alert-circle",
|
|
status='error'
|
|
)
|
|
else:
|
|
create_inappnotification(
|
|
title=f"HackerOne Program Import Failed: {handle}",
|
|
description=f"Failed to import program from hackerone with handle '{handle}'. Program details could not be fetched.",
|
|
notification_type=PROJECT_LEVEL_NOTIFICATION,
|
|
project_slug=project_slug,
|
|
icon="mdi-alert-circle",
|
|
status='error'
|
|
)
|
|
|
|
create_inappnotification(
|
|
title="HackerOne Program Import Completed",
|
|
description=f"Import process for {len(handles)} program(s) has completed.",
|
|
notification_type=PROJECT_LEVEL_NOTIFICATION,
|
|
project_slug=project_slug,
|
|
icon="mdi-check-all",
|
|
status='success'
|
|
) |