Spaces:
Sleeping
Sleeping
File size: 4,373 Bytes
35c63fd | 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 | from django.contrib import admin
from django.utils.html import format_html
from django.db import models
from .models import Project
class ProjectAdmin(admin.ModelAdmin):
"""项目管理"""
list_display = (
'name', 'user', 'source_lang', 'target_lang', 'status_display',
'progress_display', 'segment_count', 'completed_count', 'created_at'
)
list_filter = ('status', 'source_lang', 'target_lang', 'created_at', 'updated_at')
search_fields = ('name', 'user__username', 'user__email', 'description')
ordering = ('-created_at',)
date_hierarchy = 'created_at'
fieldsets = (
('基本信息', {
'fields': ('name', 'user', 'description')
}),
('语言配置', {
'fields': ('source_lang', 'target_lang')
}),
('文件信息', {
'fields': ('srt_file_path', 'video_file_path', 'concatenated_audio_url'),
'classes': ('collapse',)
}),
('TTS配置', {
'fields': ('tts_model', 'voice_mappings', 'custom_vocabulary', 'max_speed'),
'classes': ('collapse',)
}),
('统计信息', {
'fields': ('status', 'segment_count', 'completed_segment_count', 'progress_percentage', 'created_at', 'updated_at'),
'classes': ('collapse',)
}),
)
readonly_fields = ('segment_count', 'completed_segment_count', 'progress_percentage', 'created_at', 'updated_at')
def status_display(self, obj):
"""彩色显示状态"""
colors = {
'pending': '#ffc107',
'processing': '#17a2b8',
'completed': '#28a745',
'failed': '#dc3545'
}
color = colors.get(obj.status, '#6c757d')
return format_html(
'<span style="color: {}; font-weight: bold;">{}</span>',
color, obj.get_status_display()
)
status_display.short_description = "状态"
def progress_display(self, obj):
"""进度条显示"""
progress = obj.progress_percentage or 0
color = '#28a745' if progress == 100 else '#17a2b8' if progress > 50 else '#ffc107'
return format_html(
'<div style="width: 100px; background: #e9ecef; border-radius: 3px;">'
'<div style="width: {}%; background: {}; height: 20px; border-radius: 3px; text-align: center; color: white; font-size: 12px; line-height: 20px;">'
'{}%</div></div>',
progress, color, progress
)
progress_display.short_description = "进度"
def completed_count(self, obj):
"""显示完成的段落数"""
return f"{obj.completed_segment_count}/{obj.segment_count}"
completed_count.short_description = "完成情况"
def get_queryset(self, request):
"""优化查询性能"""
return super().get_queryset(request).select_related('user').prefetch_related('segments')
actions = ['mark_as_pending', 'mark_as_processing', 'mark_as_completed']
def mark_as_pending(self, request, queryset):
"""批量标记为待处理"""
updated = queryset.update(status='pending')
self.message_user(request, f'成功将 {updated} 个项目标记为待处理')
mark_as_pending.short_description = "标记为待处理"
def mark_as_processing(self, request, queryset):
"""批量标记为处理中"""
updated = queryset.update(status='processing')
self.message_user(request, f'成功将 {updated} 个项目标记为处理中')
mark_as_processing.short_description = "标记为处理中"
def mark_as_completed(self, request, queryset):
"""批量标记为已完成"""
updated = queryset.update(status='completed')
self.message_user(request, f'成功将 {updated} 个项目标记为已完成')
mark_as_completed.short_description = "标记为已完成"
def has_add_permission(self, request):
"""项目应通过前端页面创建，不允许在管理后台添加"""
return False
def get_urls(self):
"""移除添加URL"""
urls = super().get_urls()
# 过滤掉添加相关的URL
filtered_urls = []
for url in urls:
if hasattr(url, 'name') and url.name and 'add' in url.name:
continue
filtered_urls.append(url)
return filtered_urls
|