From 2108a9570f13a8d9bd07683024f6235ce35f0b42 Mon Sep 17 00:00:00 2001 From: dekzter Date: Thu, 3 Apr 2025 15:41:19 -0400 Subject: [PATCH] support custom logos --- apps/output/views.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/apps/output/views.py b/apps/output/views.py index d4e7f282..20a749a3 100644 --- a/apps/output/views.py +++ b/apps/output/views.py @@ -4,6 +4,7 @@ from apps.channels.models import Channel, ChannelProfile from apps.epg.models import ProgramData from django.utils import timezone from datetime import datetime, timedelta +import re def generate_m3u(request, profile_name=None): """ @@ -94,13 +95,14 @@ def generate_epg(request, profile_name=None): # Add channel logo if available if channel.logo: - # Use the full URL for the logo - base_url = request.build_absolute_uri('/')[:-1] logo_url = channel.logo.url # Convert to absolute URL if it's relative - if logo_url.startswith('/'): - logo_url = f"{base_url}{logo_url}" + if logo_url.startswith('/data'): + # Use the full URL for the logo + logo_uri = re.sub(r"^\/data", '', logo_url) + base_url = request.build_absolute_uri('/')[:-1] + logo_url = f"{base_url}{logo_uri}" xml_lines.append(f' ')