from django import forms
from django.contrib import admin
from django.contrib.auth import get_user_model
from django.urls import reverse
from django.utils.html import format_html
from django.utils.translation import gettext_lazy as _

# 导入相关模型
from .models import Article, Category, Tag, Links, SideBar, BlogSettings


# 自定义表单，用于文章模型的管理界面
class ArticleForm(forms.ModelForm):
    # 可以在此处自定义字段的表单小部件
    class Meta:
        model = Article
        fields = '__all__'


# 定义批量操作函数
def makr_article_publish(modeladmin, request, queryset):
    # 将选中的文章状态更新为“已发布”
    queryset.update(status='p')


def draft_article(modeladmin, request, queryset):
    # 将选中的文章状态更新为“草稿”
    queryset.update(status='d')


def close_article_commentstatus(modeladmin, request, queryset):
    # 将选中的文章评论状态更新为“关闭”
    queryset.update(comment_status='c')


def open_article_commentstatus(modeladmin, request, queryset):
    # 将选中的文章评论状态更新为“开启”
    queryset.update(comment_status='o')


# 为批量操作函数添加描述
makr_article_publish.short_description = _('Publish selected articles')
draft_article.short_description = _('Draft selected articles')
close_article_commentstatus.short_description = _('Close article comments')
open_article_commentstatus.short_description = _('Open article comments')


# 自定义文章管理界面
class ArticlelAdmin(admin.ModelAdmin):
    # 每页显示的记录数
    list_per_page = 20
    # 可搜索的字段
    search_fields = ('body', 'title')
    # 使用自定义表单
    form = ArticleForm
    # 列表显示的字段
    list_display = (
        'id',
        'title',
        'author',
        'link_to_category',
        'creation_time',
        'views',
        'status',
        'type',
        'article_order')
    # 可点击进入详情的字段
    list_display_links = ('id', 'title')
    # 列表过滤器
    list_filter = ('status', 'type', 'category')
    # 按日期层级显示
    date_hierarchy = 'creation_time'
    # 多对多字段的水平过滤器
    filter_horizontal = ('tags',)
    # 排除的字段
    exclude = ('creation_time', 'last_modify_time')
    # 是否允许在站点上查看
    view_on_site = True
    # 批量操作
    actions = [
        makr_article_publish,
        draft_article,
        close_article_commentstatus,
        open_article_commentstatus]
    # 使用原始 ID 字段显示外键
    raw_id_fields = ('author', 'category',)

    # 自定义显示分类链接
    def link_to_category(self, obj):
        info = (obj.category._meta.app_label, obj.category._meta.model_name)
        link = reverse('admin:%s_%s_change' % info, args=(obj.category.id,))
        return format_html(u'<a href="%s">%s</a>' % (link, obj.category.name))

    link_to_category.short_description = _('category')

    # 自定义表单的查询集
    def get_form(self, request, obj=None, **kwargs):
        form = super(ArticlelAdmin, self).get_form(request, obj, **kwargs)
        # 限制作者字段仅显示超级用户
        form.base_fields['author'].queryset = get_user_model(
        ).objects.filter(is_superuser=True)
        return form

    # 保存模型时的自定义逻辑
    def save_model(self, request, obj, form, change):
        super(ArticlelAdmin, self).save_model(request, obj, form, change)

    # 自定义站点查看 URL
    def get_view_on_site_url(self, obj=None):
        if obj:
            # 如果对象存在，返回其完整 URL
            url = obj.get_full_url()
            return url
        else:
            # 否则返回当前站点的域名
            from djangoblog.utils import get_current_site
            site = get_current_site().domain
            return site


# 自定义标签管理界面
class TagAdmin(admin.ModelAdmin):
    # 排除的字段
    exclude = ('slug', 'last_mod_time', 'creation_time')


# 自定义分类管理界面
class CategoryAdmin(admin.ModelAdmin):
    # 列表显示的字段
    list_display = ('name', 'parent_category', 'index')
    # 排除的字段
    exclude = ('slug', 'last_mod_time', 'creation_time')


# 自定义链接管理界面
class LinksAdmin(admin.ModelAdmin):
    # 排除的字段
    exclude = ('last_mod_time', 'creation_time')


# 自定义侧边栏管理界面
class SideBarAdmin(admin.ModelAdmin):
    # 列表显示的字段
    list_display = ('name', 'content', 'is_enable', 'sequence')
    # 排除的字段
    exclude = ('last_mod_time', 'creation_time')


# 自定义博客设置管理界面
class BlogSettingsAdmin(admin.ModelAdmin):
    pass