File size: 3,906 Bytes
b4f2238 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 | """
测试功能 Admin
提供在 Admin 后台测试登录、立案、查询等功能
"""
import logging
from django.contrib import admin
from django.shortcuts import redirect
from django.template.response import TemplateResponse
from django.urls import path
from django.utils.html import format_html
from apps.automation.models import TestCourt
from apps.automation.services.scraper.test_service import TestService
logger = logging.getLogger("apps.automation")
def _get_test_service():
"""工厂函数：创建测试服务实例"""
return TestService()
def _get_organization_service():
"""工厂函数：创建组织服务实例"""
from apps.core.dependencies.business_organization import build_organization_service
return build_organization_service()
@admin.register(TestCourt)
class TestCourtAdmin(admin.ModelAdmin):
"""
测试法院系统 Admin
使用 TestCourt 作为占位模型
提供测试功能的入口，支持 Token 捕获
"""
def has_module_perms(self, request) -> bool: # type: ignore[override]
return False
def get_model_perms(self, request):
"""返回空字典，隐藏所有权限"""
return {}
# 自定义列表页
def changelist_view(self, request, extra_context=None):
"""自定义列表页 - 显示测试选项"""
organization_service = _get_organization_service()
# 获取所有账号凭证
credentials = organization_service.get_all_credentials()
context = {
"title": "测试法院系统",
"credentials": credentials,
"opts": self.model._meta,
"has_view_permission": self.has_view_permission(request),
"site_header": self.admin_site.site_header,
"site_title": self.admin_site.site_title,
}
return TemplateResponse(request, "admin/automation/test_court_list.html", context)
def has_add_permission(self, request):
"""禁用添加功能"""
return False
def has_delete_permission(self, request, obj=None):
"""禁用删除功能"""
return False
def has_change_permission(self, request, obj=None):
"""禁用修改功能"""
return False
def get_urls(self):
"""添加自定义 URL"""
urls = super().get_urls()
custom_urls = [
path(
"test-login/<int:credential_id>/",
self.admin_site.admin_view(self.test_credential_login_view),
name="automation_testcourt_test_login",
),
]
return custom_urls + urls
def test_credential_login_view(self, request, credential_id):
"""
测试账号凭证登录
Admin 层只负责：
1. 参数验证
2. 调用 Service
3. 渲染结果
"""
from django.contrib import messages
# 1. 获取凭证（用于显示）
try:
organization_service = _get_organization_service()
credential = organization_service.get_credential(credential_id)
except Exception:
logger.warning("凭证 ID %s 不存在或获取失败", credential_id, exc_info=True)
messages.error(request, f"凭证 ID {credential_id} 不存在")
return redirect("admin:organization_accountcredential_changelist")
# 2. 调用 Service 执行测试
test_service = _get_test_service()
result = test_service.test_login(credential_id)
# 3. 渲染结果页面
context = {
"title": f"测试登录: {credential.site_name}",
"credential": credential,
"result": result,
"site_header": self.admin_site.site_header,
"site_title": self.admin_site.site_title,
}
return TemplateResponse(request, "admin/automation/test_court_result.html", context)
|