mirror of
https://github.com/eduardogsilva/routerfleet.git
synced 2025-06-21 01:25:41 +02:00
105 lines
4.6 KiB
Python
105 lines
4.6 KiB
Python
import datetime
|
|
|
|
from router_manager.models import Router
|
|
from .models import MessageChannel, Notification, MessageSettings, Message
|
|
from backup_data.models import RouterBackup
|
|
import requests
|
|
from django.utils import timezone
|
|
|
|
|
|
def send_notification_message(message: Message):
|
|
message_settings, _ = MessageSettings.objects.get_or_create(name='message_settings')
|
|
if message.status != 'pending':
|
|
return
|
|
if message.retry_count > message_settings.max_retry:
|
|
message.status = 'failed'
|
|
message.completed = timezone.now()
|
|
message.save()
|
|
return
|
|
|
|
message_response = {'status': 'pending', 'error_message': '', 'error_status_code': 0}
|
|
if message.channel.channel_type == 'callmebot':
|
|
url = f'https://api.callmebot.com/whatsapp.php?phone={message.channel.destination}&text={message.message}&apikey={message.channel.token}'
|
|
elif message.channel.channel_type == 'telegram':
|
|
url = f'https://api.telegram.org/bot{message.channel.token}/sendMessage?chat_id={message.channel.destination}&text={message.message}'
|
|
else:
|
|
message_response['status'] = 'failed'
|
|
message_response['error_message'] = 'Failed to send message: Invalid channel type'
|
|
message_response['error_status_code'] = 0
|
|
|
|
if message_response['status'] == 'pending':
|
|
try:
|
|
response = requests.get(url)
|
|
if response.status_code == 200:
|
|
message_response['status'] = 'sent'
|
|
else:
|
|
message_response['status'] = 'failed'
|
|
message_response['error_message'] = response.text
|
|
message_response['error_status_code'] = response.status_code
|
|
except:
|
|
message_response['status'] = 'failed'
|
|
message_response['error_message'] = 'Failed to send message: Request exception'
|
|
message_response['error_status_code'] = 0
|
|
|
|
if message_response['status'] == 'sent':
|
|
message.status = 'sent'
|
|
message.completed = timezone.now()
|
|
message.save()
|
|
else:
|
|
message.retry_count += 1
|
|
message.error_message = message_response['error_message']
|
|
message.error_status_code = message_response['error_status_code']
|
|
message.next_retry = timezone.now() + datetime.timedelta(seconds=message_settings.retry_interval)
|
|
message.save()
|
|
return
|
|
|
|
|
|
def notify_router_status_update(router: Router):
|
|
message_channel_list = MessageChannel.objects.filter(enabled=True)
|
|
message_settings, _ = MessageSettings.objects.get_or_create(name='message_settings')
|
|
|
|
if message_settings.concatenate_status_change:
|
|
if router.routerstatus.status_online:
|
|
if message_channel_list.filter(status_change_online=True):
|
|
Notification.objects.create(notification_type='status_online', router=router)
|
|
else:
|
|
if message_channel_list.filter(status_change_offline=True):
|
|
Notification.objects.create(notification_type='status_offline', router=router)
|
|
else:
|
|
if router.routerstatus.status_online:
|
|
for message_channel in message_channel_list.filter(status_change_online=True):
|
|
Message.objects.create(
|
|
channel=message_channel,
|
|
subject='Router status change: Online',
|
|
message=f'Router {router.name} ({router.address}) is now online'
|
|
)
|
|
else:
|
|
for message_channel in message_channel_list.filter(status_change_offline=True):
|
|
Message.objects.create(
|
|
channel=message_channel,
|
|
subject='Router status change: Offline',
|
|
message=f'Router {router.name} ({router.address}) is now offline'
|
|
)
|
|
return
|
|
|
|
|
|
def notify_backup_fail(router_backup: RouterBackup):
|
|
message_settings, _ = MessageSettings.objects.get_or_create(name='message_settings')
|
|
message_channel_list = MessageChannel.objects.filter(enabled=True, backup_fail=True)
|
|
|
|
if not message_channel_list:
|
|
return
|
|
|
|
if message_settings.concatenate_backup_fails:
|
|
Notification.objects.create(notification_type='backup_fail', router=router_backup.router, router_backup=router_backup)
|
|
else:
|
|
error_message = f'Backup {router_backup.id} failed for router {router_backup.router.name} ({router_backup.router.address})'
|
|
if router_backup.error_message:
|
|
error_message += f'\n\nError message: {router_backup.error_message}'
|
|
for message_channel in message_channel_list:
|
|
Message.objects.create(
|
|
channel=message_channel,
|
|
subject=f'Backup failed: {router_backup.id}',
|
|
message=error_message
|
|
)
|
|
return
|