mirror of
https://github.com/Dispatcharr/Dispatcharr.git
synced 2026-07-21 09:09:22 +00:00
70 lines
2.8 KiB
Python
70 lines
2.8 KiB
Python
from django.contrib import admin
|
|
from django.utils.html import format_html
|
|
from .models import M3UAccount, M3UFilter, ServerGroup, UserAgent
|
|
import json
|
|
|
|
class M3UFilterInline(admin.TabularInline):
|
|
model = M3UFilter
|
|
extra = 1
|
|
verbose_name = "M3U Filter"
|
|
verbose_name_plural = "M3U Filters"
|
|
|
|
@admin.register(M3UAccount)
|
|
class M3UAccountAdmin(admin.ModelAdmin):
|
|
list_display = ('name', 'server_url', 'server_group', 'max_streams', 'is_active', 'user_agent_display', 'vod_enabled_display', 'uploaded_file_link', 'created_at', 'updated_at')
|
|
list_filter = ('is_active', 'server_group', 'account_type')
|
|
search_fields = ('name', 'server_url', 'server_group__name')
|
|
inlines = [M3UFilterInline]
|
|
actions = ['activate_accounts', 'deactivate_accounts']
|
|
|
|
# Handle both ForeignKey and ManyToManyField cases for UserAgent
|
|
def user_agent_display(self, obj):
|
|
if hasattr(obj, 'user_agent'): # ForeignKey case
|
|
return obj.user_agent.user_agent if obj.user_agent else "None"
|
|
elif hasattr(obj, 'user_agents'): # ManyToManyField case
|
|
return ", ".join([ua.user_agent for ua in obj.user_agents.all()]) or "None"
|
|
return "None"
|
|
user_agent_display.short_description = "User Agent(s)"
|
|
|
|
def vod_enabled_display(self, obj):
|
|
"""Display whether VOD is enabled for this account"""
|
|
if obj.custom_properties:
|
|
try:
|
|
custom_props = json.loads(obj.custom_properties)
|
|
return "Yes" if custom_props.get('enable_vod', False) else "No"
|
|
except (json.JSONDecodeError, TypeError):
|
|
pass
|
|
return "No"
|
|
vod_enabled_display.short_description = "VOD Enabled"
|
|
vod_enabled_display.boolean = True
|
|
|
|
def uploaded_file_link(self, obj):
|
|
if obj.uploaded_file:
|
|
return format_html("<a href='{}' target='_blank'>Download M3U</a>", obj.uploaded_file.url)
|
|
return "No file uploaded"
|
|
uploaded_file_link.short_description = "Uploaded File"
|
|
|
|
@admin.action(description='Activate selected accounts')
|
|
def activate_accounts(self, request, queryset):
|
|
queryset.update(is_active=True)
|
|
|
|
@admin.action(description='Deactivate selected accounts')
|
|
def deactivate_accounts(self, request, queryset):
|
|
queryset.update(is_active=False)
|
|
|
|
# Add ManyToManyField for Django Admin (if applicable)
|
|
if hasattr(M3UAccount, 'user_agents'):
|
|
filter_horizontal = ('user_agents',) # Only for ManyToManyField
|
|
|
|
@admin.register(M3UFilter)
|
|
class M3UFilterAdmin(admin.ModelAdmin):
|
|
list_display = ('m3u_account', 'filter_type', 'regex_pattern', 'exclude')
|
|
list_filter = ('filter_type', 'exclude')
|
|
search_fields = ('regex_pattern',)
|
|
ordering = ('m3u_account',)
|
|
|
|
@admin.register(ServerGroup)
|
|
class ServerGroupAdmin(admin.ModelAdmin):
|
|
list_display = ('name',)
|
|
search_fields = ('name',)
|
|
|