kaauh_ats/recruitment/management/commands/verify_notifications.py

113 lines
4.9 KiB
Python

from django.core.management.base import BaseCommand
from django.contrib.auth.models import User
from recruitment.models import Notification, HiringAgency
import datetime
class Command(BaseCommand):
help = 'Verify the notification system is working correctly'
def add_arguments(self, parser):
parser.add_argument(
'--detailed',
action='store_true',
help='Show detailed breakdown of notifications',
)
def handle(self, *args, **options):
self.stdout.write(self.style.SUCCESS('🔍 Verifying Notification System'))
self.stdout.write('=' * 50)
# Check notification counts
total_notifications = Notification.objects.count()
pending_notifications = Notification.objects.filter(status='PENDING').count()
sent_notifications = Notification.objects.filter(status='SENT').count()
failed_notifications = Notification.objects.filter(status='FAILED').count()
self.stdout.write(f'\n📊 Notification Counts:')
self.stdout.write(f' Total Notifications: {total_notifications}')
self.stdout.write(f' Pending: {pending_notifications}')
self.stdout.write(f' Sent: {sent_notifications}')
self.stdout.write(f' Failed: {failed_notifications}')
# Agency messaging system has been removed - replaced by Notification system
self.stdout.write(f'\n💬 Message System:')
self.stdout.write(f' Agency messaging system has been replaced by Notification system')
# Check admin user notifications
admin_users = User.objects.filter(is_staff=True)
self.stdout.write(f'\n👤 Admin Users ({admin_users.count()}):')
for admin in admin_users:
admin_notifications = Notification.objects.filter(recipient=admin).count()
admin_unread = Notification.objects.filter(recipient=admin, status='PENDING').count()
self.stdout.write(f' {admin.username}: {admin_notifications} notifications ({admin_unread} unread)')
# Check agency notifications
# Note: Current Notification model only supports User recipients, not agencies
# Agency messaging system has been removed
agencies = HiringAgency.objects.all()
self.stdout.write(f'\n🏢 Agencies ({agencies.count()}):')
for agency in agencies:
self.stdout.write(f' {agency.name}: Agency messaging system has been removed')
# Check notification types
if options['detailed']:
self.stdout.write(f'\n📋 Detailed Notification Breakdown:')
# By type
for notification_type in ['email', 'in_app']:
count = Notification.objects.filter(notification_type=notification_type).count()
if count > 0:
self.stdout.write(f' {notification_type}: {count}')
# By status
for status in ['pending', 'sent', 'read', 'failed', 'retrying']:
count = Notification.objects.filter(status=status).count()
if count > 0:
self.stdout.write(f' {status}: {count}')
# System health check
self.stdout.write(f'\n🏥 System Health Check:')
issues = []
# Check for failed notifications
if failed_notifications > 0:
issues.append(f'{failed_notifications} failed notifications')
# Check for admin users without notifications
admin_with_no_notifications = admin_users.filter(
notifications__isnull=True
).count()
if admin_with_no_notifications > 0 and total_notifications > 0:
issues.append(f'{admin_with_no_notifications} admin users with no notifications')
if issues:
self.stdout.write(self.style.WARNING(' ⚠️ Issues found:'))
for issue in issues:
self.stdout.write(f' - {issue}')
else:
self.stdout.write(self.style.SUCCESS(' ✅ No issues detected'))
# Recent activity
recent_notifications = Notification.objects.filter(
created_at__gte=datetime.datetime.now() - datetime.timedelta(hours=24)
).count()
self.stdout.write(f'\n🕐 Recent Activity (last 24 hours):')
self.stdout.write(f' New notifications: {recent_notifications}')
# Summary
self.stdout.write(f'\n📋 Summary:')
if total_notifications > 0 and failed_notifications == 0:
self.stdout.write(self.style.SUCCESS(' ✅ Notification system is working correctly'))
elif failed_notifications > 0:
self.stdout.write(self.style.WARNING(' ⚠️ Notification system has some failures'))
else:
self.stdout.write(self.style.WARNING(' ⚠️ No notifications found - system may not be active'))
self.stdout.write('\n' + '=' * 50)
self.stdout.write(self.style.SUCCESS('✨ Verification complete!'))