You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
2.4 KiB
58 lines
2.4 KiB
import random
|
|
import string
|
|
from django.db.models.signals import post_save
|
|
# from django.db.transaction import DatabaseError
|
|
from django.dispatch import receiver
|
|
from django.conf import settings
|
|
# from django.apps import apps
|
|
# from django.db.models import Q
|
|
|
|
from .models import Club, FailedApiCall, Log
|
|
import requests
|
|
|
|
# Others
|
|
|
|
def generate_unique_code():
|
|
characters = string.ascii_letters + string.digits
|
|
while True:
|
|
code = ''.join(random.sample(characters, 3))
|
|
if not Club.objects.filter(broadcast_code=code).exists():
|
|
return code
|
|
|
|
@receiver(post_save, sender=Club)
|
|
def assign_unique_code(sender, instance, created, **kwargs):
|
|
if created and not instance.broadcast_code:
|
|
instance.broadcast_code = generate_unique_code()
|
|
instance.save()
|
|
|
|
DISCORD_FAILED_CALLS_WEBHOOK_URL = 'https://discord.com/api/webhooks/1248191778134163486/sSoTL6cULCElWr2YFwyllsg7IXxHcCx_YMDJA_cUHtVUU4WOfN-5M7drCJuwNBBfAk9a'
|
|
DISCORD_LOGS_WEBHOOK_URL = 'https://discord.com/api/webhooks/1257987637449588736/TtOUwzYgSlQH2d3Ps7SfIKRcFALQVa3hfkC-j9K4_UAcWtsfiw4v8NUPbnX2_ZPOYzuv'
|
|
|
|
@receiver(post_save, sender=FailedApiCall)
|
|
def notify_discord_on_create(sender, instance, created, **kwargs):
|
|
notify_object_creation_on_discord(created, instance, DISCORD_FAILED_CALLS_WEBHOOK_URL)
|
|
|
|
@receiver(post_save, sender=Log)
|
|
def notify_log_creation_on_discord(sender, instance, created, **kwargs):
|
|
notify_object_creation_on_discord(created, instance, DISCORD_LOGS_WEBHOOK_URL)
|
|
|
|
# WARNING: using this method requires the instance to have a discord_string method
|
|
def notify_object_creation_on_discord(created, instance, webhook_url):
|
|
if created:
|
|
default_db_engine = settings.DATABASES['default']['ENGINE']
|
|
if default_db_engine != 'django.db.backends.sqlite3':
|
|
if hasattr(instance, 'discord_string'):
|
|
message = f'New {instance.__class__.__name__} created: {instance.discord_string()}'
|
|
else:
|
|
message = "no message. Please configure 'discord_string' on your instance"
|
|
send_discord_message(webhook_url, message)
|
|
|
|
def send_discord_message(webhook_url, content):
|
|
data = {
|
|
"content": content
|
|
}
|
|
response = requests.post(webhook_url, json=data)
|
|
if response.status_code != 204:
|
|
raise ValueError(
|
|
f'Error sending message to Discord webhook: {response.status_code}, {response.text}'
|
|
)
|
|
|