58 lines
1.6 KiB
Python
Executable File
58 lines
1.6 KiB
Python
Executable File
#!/usr/bin/python
|
|
"""Channel management for IPMPV."""
|
|
|
|
import re
|
|
import requests
|
|
import sys
|
|
from utils import m3u_url
|
|
|
|
def get_channels():
|
|
"""
|
|
Get a list of channels from the M3U playlist.
|
|
|
|
Returns:
|
|
list: A list of channel dictionaries with name, url, logo, and group.
|
|
"""
|
|
if m3u_url:
|
|
try:
|
|
response = requests.get(m3u_url)
|
|
response.raise_for_status() # Raise exception for HTTP errors
|
|
except requests.RequestException as e:
|
|
print(f"Error fetching M3U playlist: {e}")
|
|
return []
|
|
else:
|
|
print("Error: IPMPV_M3U_URL not set. Please set this environment variable to the URL of your IPTV list, in M3U format.")
|
|
sys.exit(1)
|
|
|
|
lines = response.text.splitlines()
|
|
|
|
channels = []
|
|
regex = re.compile(r'tvg-logo="(.*?)".*?group-title="(.*?)"', re.IGNORECASE)
|
|
|
|
for i in range(len(lines)):
|
|
if lines[i].startswith("#EXTINF"):
|
|
match = regex.search(lines[i])
|
|
logo = match.group(1) if match else ""
|
|
group = match.group(2) if match else "Other"
|
|
name = lines[i].split(",")[-1]
|
|
url = lines[i + 1]
|
|
|
|
channels.append({"name": name, "url": url, "logo": logo, "group": group})
|
|
|
|
return channels
|
|
|
|
def group_channels(channels):
|
|
"""
|
|
Group channels by their group title.
|
|
|
|
Args:
|
|
channels (list): List of channel dictionaries.
|
|
|
|
Returns:
|
|
dict: Dictionary of channel groups.
|
|
"""
|
|
grouped_channels = {}
|
|
for channel in channels:
|
|
grouped_channels.setdefault(channel["group"], []).append(channel)
|
|
return grouped_channels
|