from django.contrib import admin
from .models import Group, Membership, Post # 导入所有需要的模型

# 为 Membership 模型创建一个内联 Admin 类
class MembershipInline(admin.TabularInline):
    model = Membership
    extra = 1 # 默认显示一个空的额外行，方便添加新成员
    autocomplete_fields = ['user'] # 假设 CustomUserAdmin 中设置了 search_fields

# 为 Group 模型创建一个 Admin 类
class GroupAdmin(admin.ModelAdmin):
    list_display = ('name', 'creator', 'created_at', 'get_tags_display')
    list_filter = ('created_at', 'tags')
    search_fields = ('name', 'description', 'creator__email', 'creator__nickname')
    
    inlines = [MembershipInline] # 将 MembershipInline 嵌入
    
    filter_horizontal = ('tags',) # 为 tags 字段使用 filter_horizontal

    def get_tags_display(self, obj):
        return ", ".join([tag.name for tag in obj.tags.all()])
    get_tags_display.short_description = '相关标签'

# 为 Membership 模型创建一个 Admin 类 (可选，但方便单独管理)
class MembershipAdmin(admin.ModelAdmin):
    list_display = ('user_display', 'group', 'date_joined') # 假设 CustomUser 有 display_name
    list_filter = ('group', 'date_joined')
    search_fields = ('user__email', 'user__nickname', 'group__name')
    autocomplete_fields = ['user', 'group']

    def user_display(self, obj):
        # 确保你的 CustomUser 模型有一个 display_name 属性或方法
        # 如果没有，可以直接用 obj.user.email 或 obj.user.nickname
        if hasattr(obj.user, 'display_name'):
            return obj.user.display_name
        return obj.user.email # 或者其他你想显示的字段
    user_display.short_description = '用户'
    user_display.admin_order_field = 'user__email'


# 为 Post 模型创建一个 Admin 类
class PostAdmin(admin.ModelAdmin):
    list_display = ('title_or_author', 'group', 'author_display', 'created_at', 'short_content')
    list_filter = ('group', 'created_at', 'author')
    search_fields = ('title', 'content', 'author__email', 'author__nickname', 'group__name')
    autocomplete_fields = ['group', 'author'] # 为外键启用自动完成

    def title_or_author(self, obj):
        return obj.title if obj.title else f"来自 {self.author_display(obj)} 的动态"
    title_or_author.short_description = '标题/作者动态'

    def author_display(self, obj):
        if hasattr(obj.author, 'display_name'):
            return obj.author.display_name
        return obj.author.email # 或者其他你想显示的字段
    author_display.short_description = '作者'
    author_display.admin_order_field = 'author__email' # 允许按作者邮箱排序

    def short_content(self, obj):
        return obj.content[:50] + '...' if len(obj.content) > 50 else obj.content
    short_content.short_description = '内容预览'
# PostAdmin 类定义到此结束


# 注册所有模型到Admin站点
admin.site.register(Group, GroupAdmin)
admin.site.register(Membership, MembershipAdmin)
admin.site.register(Post, PostAdmin) # PostAdmin 的注册现在在类定义之后