from django.contrib import admin
from django.utils.html import format_html
from django.utils import timezone
 
@admin.register(Article)
class ArticleAdmin(admin.ModelAdmin):
    list_display = ('title', 'author_name', 'status_badge', 'created_at', 'word_count_display')
    list_filter = ('status', 'author', 'created_at')
    search_fields = ('title', 'content', 'author__name')
    readonly_fields = ('created_at', 'updated_at', 'word_count_display')
    date_hierarchy = 'created_at'
    ordering = ('-created_at',)
 
    fieldsets = (
        ('Content', {'fields': ('title', 'slug', 'content', 'summary')}),
        ('Metadata', {'fields': ('author', 'status', 'published_at')}),
        ('System', {'fields': ('created_at', 'updated_at'), 'classes': ('collapse',)}),
    )
 
    def author_name(self, obj):
        return obj.author.name
    author_name.short_description = 'Author'
    author_name.admin_order_field = 'author__name'
 
    def status_badge(self, obj):
        colors = {'draft': '#6c757d', 'published': '#28a745', 'archived': '#dc3545'}
        return format_html(
            '<span style="background:{};color:white;padding:3px 8px;border-radius:3px">{}</span>',
            colors.get(obj.status, '#6c757d'), obj.status.upper()
        )
    status_badge.short_description = 'Status'
 
    def word_count_display(self, obj):
        return f"{len(obj.content.split())} words"
    word_count_display.short_description = 'Word Count'