# -*- coding: utf-8 -*-
from urllib3.packages.six import ensure_text
import re
class PlaylistConfig():

    # Default playlist format
    m3uemptyheader = u'#EXTM3U\n'
    m3uheader = u'#EXTM3U deinterlace=1 m3uautoload=1 cache=1000\n'
    # If you need the #EXTGRP field put this #EXTGRP:{group}\n after {name}\n.
    m3uchanneltemplate = \
       u'#EXTINF:-1 group-title="{group}" tvg-name="{tvg}" tvg-id="{tvgid}" tvg-logo="{logo}",{name}\n#EXTGRP:{group}\n{url}\n'

    # Playlist sorting options.
    sortByName = False
    sortByGroup = True

    # Channel names mapping. You may use this to rename channels.
    m3uchannelnames = dict()
    # Examples:
    m3uchannelnames['Amedia 1'] = 'A1'
    m3uchannelnames['Amedia 2'] = 'A2'
    m3uchannelnames['Da Vinci Learning'] = 'Da Vinci'
    m3uchannelnames['SET'] = 'Sony channel'
    m3uchannelnames['SET HD'] = 'Sony channel HD'
    m3uchannelnames['History 2 HD'] = 'H2 HD'
    m3uchannelnames['5 канал'] = 'Пятый канал'
    m3uchannelnames['TV XXI (TV21)'] = 'TV XXI'
    m3uchannelnames['TV1000 Action East'] = 'TV 1000 Action'
    m3uchannelnames['TV 1000 Action East'] = 'TV 1000 Action'
    m3uchannelnames['TV1000 Русское кино'] = 'TV 1000 Русское кино'
    m3uchannelnames['Enter Film'] = 'Enter-фильм'
    m3uchannelnames['Кинопоказ 1 HD'] = 'Кинопоказ HD-1'
    m3uchannelnames['Кинопоказ 2 HD'] = 'Кинопоказ HD-2'
    m3uchannelnames['Travel+Adventure'] = 'Travel + adventure'
    m3uchannelnames['Travel + adventure HD'] ='Travel+Adventure HD'
    m3uchannelnames['HD Life'] = 'HDL'
    m3uchannelnames['ID Xtra'] = 'ID Investigation Discovery'
    m3uchannelnames['Первый'] = 'Первый канал'
    m3uchannelnames['ТВ3'] = 'ТВ 3'
    m3uchannelnames['КХЛ'] = 'КХЛ ТВ'
    m3uchannelnames['Канал Disney'] = 'Disney Channel'
    m3uchannelnames['Boomerаng TV'] = 'Boomerаng'
    m3uchannelnames['Nick Jr.'] = 'Nick Jr'
    m3uchannelnames['Бобёр']  = 'Бобер'
    m3uchannelnames['Наука'] = 'Наука 2.0'
    m3uchannelnames['Russian Travel Guide'] = 'RTG TV'
    m3uchannelnames['Иллюзион +'] = 'Иллюзион+'
    m3uchannelnames['РТВ - Любимое кино'] = 'Наше Любимое Кино'
    m3uchannelnames['ТВ Центр'] = 'ТВЦ'
    m3uchannelnames['VH1'] = 'VH1 European'
    m3uchannelnames['1 HD'] = '1HD Music Television'
    m3uchannelnames['1Music (Hungary)'] = '1 Music Channel (Hungary)'

    # Similar to m3uchannelnames but for groups
    m3ugroupnames = dict()
    m3ugroupnames['kids'] = 'Детские'
    m3ugroupnames['music'] = 'Музыка'
    m3ugroupnames['movies'] = 'Фильмы'
    m3ugroupnames['sport'] = 'Спорт'
    #m3ugroupnames[''] = 'Общие'
    m3ugroupnames['educational'] = 'Познавательные'
    m3ugroupnames['informational'] = 'Новостные'
    m3ugroupnames['entertaining'] = 'Развлекательные'
    m3ugroupnames['erotic'] = 'Эротика'
    m3ugroupnames['erotic_18_plus'] = 'Эротика'
    m3ugroupnames['18+'] = 'Эротика'
    #m3ugroupnames[''] = 'Мужские'
    m3ugroupnames['regional'] = 'Региональные'
    m3ugroupnames['religion'] = 'Религиозные'

    # Channel name to tvg name mappings.
    m3utvgnames = dict()
    # m3utvgnames['Channel name'] = 'Tvg_name'
    m3utvgnames['A1'] = 'Amedia 1'
    m3utvgnames['A2'] = 'Amedia 2'
    m3utvgnames['TV 1000'] = 'TV1000'
    m3utvgnames['TV 1000 HD'] = 'TV1000'
    m3utvgnames['TV 1000 Action'] = 'TV1000 Action'
    m3utvgnames['TV 1000 Action HD'] = 'TV1000 Action'
    m3utvgnames['TV 1000 Premium'] = 'TV1000 Premium Baltic'
    m3utvgnames['TV 1000 World Kino'] = 'TV1000 World Kino'
    m3utvgnames['TV 1000 Русское кино'] = 'TV1000 Русское кино'
    m3utvgnames['TV 1000 Русское кино HD'] = 'TV1000 Русское кино'
    m3utvgnames['ViP Comedy HD'] = 'ViP Comedy'
    m3utvgnames['ViP Megahit HD'] = 'ViP Megahit'
    m3utvgnames['ViP Premiere HD'] = 'ViP Premiere'
    m3utvgnames['ViP Serial HD'] = 'ViP Serial'
    m3utvgnames['Epic Drama HD'] = 'Epic Drama'
    m3utvgnames['Disney Channel (+2)'] = 'Disney +2'
    m3utvgnames['Disney Channel (+4)'] = 'Disney +4'
    m3utvgnames['Disney Channel (+7)'] = 'Disney +7'
    m3utvgnames['Карусель (+3)'] = 'Карусель +3'
    m3utvgnames['Карусель (+7)'] = 'Карусель +7'
    m3utvgnames['Малятко ТВ'] = 'Малятко TV'
    m3utvgnames['Пиксель ТВ'] = 'Пиксель'
    m3utvgnames['SET'] = 'Sony ТВ'
    m3utvgnames['SET HD'] = 'Sony ТВ HD'
    m3utvgnames['Sony channel'] = 'Sony ТВ'
    m3utvgnames['Sony channel HD'] = 'Sony ТВ HD'
    m3utvgnames['Sony Sci-Fi'] = 'Sony SCI-FI'
    m3utvgnames['A HBO HD'] = 'Amedia Premium HD'
    m3utvgnames['Star Cinema HD'] = 'Star Cinema'
    m3utvgnames['Amedia Hit HD'] = 'Amedia Hit'
    m3utvgnames['KinoTV Polska'] = 'Kino Polska'
    m3utvgnames['FilmBox Premium HD'] = 'FilmBox Premium Polska'
    m3utvgnames['Paramount Comedy HD (Россия)'] = 'Paramount Comedy'
    m3utvgnames['Zee TV Россия'] = 'Zee-TV'
    m3utvgnames['Кинохит HD'] = 'Кинохит'
    m3utvgnames['Дом кино Премиум'] = 'Дом кино Премиум'
    m3utvgnames['Кинопремьера HD'] = 'Кинопремьера'
    m3utvgnames['Мужское кино HD'] = 'Мужское кино'
    m3utvgnames['Наш кинороман HD'] = 'Наш кинороман'
    m3utvgnames['Наше любимое кино'] = 'Наше любимое'
    m3utvgnames['Иллюзион+'] = 'Иллюзион +'
    m3utvgnames['КиноПремиум HD'] = 'КиноПремиумHD'
    m3utvgnames['Кинопоказ HD-1'] = 'Шокирующее'
    m3utvgnames['Кинопоказ HD-2'] = 'Кинопоказ HD2'
    m3utvgnames['НСТ'] = 'Настоящее Страшное Телевидение'
    m3utvgnames['BOLT HD (UA)'] = 'BOLT'
    m3utvgnames['Шокирующее HD'] = 'Шокирующее'
    m3utvgnames['Классика Кино HD'] = 'Классика Кино'
    m3utvgnames['Пятый канал (+2)'] = 'Пятый канал +2'
    m3utvgnames['РЕН ТВ'] = 'Рен ТВ'
    m3utvgnames['РЕН ТВ HD'] = 'Рен ТВ'
    m3utvgnames['РЕН ТВ (+2)'] = 'Рен ТВ +2'
    m3utvgnames['Рен ТВ (+4)'] = 'Рен ТВ +4'
    m3utvgnames['ТВ 3'] = 'ТВ3'
    m3utvgnames['ТВ 3 (+2)'] = 'ТВ3 +2'
    m3utvgnames['ТВ 3 (+3)'] = 'ТВ3 +3'
    m3utvgnames['ТВ 3 (+4)'] = 'ТВ3 +4'
    m3utvgnames['ТВ 3 (+7)'] = 'ТВ3 +7'
    m3utvgnames['Россия 1 (+2)'] = 'Россия 1 +2'
    m3utvgnames['Россия 1 (+4)'] = 'Россия 1 +4'
    m3utvgnames['Россия 1 (+6)'] = 'Россия 1 +6'
    m3utvgnames['Россия 1 (+8)'] = 'Россия 1 +8'
    m3utvgnames['Домашний (+2)'] = 'Домашний +2'
    m3utvgnames['Че (+4)'] = 'Че +4'
    m3utvgnames['Че (+7)'] = 'Че +7'
    m3utvgnames['Че (+4)!'] = 'Че +4'
    m3utvgnames['HD Life'] = 'HDL'
    m3utvgnames['ID Xtra'] = 'ID Investigation Discovery'
    m3utvgnames['Авто 24'] = 'Авто24'
    m3utvgnames['Дайвинг ТВ HD'] = 'Дайвинг.TV'
    m3utvgnames['Animal Planet'] = 'Animal Planet Россия'
    m3utvgnames['Animal Planet HD'] = 'Animal Planet Россия HD'
    m3utvgnames['Discovery Channel'] = 'Discovery Россия'
    m3utvgnames['Discovery Channel HD'] = 'Discovery Россия HD'
    m3utvgnames['Discovery Historia Polska'] = 'Discovery Historia'
    m3utvgnames['History Channel'] = 'History'
    m3utvgnames['History Channel HD'] = 'History HD'
    m3utvgnames['ID Investigation Discovery'] = 'ID Xtra (Россия)'
    m3utvgnames['ID:Investigation Discovery Europe HD'] = 'ID Xtra (Европа)'
    m3utvgnames['HDL'] = 'HDL (HD Life)'
    m3utvgnames['Da Vinci'] = 'Da Vinci Learning Россия'
    m3utvgnames['H2 HD'] = 'H2'
    m3utvgnames['Russia Today Doc.'] = 'RT doc'
    m3utvgnames['Russia Today Doc HD'] = 'RT doc'
    m3utvgnames['Travel + adventure'] = 'Travel+Adventure'
    m3utvgnames['Viasat Explore'] = 'Viasat Explore Россия'
    m3utvgnames['Viasat Nature East'] = 'Viasat Nature'
    m3utvgnames['Viasat Nature-History HD'] = 'Viasat Nature/History HD'
    m3utvgnames['Загородный HD'] = 'Загородный'
    m3utvgnames['Загородный International HD'] = 'Загородный Int'
    m3utvgnames['Зоо ТВ'] = 'Zoo TV (Зоо ТВ)'
    m3utvgnames['Зима'] = 'Зима ТВ (Чебоксары)'
    m3utvgnames['Драйв ТВ'] = 'Драйв'
    m3utvgnames['Охотник и рыболов Int HD'] = 'Охотник и Рыболов Международный'
    m3utvgnames['Русский экстрим HD'] = 'Русский экстрим'
    m3utvgnames['Нано ТВ'] = 'NANO'
    m3utvgnames['Загородный'] = 'Загородный HD'
    m3utvgnames['Россия К'] = 'Россия Культура'
    m3utvgnames['1 Music Channel (Hungary)'] = '1 Music Channel Hungary'
    m3utvgnames['1HD Music Television'] = '1 HD Music Television'
    m3utvgnames['1 MUSIC CHANNEL'] = '1 Music Channel (Romania)'
    m3utvgnames['VH1 European'] = 'VH1 Europe'
    m3utvgnames['MTV Россия'] = 'MTV Russia'
    m3utvgnames['Music Box Gold'] = 'Musicbox Gold'
    m3utvgnames['Music Box RU'] = 'MusicBox TV'
    m3utvgnames['Music Box RU HD'] = 'MusicBox TV'
    m3utvgnames['Music Box UA'] = 'MusicBox UA'
    m3utvgnames['Music Box UA HD'] = 'MusicBox UA HD'
    m3utvgnames['MTV Россия HD'] = 'MTV Russia'
    m3utvgnames['Шансон ТВ'] = 'Шансон-ТВ'
    m3utvgnames['5 канал (Украина)'] = '5 канал Украина'
    m3utvgnames['24 Украина'] = '24 (Телеканал новостей 24)'
    m3utvgnames['UA: Буковина'] = 'UA:Буковина'
    m3utvgnames['UA: Одесса'] = 'UA:Одеса'
    m3utvgnames['UA: Крим'] = 'UA:Крим'
    m3utvgnames['UA: Культура'] = 'UA:Культура'
    m3utvgnames['UA: Перший'] = 'UA:Перший'
    m3utvgnames['UA: TV'] = 'UA:TV'
    m3utvgnames['UA: Житомир'] = 'UA:Житомир'
    m3utvgnames['Беларусь 4 Могилёв'] = 'Беларусь 4 Могилев'
    m3utvgnames['8 канал'] = '8 канал (Триколор)'
    m3utvgnames['Матч ТВ'] = 'Матч!'
    m3utvgnames['Матч ТВ HD'] = 'Матч!'
    m3utvgnames['Матч! Футбол 1 Резерв 1'] = 'Матч! Футбол 1'
    m3utvgnames['Матч! Футбол 1 Резерв 2'] = 'Матч! Футбол 1'
    m3utvgnames['Матч! Футбол 1 Резерв 3'] = 'Матч! Футбол 1'
    m3utvgnames['Матч! Футбол 1 HD Резерв 1'] = 'Матч! Футбол 1 HD'
    m3utvgnames['Матч! Футбол 1 HD Резерв 2'] = 'Матч! Футбол 1 HD'
    m3utvgnames['Матч! Футбол 1 HD Резерв 3'] = 'Матч! Футбол 1 HD'
    m3utvgnames['Футбoл'] = 'Футбол Россия'
    m3utvgnames['Футбoл HD'] = 'Футбол Россия'
    m3utvgnames['Первый канал (4:3)'] = 'Первый канал'
    m3utvgnames['Первый канал (+2)'] = 'Первый канал +2'
    m3utvgnames['Первый канал (СНГ)'] = 'Первый канал СНГ'
    m3utvgnames['Первый канал Евразия'] = 'Первый канал - Евразия'
    m3utvgnames['Правда тут'] = 'ПравдаТУТ'
    m3utvgnames['РЕН ТВ Балтийский'] = 'РЕН ТВ Baltic'
    m3utvgnames['Россия HD'] = 'Россия 1 HD'
    m3utvgnames['Мир Премиум HD'] = 'Мир Premium (Мир HD)'
    m3utvgnames['СКИФ (Беларусь)'] = 'СКИФ Витебск'
    m3utvgnames['ТВЦ'] = 'ТВ Центр'
    m3utvgnames['ТВ Центр (+4)'] = 'ТВ Центр +4'
    m3utvgnames['ТВ Центр Международный'] = 'ТВ Центр Международный (TVCI)'
    m3utvgnames['ТВ 3 (Беларусь)'] = 'ТВ3 Беларусь'
    m3utvgnames['11 канал HD (Пенза)'] = '11 канал Наш Дом (Пенза)'
    m3utvgnames['3+ (Латвия)'] = '3+ Latvia'
    m3utvgnames['3+ (Estonia)'] = '3+ Estonia'
    m3utvgnames['360 градусов'] = '360° Подмосковье'
    m3utvgnames['360 градусов HD'] = '360° Подмосковье HD'
    m3utvgnames['360 Новости'] = '360° Новости'
    m3utvgnames['4 канал HD (Киев)'] = '4 канал Украина (RTi)'
    m3utvgnames['7 канал (Казахстан)'] = '7 канал Казахстан'
    m3utvgnames['7 канал (Одесса)'] = '7 канал Одесса'
    m3utvgnames['8 канал (Беларусь)'] = '8 канал Беларусь'
    m3utvgnames['8 канал (Красноярск)'] = '8 канал Красноярский край'
    m3utvgnames['Armenia 1 Satellite'] = 'Armenia 1 TV Satellite'
    m3utvgnames['Arte Network'] = 'ARTE'
    m3utvgnames['ATV Azerbaycan'] = 'AzTV'
    m3utvgnames['ATV HD'] = 'AzTV'
    m3utvgnames['Blue Hustler'] = 'Blue Hustler Россия'
    m3utvgnames['Brazzers TV Europe'] = 'Brazzers TV Europe (Россия)'
    m3utvgnames['MULTIMANIA TV'] = 'Мультимания'

    # Filtering rules
    # Empty dict skips filtering, all the channels are "whitelisted".
    # Filter items are iterated in alphabetical order.
    # The channel and group names are matched against case-insensitive regular expressions.
    # It is a good idea to specify massive drop at the beginning to minimize the amount of further processing.
    # The very last filter item in the example below will match and drop everything that wasn't matched before.
    m3ufilter = {}
    #m3ufilter = {
    #    '10-adult': {
    #        'allowed': False,
    #        'group': [
    #            "Erot.*",
    #            "Эрот.*",
    #            "18+?",
    #            "Relig.*",
    #            "Религ.*",
    #        ],
    #    },
    #    '20-mult': {
    #        'allowed': True,
    #        'group': [
    #            "Entert.*",
    #            "Child.*",
    #            "Kids",
    #            "Дет.*",
    #            "Без категории",
    #            "",
    #        ],
    #        'channel': [
    #            "Cartoon.*",
    #            "Disney.*",
    #            "2x2.*"
    #        ],
    #    },
    #    '30-sci': {
    #        'allowed': True,
    #        "channel": [
    #            "Discovery.*",
    #            "Nat.*Geo.*",
    #        ],
    #    },
    #    '99-default': {
    #        'allowed': False,
    #        'channel': [".*"],
    #    },
    #}

    @staticmethod
    def prepareFilter():
        prepared = []
        for filter in sorted(PlaylistConfig.m3ufilter.keys()):
            item = {
                'allowed': False
            }
            if PlaylistConfig.m3ufilter[filter]['allowed'] == True:
                item['allowed'] = True
            if 'group' in PlaylistConfig.m3ufilter[filter]:
                item['group'] = "(" + ")|(".join(PlaylistConfig.m3ufilter[filter]['group']) + ")"
            if 'channel' in PlaylistConfig.m3ufilter[filter]:
                item['channel'] = "(" + ")|(".join(PlaylistConfig.m3ufilter[filter]['channel']) + ")"
            prepared.append(item)
        return prepared

    @staticmethod
    def filterItem(item, filter):
        for each_filter in filter:
            proceedWithChannel = False
            if 'group' in each_filter:
                if re.match(each_filter['group'], item['group'] or '', re.IGNORECASE):
                    proceedWithChannel = True
                # don't check channel if group specified but not matched
            else:
                proceedWithChannel = True

            if proceedWithChannel:
                if 'channel' in each_filter:
                    if re.match(each_filter['channel'], item['name'] or '', re.IGNORECASE) or re.match(each_filter['channel'], item['tvg'] or '', re.IGNORECASE):
                        return each_filter['allowed']
                    # proceed to next filter if channel specified but not matched
                else:
                    return each_filter['allowed']

    # This comparator is used for the playlist sorting.
    @staticmethod
    def sortItems(itemlist):
        if PlaylistConfig.sortByGroup: return sorted(itemlist, key=lambda x:x['group'])
        elif PlaylistConfig.sortByName: return sorted(itemlist, key=lambda x:x['name'])
        return itemlist

    # This method can be used to change a channel info such as name, group etc.
    # The following fields can be changed:
    #
    #    name - channel name
    #    url - channel URL
    #    tvg - channel tvg name
    #    tvgid - channel tvg id
    #    group - channel group
    #    logo - channel logo
    @staticmethod
    def changeItem(item):
        PlaylistConfig._changeItemByDict(item, 'name', PlaylistConfig.m3uchannelnames)
        PlaylistConfig._changeItemByDict(item, 'group', PlaylistConfig.m3ugroupnames)
        PlaylistConfig._changeItemByDict(item, 'name', PlaylistConfig.m3utvgnames, 'tvg')

    @staticmethod
    def _changeItemByDict(item, key, replacementsDict, setKey=None):
        if len(replacementsDict) > 0:
           value = replacementsDict.get(item.get(key))
           if value: item[key if setKey is None else setKey] = ensure_text(value)

    xml_template = """<?xml version="1.0" encoding="utf-8"?>
    <items>
    <playlist_name>Playlist</playlist_name>

    %(items)s

    </items>
    """

    xml_channel_template = """
    <channel>
      <title><![CDATA[%(title)s]]></title>
      <description><![CDATA[<tr><td>%(description_title)s</td></tr>]]></description>
      <playlist_url>%(hostport)s%(url)s</playlist_url>
    </channel>
    """

    xml_stream_template = """
    <channel>
      <title><![CDATA[%(title)s]]></title>
      <description><![CDATA[<tr><td>%(description_title)s</td></tr>]></description>
      <stream_url><![CDATA[%(hostport)s%(url)s]]></stream_url>
    </channel>
    """