addon/plugin.video.alfa/channels/doomtv.py

# -*- coding: utf-8 -*-

import re
import urllib
import urlparse

from channels import autoplay
from channels import filtertools
from core import httptools
from core import jsontools
from core import scrapertools
from core import servertools
from core import tmdb
from core.item import Item
from platformcode import config, logger
from channelselector import get_thumb

IDIOMAS = {'Latino': 'Latino'}
list_language = IDIOMAS.values()
list_quality = []
list_servers = ['dostream', 'openload']

host = 'http://doomtv.net/'


def mainlist(item):
    logger.info()

    autoplay.init(item.channel, list_servers, list_quality)

    itemlist = []

    itemlist.append(
        item.clone(title="Todas",
                   action="lista",
                   thumbnail=get_thumb('all', auto=True),
                   fanart='https://s18.postimg.cc/fwvaeo6qh/todas.png',
                   url='%s%s'%(host,'peliculas/'),
                   first=0

                   ))

    itemlist.append(
        item.clone(title="Generos",
                   action="seccion",
                   thumbnail=get_thumb('genres', auto=True),
                   fanart='https://s3.postimg.cc/5s9jg2wtf/generos.png',
                   url='%s%s' % (host, 'peliculas/'),
                   ))

    itemlist.append(
        item.clone(title="Mas Vistas",
                   action="lista",
                   thumbnail=get_thumb('more watched', auto=True),
                   fanart='https://s9.postimg.cc/wmhzu9d7z/vistas.png',
                   url='%s%s'%(host,'top-imdb/'),
                   first=0
                   ))

    itemlist.append(
        item.clone(title="Buscar",
                   action="search",
                   url='http://doomtv.net/?s=',
                   thumbnail=get_thumb('search', auto=True),
                   fanart='https://s30.postimg.cc/pei7txpa9/buscar.png'
                   ))

    autoplay.show_option(item.channel, itemlist)

    return itemlist


def get_source(url, referer=None):
    logger.info()
    if referer is None:
        data = httptools.downloadpage(url).data
    else:
        data = httptools.downloadpage(url, headers={'Referer':referer}).data
    data = re.sub(r'\n|\r|\t|&nbsp;|<br>|\s{2,}', "", data)
    return data

def lista(item):
    logger.info()

    itemlist = []
    next = False

    data = get_source(item.url)
    patron = 'movie-id=.*?href="([^"]+)" data-url.*?quality">([^<]+)<.*?img data-original="([^"]+)" class.*?'
    patron += '<h2>([^<]+)<\/h2>.*?<p>([^<]+)<\/p>'

    matches = re.compile(patron, re.DOTALL).findall(data)

    first = item.first
    last = first + 19
    if last > len(matches):
        last = len(matches)
        next = True

    for scrapedurl, quality, scrapedthumbnail, scrapedtitle, plot in matches[first:last]:

        url = host+scrapedurl
        thumbnail = 'https:'+scrapedthumbnail.strip()
        filtro_thumb = thumbnail.replace("https://image.tmdb.org/t/p/w185", "")
        filtro_list = {"poster_path": filtro_thumb.strip()}
        filtro_list = filtro_list.items()
        title = scrapedtitle
        fanart = ''
        plot = plot
        itemlist.append(
            Item(channel=item.channel,
                 action='findvideos',
                 title=title,
                 url=url,
                 thumbnail=thumbnail,
                 plot=plot,
                 infoLabels={'filtro': filtro_list},
                 fanart=fanart,
                 contentTitle=title
                 ))
    tmdb.set_infoLabels_itemlist(itemlist, seekTmdb = True)

    if not next:
        url_next_page = item.url
        first = last
    else:
        url_next_page = scrapertools.find_single_match(data, "<li class='active'>.*?class='page larger' href='([^']+)'")
        first = 0

    if url_next_page:
        itemlist.append(item.clone(title="Siguiente >>", url=url_next_page, action='lista', first=first))

    return itemlist


def seccion(item):
    logger.info()

    itemlist = []
    duplicado = []
    data = get_source(item.url)

    patron = 'menu-item-object-category menu-item-\d+"><a href="([^"]+)">([^<]+)<\/a><\/li>'

    matches = re.compile(patron, re.DOTALL).findall(data)

    for scrapedurl, scrapedtitle in matches:
        url = host+scrapedurl
        title = scrapedtitle
        thumbnail = ''
        if url not in duplicado:
            itemlist.append(
                Item(channel=item.channel,
                     action='lista',
                     title=title,
                     url=url,
                     thumbnail=thumbnail,
                     first=0
                     ))
    return itemlist


def search(item, texto):
    logger.info()
    texto = texto.replace(" ", "+")
    item.url = item.url + texto
    item.first=0
    if texto != '':
        return lista(item)


def newest(categoria):
    logger.info()
    itemlist = []
    item = Item()
    try:
        if categoria in ['peliculas', 'latino']:
            item.url = host +'peliculas/page/1'
        elif categoria == 'infantiles':
            item.url = host + 'categoria/animacion/'
        elif categoria == 'terror':
            item.url = host + '/categoria/terror/'
        item.first=0
        itemlist = lista(item)
        if itemlist[-1].title == 'Siguiente >>>':
            itemlist.pop()
    except:
        import sys
        for line in sys.exc_info():
            logger.error("{0}".format(line))
        return []

    return itemlist

def findvideos(item):
    logger.info()
    itemlist = []

    data = get_source(item.url)
    patron = 'id="(tab\d+)"><div class="movieplay">.*?src="([^"]+)"'
    matches = re.compile(patron, re.DOTALL).findall(data)

    for option, urls in matches:
        language = 'Latino'
        if 'http' not in urls:
            urls = 'https:'+urls
        if not config.get_setting('unify'):
            title = ' [%s]' % language
        else:
            title = '%s'
        new_item = Item(
                        channel=item.channel,
                        url=urls,
                        title= '%s'+ title,
                        contentTitle=item.title,
                        action='play',
                        language = IDIOMAS[language],
                        infoLabels = item.infoLabels
                        )
        itemlist.append(new_item)
    itemlist = servertools.get_servers_itemlist(itemlist, lambda i: i.title % i.server.capitalize())

    # Requerido para FilterTools

    itemlist = filtertools.get_links(itemlist, item, list_language)

    # Requerido para AutoPlay

    autoplay.start(itemlist, item)

    if config.get_videolibrary_support() and len(itemlist) > 0 and item.extra != 'findvideos':
        itemlist.append(
            Item(channel=item.channel,
                 title='[COLOR yellow]Añadir esta pelicula a la videoteca[/COLOR]',
                 url=item.url,
                 action="add_pelicula_to_library",
                 extra="findvideos",
                 contentTitle=item.contentTitle,
                 ))


    return itemlist