def parse(data, content_type=None):
import xbmc
import xml.etree.ElementTree as ET
from itertools import izip_longest
from concurrent import futures
from contextlib import nested, closing
from streamajoker.utils import SafeDialogProgress, get_quality_from_name, get_show_info_from_name, normalize_release_tags
from streamajoker import tmdb
root = ET.fromstring(data)
def _text(node, path):
n = node.find(path)
if n is not None:
return n.text
def _attr(node, path, attrib):
n = node.find(path)
if n is not None:
return n.attrib.get(attrib)
items = []
for node in root.getiterator("item"):
item = {
"title": _text(node, "title"),
"description": _text(node, "description"),
"category": _text(node, "category"),
"pub_date": _text(node, "pubDate"),
"seeds": _text(node, ".//{%(torrent)s}seeds" % NSMAP) or _text(node, "numSeeders") or _text(node, "seeders"),
"peers": _text(node, ".//{%(torrent)s}peers" % NSMAP) or _text(node, "numLeechers") or _text(node, "leechers"),
"content_length": _text(node, ".//{%(torrent)s}contentLength" % NSMAP) or _attr(node, ".//enclosure", "length"),
"href": _text(node, ".//{%(torrent)s}magnetURI" % NSMAP) or _attr(node, ".//enclosure", "url") or _text(node, "./link"),
"filename": _text(node, ".//{%(torrent)s}fileName" % NSMAP),
}
find_image(item)
check_imdb_id(item)
items.append(item)
tmdb_list = []
if content_type:
with closing(SafeDialogProgress(delay_close=0)) as dialog:
dialog.create(plugin.name)
dialog.update(percent=0, line1="Fetching torrent information...", line2="", line3="")
with futures.ThreadPoolExecutor(max_workers=POOL_WORKERS) as pool:
futures = []
for item in items:
if item.get("imdb_id"):
futures.append(pool.submit(tmdb.get, item["imdb_id"]))
else:
futures.append(None)
state = {"done": 0}
def on_item(future):
state["done"] += 1
dialog.update(
percent=int(state["done"] * 100.0 / len(filter(None, futures))),
)
[future.add_done_callback(on_item) for future in futures if future]
while not all(future.done() for future in futures if future):
if dialog.iscanceled():
return
xbmc.sleep(100)
tmdb_list = [future and future.result() or None for future in futures]
for item, tmdb_data in izip_longest(items, tmdb_list):
if tmdb_data:
list_item = tmdb.get_list_item(tmdb_data)
release_tags = normalize_release_tags(item["title"], list_item["label"])
if release_tags:
list_item["label"] = "%s (%s)" % (list_item["label"], release_tags)
else:
list_item = {
"label": item["title"],
"icon": item.get("img") or "",
"thumbnail": item.get("img") or "",
"info": {
"genre": item["category"],
}
}
list_item.update({
"path": plugin.url_for("play", uri=item["href"]),
"is_playable": True,
})
list_item.setdefault("info", {}).update({
"genre": "%s (S:%s P:%s)" % (list_item.get("info", {}).get("genre") or "", item["seeds"], item["peers"]),
})
list_item.setdefault("stream_info", {}).update(get_quality_from_name(item["title"]))
yield list_item
def yify_show_data(callback):
import xbmc
import xbmcgui
from contextlib import nested, closing
from itertools import izip, chain
from concurrent import futures
from streamajoker import tmdb
from streamajoker.utils import url_get_json, terminating, SafeDialogProgress
plugin.set_content("movies")
args = dict((k, v[0]) for k, v in plugin.request.args.items())
current_page = int(args["set"])
limit = int(args["limit"])
with closing(SafeDialogProgress(delay_close=0)) as dialog:
dialog.create(plugin.name)
dialog.update(percent=0, line1="Fetching movie information...", line2="", line3="")
try:
search_result = url_get_json("%s/api/list.json" % BASE_URL, params=args, headers=HEADERS)
except:
plugin.notify("Unable to connect to %s." % BASE_URL)
raise
movies = search_result.get("MovieList") or []
if not movies:
return
state = {"done": 0}
def on_movie(future):
data = future.result()
state["done"] += 1
dialog.update(
percent=int(state["done"] * 100.0 / len(movies)),
line2=data.get("title") or data.get("MovieTitleClean") or "",
)
with futures.ThreadPoolExecutor(max_workers=2) as pool_tmdb:
tmdb_list = [pool_tmdb.submit(tmdb.get, movie["ImdbCode"]) for movie in movies]
[future.add_done_callback(on_movie) for future in tmdb_list]
while not all(job.done() for job in tmdb_list):
if dialog.iscanceled():
return
xbmc.sleep(100)
tmdb_list = map(lambda job: job.result(), tmdb_list)
for movie, tmdb_meta in izip(movies, tmdb_list):
if tmdb_meta:
item = tmdb.get_list_item(tmdb_meta)
if args.get("quality") == "all" and movie["Quality"] != "720p":
item["label"] = "%s (%s)" % (item["label"], movie["Quality"])
item.update({"path": plugin.url_for("play", uri=movie["TorrentMagnetUrl"]), "is_playable": True})
item.setdefault("info", {}).update(
{
"count": movie["MovieID"],
"genre": "%s (%s S:%s P:%s)"
% (item["info"]["genre"], movie["Size"], movie["TorrentSeeds"], movie["TorrentPeers"]),
"plot_outline": tmdb_meta["overview"],
"video_codec": "h264",
}
)
width = 1920
height = 1080
if movie["Quality"] == "720p":
width = 1280
height = 720
item.setdefault("stream_info", {}).update(
{"video": {"codec": "h264", "width": width, "height": height}, "audio": {"codec": "aac"}}
)
yield item
if current_page < (int(search_result["MovieCount"]) / limit):
next_args = args.copy()
next_args["set"] = int(next_args["set"]) + 1
yield {"label": ">> Next page", "path": plugin.url_for(callback, **next_args)}