import requests from rocketry import Rocketry from rocketry.conds import every from sec import GITLAB_API_TOKEN, DISCORD_WEBHOOK_URL GITLAB_API_URL = 'https://gitlab.com/api/v4/projects/codebreaker-challenge%2F2024%2Foutreach-2024/issues?state=opened' GITLAB_PROJECT_URL = 'https://gitlab.com/codebreaker-challenge/2024/outreach-2024' # Headers for GitLab API request headers = { 'Private-Token': GITLAB_API_TOKEN } # Dictionary to keep track of reported issues and their update times reported_issues = {} # Function to send a message to Discord def send_discord_notification(title, email, subject, message, issue_url): data = { "content": f"**New Issue**\n**Email:** {email}\n**Subject:** {subject}\n**Message:** {message}\n**Link:** {issue_url}" } response = requests.post(DISCORD_WEBHOOK_URL, json=data) if response.status_code == 204: print("Notification sent successfully.") else: print(f"Failed to send notification. Status code: {response.status_code}") # Function to check for new issues def check_for_new_issues(): response = requests.get(GITLAB_API_URL, headers=headers) if response.status_code == 200: issues = response.json() for issue in issues: issue_id = issue['id'] updated_at = issue['updated_at'] if issue_id not in reported_issues or reported_issues[issue_id] != updated_at: title = issue['title'] email, subject = title.split(' - ', 1) message = issue['description'][:50] + '...' if len(issue['description']) > 50 else issue['description'] issue_url = f"{GITLAB_PROJECT_URL}/-/issues/{issue['iid']}" send_discord_notification(title, email, subject, message, issue_url) reported_issues[issue_id] = updated_at else: print(f"Failed to fetch issues. Status code: {response.status_code}") # Create a Rocketry app app = Rocketry() # Schedule the task to run every 30 seconds @app.task(every('20 seconds')) def scheduled_task(): check_for_new_issues() # Run the Rocketry app if __name__ == "__main__": app.run()