Приводим в порядок плейлист Торрент-ТВ («Суперпомойка»)05.02.2019 18:49

PLAYLIST_LOAD_URL = "http://91.92.66.82/trash/ttv-list/ttv.all.tag.player.m3u"
TEMPLATE_SAVE_PATH = "/opt/etc/ttv/template.txt"
FAVORITES_LOAD_PATH = "/opt/etc/ttv/favorites.txt"
PLAYLIST_SAVE_PATH = "/opt/etc/ttv/playlist.m3u"
LOGOS_URL = "https://raw.githubusercontent.com/AlexELEC/channel-logos/master/logos/{}.png"
#LOGOS_URL = "{}.png"
STREAM_URL = "http://127.0.0.1:6878/ace/getstream?id={}&.mp4"
#STREAM_URL = "acestream://{}"
EPG_LINKS = "https://teleguide.info/download/new3/xmltv.xml.gz"
#EPG_LINKS = "https://teleguide.info/download/new3/xmltv.xml.gz,http://programtv.ru/xmltv.xml.gz,http://api.torrent-tv.ru/ttv.xmltv.xml.gz"

import re
import urllib.request
import os
from operator import itemgetter as i
from functools import cmp_to_key
from urllib.parse import urlencode

def cmp(a, b):
        return (a > b) - (a < b) 
        
def multikeysort(items, columns):
        comparers = [
                ((i(col[1:].strip()), -1) if col.startswith('-') else (i(col.strip()), 1))
                for col in columns
        ]
        def comparer(left, right):
                comparer_iter = (
                        cmp(fn(left), fn(right)) * mult
                        for fn, mult in comparers
                )
                return next((result for result in comparer_iter if result), 0)
        return sorted(items, key=cmp_to_key(comparer))
        
def loadChannels(content):
        lines = content.splitlines()
        
        returnChannels = {}
        
        pattern = re.compile("group-title=\"(.*?)\"")
        channelName = ""
        channelGroup = ""
        channelStreamID = ""
        waitURI = False
        
        for line in lines:
                if line.startswith("acestream"):
                        if waitURI:
                                channelStreamID = line[12:]
                                HD = False
                                if "HD" in channelName:
                                        HD = True
                                tmpDict = {"name": channelName, "group": channelGroup, "stream": channelStreamID, "hd": HD}
                                returnChannels[channelName] = tmpDict
                                waitURI = False
                elif line.startswith("#EXTINF"):
                        x = line.split("\",")
                        if (len(x) != 2):
                                continue
                        channelName = x[1]
                        match = pattern.search(x[0])
                        if match:
                                channelGroup = match.group(1)
                        else:
                                channelGroup = "Общие"
                        waitURI = True
        return returnChannels
        
def saveTemplate(content, channels, path):
        lines = content.splitlines()
        pattern = re.compile("group-title=\"(.*?)\"")
        
        waitURI = False
        
        channelName = ""
        channelReplace = ""
        channelNewName = ""
        channelEPG = ""
        channelGroup = ""
        channelStreamID = ""
        groupDict = {}
        currentGroup = 1
        template=""
        for line in lines:
                if line.startswith("acestream"):
                        if waitURI:
                                channelStreamID = line[12:]
                                template += channelName
                                template += "/"
                                template += channelReplace
                                template += "/"
                                template += channelNewName
                                template += "/"
                                template += channelEPG
                                template += "/"
                                template += channelGroup
                                template += "\n"
                                waitURI = False
                elif line.startswith("#EXTINF"):
                        x = line.split("\",")
                        if (len(x) != 2):
                                continue
                        channelName = x[1]
                        channelNewName = x[1]
                        channelEPG = x[1]
                        if (channels.get(channelName + " HD") != None):
                                channelReplace = channelName + " HD"
                        else:
                                channelReplace = "-"
                        match = pattern.search(x[0])
                        if match:
                                channelGroup = match.group(1)
                                if (groupDict.get(channelGroup)):
                                        channelGroup = groupDict.get(channelGroup)
                                else:
                                        newGroupName = "{:02d}_{}".format(currentGroup, channelGroup)
                                        currentGroup += 1
                                        groupDict[channelGroup] = newGroupName
                                        channelGroup = newGroupName
                        else:
                                channelGroup = "00_Unsigned"
                        waitURI = True
        file = open(path,'w', encoding='utf-8')
        file.write(template)
        file.close()
        return

def loadFavorites(content):
        returnChannels = {}
        lines = content.splitlines()
        for line in lines:
                parts = line.split('/')
                if len(parts) == 5:
                        tmpDict = {"name": parts[0], "replace": parts[1], "newName": parts[2], "EPG": parts[3], "group": parts[4]}
                        returnChannels[parts[0]] = tmpDict
        return returnChannels

