from django.contrib import admin
from simpleui.admin import AjaxAdmin

from pr import models


@admin.register(models.AIConfig)
class AIConfigAdmin(AjaxAdmin):
    """Admin配置"""

    list_display = ["api_base", "api_key", "llm_model"]
    readonly_fields = ["create_by", "delete_at", "detail"]
    top_html = ' <el-alert title="可配置多个AI模型厂商!" type="success"></el-alert>'

    def save_model(self, request, obj, form, change):
        obj.create_by = request.user.username
        return super().save_model(request, obj, form, change)


@admin.register(models.GitConfig)
class GitConfigAdmin(AjaxAdmin):
    """Admin配置"""

    list_display = ["git_name", "git_type", "git_url", "access_token"]
    readonly_fields = ["create_by", "delete_at", "detail"]
    top_html = '<el-alert title="可配置多个Git服务上!" type="success"></el-alert>'

    def save_model(self, request, obj, form, change):
        obj.create_by = request.user.username
        return super().save_model(request, obj, form, change)


@admin.register(models.ProjectConfig)
class ProjectConfigAdmin(AjaxAdmin):
    """Admin配置"""

    list_display = ["project_id", "project_name", "project_secret", "commands", "is_enable"]
    readonly_fields = ["create_by", "delete_at", "detail"]
    top_html = '<el-alert title="可配置多个项目!" type="success"></el-alert>'

    def save_model(self, request, obj, form, change):
        obj.create_by = request.user.username
        return super().save_model(request, obj, form, change)