Site/programmer/admin.py

from django.contrib import admin
from django.utils import timezone
from django.utils.html import format_html
from django.urls import path
from django.shortcuts import render
from django.http import HttpResponseRedirect
from django.contrib import messages
from .models import *


class ProgrammerAdmin(admin.ModelAdmin):
    list_display = ('id', 'title', 'time_create', 'photo', 'is_published')
    list_display_links = ('id', 'title')
    search_fields = ('title', 'content')
    list_editable = ('is_published',)
    list_filter = ('time_create', 'is_published')


class RecallAdmin(admin.ModelAdmin):
    list_display = ('id', 'title', 'time_create', 'scan', 'is_published')
    list_display_links = ('id', 'title')
    search_fields = ('title', 'content')
    list_editable = ('is_published',)
    list_filter = ('time_create', 'is_published')


class SolutionAdmin(admin.ModelAdmin):
    list_display = ('id', 'title', 'time_create', 'is_published')
    list_display_links = ('id', 'title')
    search_fields = ('title', 'description', 'implementation')
    list_editable = ('is_published',)
    list_filter = ('time_create', 'is_published')


class HomeAdmin(admin.ModelAdmin):
    list_display = ('id', 'title', 'time_create', 'is_published')
    list_display_links = ('id', 'title')
    search_fields = ('title', 'content')
    list_editable = ('is_published',)
    list_filter = ('time_create', 'is_published')


@admin.register(CallbackRequest)
class CallbackAdmin(admin.ModelAdmin):
    list_display = ('name', 'phone', 'email', 'time_create', 'is_processed', 'is_read', 'new_badge')
    list_display_links = ('name', 'phone')
    list_editable = ('is_processed', 'is_read')
    list_filter = ('time_create', 'is_processed', 'is_read')
    search_fields = ('name', 'phone', 'email')
    readonly_fields = ('time_create',)
    # actions = ['mark_as_read', 'mark_as_unread', 'mark_as_processed']
    actions = ['mark_as_read', 'mark_as_unread', 'mark_as_processed', 'resend_notification']

    def resend_notification(self, request, queryset):
        from .utils.email_notifications import send_callback_notification
        count = 0
        for callback in queryset:
            success = send_callback_notification(callback)
            if success:
                count += 1
        self.message_user(request, f'Уведомления отправлены для {count} заявок')

    resend_notification.short_description = "Переотправить email уведомления"

    def new_badge(self, obj):
        if not obj.is_read:
            return format_html('<span style="color: red; font-weight: bold;">🆕 НОВАЯ</span>')
        return ""

    new_badge.short_description = 'Статус'

    def get_queryset(self, request):
        # Показываем количество непрочитанных в заголовке
        unread_count = CallbackRequest.objects.filter(is_read=False).count()
        if unread_count > 0:
            self.message_user(
                request,
                f'У вас {unread_count} непрочитанных заявок!',
                messages.WARNING
            )
        return super().get_queryset(request)

    def mark_as_read(self, request, queryset):
        updated = queryset.update(is_read=True)
        self.message_user(request, f'{updated} заявок отмечены как прочитанные')

    mark_as_read.short_description = "Отметить как прочитанные"

    def mark_as_unread(self, request, queryset):
        updated = queryset.update(is_read=False)
        self.message_user(request, f'{updated} заявок отмечены как непрочитанные')

    mark_as_unread.short_description = "Отметить как непрочитанные"

    def mark_as_processed(self, request, queryset):
        updated = queryset.update(is_processed=True)
        self.message_user(request, f'{updated} заявок отмечены как обработанные')

    mark_as_processed.short_description = "Отметить как обработанные"

    # Добавляем кастомное представление для статистики
    def get_urls(self):
        urls = super().get_urls()
        custom_urls = [
            path('callback-stats/', self.admin_site.admin_view(self.callback_stats), name='callback_stats'),
        ]
        return custom_urls + urls

    def callback_stats(self, request):
        today = timezone.now().date()
        week_ago = today - timezone.timedelta(days=7)

        stats = {
            'total': CallbackRequest.objects.count(),
            'today': CallbackRequest.objects.filter(time_create__date=today).count(),
            'week': CallbackRequest.objects.filter(time_create__date__gte=week_ago).count(),
            'unread': CallbackRequest.objects.filter(is_read=False).count(),
            'unprocessed': CallbackRequest.objects.filter(is_processed=False).count(),
        }

        context = {
            **self.admin_site.each_context(request),
            'title': 'Статистика заявок',
            'stats': stats,
        }
        return render(request, 'admin/callback_stats.html', context)

class ProgrammerAdmin(admin.ModelAdmin):
    list_display = ('id', 'title', 'time_create', 'photo', 'is_published')
    list_display_links = ('id', 'title')
    search_fields = ('title', 'content')
    list_editable = ('is_published',)
    list_filter = ('time_create', 'is_published')

class RecallAdmin(admin.ModelAdmin):
    list_display = ('id', 'title', 'time_create', 'scan', 'is_published')
    list_display_links = ('id', 'title')
    search_fields = ('title', 'content')
    list_editable = ('is_published',)
    list_filter = ('time_create', 'is_published')

class SolutionAdmin(admin.ModelAdmin):
    list_display = ('id', 'title', 'time_create', 'is_published')
    list_display_links = ('id', 'title')
    search_fields = ('title', 'description', 'implementation')
    list_editable = ('is_published',)
    list_filter = ('time_create', 'is_published')

class HomeAdmin(admin.ModelAdmin):
    list_display = ('id', 'title', 'time_create', 'is_published')
    list_display_links = ('id', 'title')
    search_fields = ('title', 'content')
    list_editable = ('is_published',)
    list_filter = ('time_create', 'is_published')

@admin.register(PageView)
class PageViewAdmin(admin.ModelAdmin):
    list_display = ['url', 'timestamp', 'ip_address']
    list_filter = ['timestamp', 'url']
    search_fields = ['url', 'ip_address']
    date_hierarchy = 'timestamp'

    def get_queryset(self, request):
        return super().get_queryset(request).order_by('-timestamp')

admin.site.register(Competence, ProgrammerAdmin)
admin.site.register(Recall, RecallAdmin)
admin.site.register(Solution, SolutionAdmin)
admin.site.register(Home, HomeAdmin)