def savePlaylist(channels, favorites, path):
        returnChannels = []
        currentChannels = set()
        for key, chDict in favorites.items():
                if chDict["replace"] != "-":
                        if favorites.get(chDict["replace"]) != None and channels.get(chDict["replace"]) != None:
                                currentChannels.add(chDict["replace"])
                        elif channels.get(chDict["name"]) != None:
                                currentChannels.add(chDict["name"])
                elif channels.get(chDict["name"]) != None:
                        currentChannels.add(chDict["name"])
        for ch in currentChannels:
                chFromFavorites = favorites.get(ch)
                chFromChannels = channels.get(ch)
                tmpDict = {"name": chFromFavorites.get("newName"), "oldName": chFromFavorites.get("name"), "EPG": chFromFavorites.get("EPG"), "group": chFromFavorites.get("group"), "stream": chFromChannels.get("stream"), "hd": chFromChannels.get("hd")}
                returnChannels.append(tmpDict)
        result = multikeysort(returnChannels, ['group', '-hd', 'name'])
        template=""
        template += "#EXTM3U url-tvg="
        template += "\""
        template += EPG_LINKS
        template += "\""
        template += "\n"
        for n in result:
                group = n.get("group")
                if group.find("_", 2, 3) != -1:
                        group = group[3:]
                template += "#EXTINF:-1 tvg-name=\"{}\" tvg-logo=\"{}\" group-title=\"{}\" ,{}".format(n.get("EPG"), LOGOS_URL.format(urllib.parse.quote(n.get("oldName"))), group, n.get("name"))
                template += "\n"
                template += STREAM_URL.format(n.get("stream"))
                template += "\n"
        file = open(path,'w', encoding='utf-8')
        file.write(template)
        file.close()
        return result
        
response = urllib.request.urlopen(PLAYLIST_LOAD_URL)
content = response.read().decode("utf-8")
channels = loadChannels(content)

if channels == None or (len(channels.keys()) == 0):
        exit()

saveTemplate(content, channels, TEMPLATE_SAVE_PATH)

exists = os.path.isfile(FAVORITES_LOAD_PATH)

if exists:
        file = open(FAVORITES_LOAD_PATH,'r', encoding='utf-8')
        content = file.read()
        favorites = loadFavorites(content)
        savePlaylist(channels, favorites, PLAYLIST_SAVE_PATH)

#!/bin/sh
PATH=/opt/sbin:/opt/bin:/opt/usr/sbin:/opt/usr/bin:/usr/sbin:/usr/bin:/sbin:/bin

python3 /opt/etc/ttv/ttv.py
echo "Content-Type: text/plain; charset=UTF-8"
echo ""
echo "$(cat /opt/etc/ttv/playlist.m3u)"

Amedia Premium/Amedia Premium HD/Amedia Premium/Amedia Premium/11_Фильмы и сериалы
Amedia Premium HD/-/Amedia Premium HD/Amedia Premium HD/11_Фильмы и сериалы
Дождь/Дождь HD/Дождь/Дождь/12_Общие
Дождь HD/-/Дождь HD/Дождь HD/12_Общие
Viasat History/-/Viasat History/Viasat History/01_Познавательные
Discovery Channel/Discovery Channel HD/Discovery Channel/Discovery Channel/01_Познавательные
Discovery Channel HD/-/Discovery Channel HD/Discovery Channel HD/01_Познавательные
Discovery Science/Discovery Science HD/Discovery Science/Discovery Science/01_Познавательные
Discovery Science HD/-/Discovery Science HD/Discovery Science HD/01_Познавательные
Amedia Hit/Amedia Hit HD/Amedia Hit/Amedia Hit/11_Фильмы и сериалы
Amedia Hit HD/-/Amedia Hit HD/Amedia Hit HD/11_Фильмы и сериалы
Матч ТВ/Матч ТВ HD/Матч!/Матч!/03_Спортивные
Матч ТВ HD/-/Матч! HD/Матч!/03_Спортивные
Eurosport 1/Eurosport 1 HD/Eurosport 1/Eurosport 1/03_Спортивные
Eurosport 1 HD/-/Eurosport 1 HD/Eurosport 1 HD/03_Спортивные
Paramount Comedy HD (Россия)/-/Paramount Comedy HD/Paramount Comedy HD (Россия)/11_Фильмы и сериалы
Матч! Футбол 1/Матч! Футбол 1 HD/Матч! Футбол 1/Матч! Футбол 1/03_Спортивные
Матч! Футбол 1 HD/-/Матч! Футбол 1 HD/Матч! Футбол 1 HD/03_Спортивные
Россия 1/Россия HD/Россия 1/Россия 1/13_Зомби-пропаганда
Россия HD/-/Россия HD/Россия HD/13_Зомби-пропаганда