#!/usr/bin/env python
# -*- coding: UTF-8 -*-
#
# Imports
#
from __future__ import absolute_import
from future import standard_library
standard_library.install_aliases()
from builtins import str
from builtins import object
import os
import requests
import sys
import urllib.request, urllib.parse, urllib.error
import re
import xbmcgui
import xbmcplugin
from .hak5_const import IMAGES_PATH, HEADERS, LANGUAGE, convertToUnicodeString, log, getSoup
#
# Main class
#
class Main(object):
def __init__(self):
# Get the command line arguments
# Get the plugin url in plugin:// notation
self.plugin_url = sys.argv[0]
# Get the plugin handle as an integer number
self.plugin_handle = int(sys.argv[1])
log("ARGV", repr(sys.argv))
# Parse parameters...
self.plugin_category = urllib.parse.parse_qs(urllib.parse.urlparse(sys.argv[2]).query)['plugin_category'][0]
self.video_list_page_url = urllib.parse.parse_qs(urllib.parse.urlparse(sys.argv[2]).query)['url'][0]
self.next_page_possible = urllib.parse.parse_qs(urllib.parse.urlparse(sys.argv[2]).query)['next_page_possible'][0]
self.video_list_page_url = str(self.video_list_page_url).replace('https', 'http')
log("self.video_list_page_url", self.video_list_page_url)
#
# Get the videos...
#
self.getVideos()
#
# Get videos...
#
def getVideos(self):
#
# Init
#
# Create a list for our items.
listing = []
#
# Get HTML page
#
response = requests.get(self.video_list_page_url, headers=HEADERS)
html_source = response.text
html_source = convertToUnicodeString(html_source)
# Parse response
soup = getSoup(html_source)
#
# log("html_source", html_source)
#Season 1
seasons = soup.findAll('a', attrs={'href': re.compile("^" + "https://www.hak5.org/category/episodes/season")})
log("len(seasons", len(seasons))
for season in seasons:
# log("season", season)
# let's skip these links
#
if str(season).find("class=") > 0:
log("skipped season that contains class=", season)
continue
# let's skip these links
# Season 22
if str(season).find("rel=") > 0:
log("skipped season that contains rel=", season)
continue
# let's skip links that don't contain the word season
# Pineapple Uni
if str(season).find("season") > 0:
pass
else:
log("skipped season that does not contain the word season", season)
continue
url = season['href']
thumbnail_url = ''
title = season.text
add_sort_methods()
context_menu_items = []
# Add refresh option to context menu
context_menu_items.append((LANGUAGE(30104), 'Container.Refresh'))
# Add to list...
list_item = xbmcgui.ListItem(label=title, thumbnailImage=thumbnail_url)
list_item.setArt({'thumb': thumbnail_url, 'icon': thumbnail_url,
'fanart': os.path.join(IMAGES_PATH, 'fanart-blur.jpg')})
list_item.setProperty('IsPlayable', 'false')
parameters = {"action": "list-episodes", "season_name": title, "url": url, "next_page_possible": "False",
"title": title}
url = self.plugin_url + '?' + urllib.parse.urlencode(parameters)
is_folder = True
# Adding context menu items to context menu
list_item.addContextMenuItems(context_menu_items, replaceItems=False)
# Add our item to the listing as a 3-element tuple.
listing.append((url, list_item, is_folder))
# Add our listing to Kodi.
# Large lists and/or slower systems benefit from adding all items at once via addDirectoryItems
# instead of adding one by ove via addDirectoryItem.
xbmcplugin.addDirectoryItems(self.plugin_handle, listing, len(listing))
# Disable sorting
xbmcplugin.addSortMethod(handle=self.plugin_handle, sortMethod=xbmcplugin.SORT_METHOD_NONE)
# Finish creating a virtual folder.
xbmcplugin.endOfDirectory(self.plugin_handle)
def add_sort_methods():
sort_methods = [xbmcplugin.SORT_METHOD_UNSORTED,xbmcplugin.SORT_METHOD_LABEL,xbmcplugin.SORT_METHOD_DATE,xbmcplugin.SORT_METHOD_DURATION,xbmcplugin.SORT_METHOD_EPISODE]
for method in sort_methods:
xbmcplugin.addSortMethod(int(sys.argv[1]), sortMethod=method)