def __init__(self, _kodi): ''' Initializes TV-guide object ''' self._kodi = _kodi self._favorites = favorites.Favorites(_kodi) self._proxies = _kodi.get_proxies() install_opener(build_opener(ProxyHandler(self._proxies))) self._showfanart = _kodi.get_setting('showfanart', 'true') == 'true'
def __init__(self, _kodi): ''' Initialize searchtes, relies on XBMC vfs ''' self._kodi = _kodi self._favorites = favorites.Favorites(_kodi) self._apihelper = vrtapihelper.VRTApiHelper(_kodi, self._favorites) self._search_history = _kodi.get_userdata_path( ) + 'search_history.json'
class TestSearch(unittest.TestCase): _favorites = favorites.Favorites(addon.kodi) _apihelper = vrtapihelper.VRTApiHelper(addon.kodi, _favorites) def test_search_journaal(self): ''' Test for journaal ''' search_items, sort, ascending, content = self._apihelper.get_search_items( 'journaal', page=1) # Test we get a non-empty search result self.assertEqual(len(search_items), 50) self.assertEqual(sort, 'dateadded') self.assertFalse(ascending) self.assertEqual(content, 'episodes') def test_search_journaal_page2(self): ''' Test for journaal ''' search_items, sort, ascending, content = self._apihelper.get_search_items( 'journaal', page=2) # Test we get a non-empty search result self.assertEqual(len(search_items), 50) self.assertEqual(sort, 'dateadded') self.assertFalse(ascending) self.assertEqual(content, 'episodes') def test_search_journaal_page3(self): ''' Test for journaal ''' search_items, sort, ascending, content = self._apihelper.get_search_items( 'journaal', page=3) # Test we get a non-empty search result self.assertEqual(len(search_items), 50) self.assertEqual(sort, 'dateadded') self.assertFalse(ascending) self.assertEqual(content, 'episodes') def test_search_weer(self): ''' Test for journaal ''' search_items, sort, ascending, content = self._apihelper.get_search_items( 'weer', page=1) # Test we get a non-empty search result self.assertEqual(len(search_items), 50) self.assertEqual(sort, 'dateadded') self.assertFalse(ascending) self.assertEqual(content, 'episodes')
class TestFavorites(unittest.TestCase): _favorites = favorites.Favorites(addon.kodi) def test_follow_unfollow(self): programs = [ { 'program_title': 'Winteruur', 'program': 'winteruur' }, { 'program_title': 'De Campus Cup', 'program': 'de-campus-cup' }, { 'program_title': 'Terug naar Siberië', 'program': 'terug-naar-siberie' }, { 'program_title': '22/3 - 1 jaar later - het onderzoek', 'program': '22-3-1-jaar-later---het-onderzoek' }, { 'program_title': 'Belle & Sebastian', 'program': 'belle---sebastian' }, ] for program_item in programs: program_title = program_item.get('program_title') program = program_item.get('program') self._favorites.follow(program=program, title=program_title) self.assertTrue(self._favorites.is_favorite(program)) self._favorites.unfollow(program=program, title=program_title) self.assertFalse(self._favorites.is_favorite(program)) self._favorites.follow(program=program, title=program_title) self.assertTrue(self._favorites.is_favorite(program)) def test_programs(self): programs = self._favorites.programs() self.assertTrue(programs) print(programs) def test_titles(self): titles = self._favorites.titles() self.assertTrue(titles) print(sorted(titles))
def Search(url): keyword = utils._get_keyboard(heading="Searching for...") if (not keyword): return False, 0 try: response = utils.getHtml(url + keyword) except: utils.notify('Model not found - try again') return None match = re.compile("p_signupargs: 'smid%3D([^']+)'", re.DOTALL | re.IGNORECASE).findall(response) if match: utils.notify('Found ' + keyword + ' adding to favorites now') img = "http://m1.nsimg.net/media/snap/" + match[0] + ".jpg" performerID = match[0] name = keyword favorites.Favorites('add', 'Playvid', name, performerID, img)
def favorites_refresh(): ''' The API interface to refresh the favorites cache ''' from resources.lib import favorites favorites.Favorites(kodi).get_favorites(ttl=0)
def unfollow(program, title): ''' The API interface to unfollow a program used by the context menu ''' from resources.lib import favorites favorites.Favorites(kodi).unfollow(program=program, title=title)
pornkinox.Search(url, keyword) elif mode == 474: pornkinox.Categories(url) elif mode == 475: camsoda.Main() elif mode == 476: camsoda.List(url) elif mode == 478: camsoda.Playvid(url, name) elif mode == 479: camsoda.clean_database(True) elif mode == 480: naked.Main() elif mode == 481: naked.List(url) elif mode == 482: naked.Playvid(url, name) elif mode == 483: naked.clean_database(True) elif mode == 900: favorites.Favorites(fav, favmode, name, url, img) elif mode == 901: favorites.List() elif mode == 902: utils.newSearch(url, channel) elif mode == 903: utils.clearSearch() xbmcplugin.endOfDirectory(utils.addon_handle)
class TestVRTPlayer(unittest.TestCase): _favorites = favorites.Favorites(addon.kodi) _apihelper = vrtapihelper.VRTApiHelper(addon.kodi, _favorites) _vrtplayer = vrtplayer.VRTPlayer(addon.kodi) def test_show_videos_single_episode_shows_videos(self): program = 'marathonradio' episode_items, sort, ascending, content = self._apihelper.get_episode_items(program=program) self.assertTrue(episode_items, msg=program) self.assertEqual(sort, 'dateadded') self.assertFalse(ascending) self.assertEqual(content, 'episodes') self._vrtplayer.show_episodes(program) # self.assertTrue(self._kodi.show_listing.called) def test_show_videos_single_season_shows_videos(self): program = 'het-weer' episode_items, sort, ascending, content = self._apihelper.get_episode_items(program=program) self.assertTrue(episode_items, msg=program) self.assertEqual(sort, 'dateadded') self.assertFalse(ascending) self.assertEqual(content, 'episodes') self._vrtplayer.show_episodes(program) # self.assertTrue(self._kodi.show_listing.called) def test_show_videos_multiple_seasons_shows_videos(self): program = 'pano' episode_items, sort, ascending, content = self._apihelper.get_episode_items(program=program) self.assertTrue(episode_items) self.assertEqual(sort, 'label') self.assertFalse(ascending) self.assertEqual(content, 'seasons') self._vrtplayer.show_episodes(program) # self.assertTrue(self._kodi.show_listing.called) def test_show_videos_specific_seasons_shows_videos(self): program = 'thuis' episode_items, sort, ascending, content = self._apihelper.get_episode_items(program=program) self.assertTrue(episode_items, msg=program) self.assertEqual(sort, 'label') self.assertFalse(ascending) self.assertEqual(content, 'seasons') self._vrtplayer.show_episodes(program) # self.assertTrue(self._kodi.show_listing.called) def test_categories_scraping(self): ''' Test to ensure our hardcoded categories conforms to scraped categories ''' # Remove thumbnails from scraped categories first categories_scraped = [dict(id=c['id'], name=c['name']) for c in self._apihelper.get_categories()] categories_stored = [dict(id=c['id'], name=c['name']) for c in CATEGORIES] self.assertEqual(categories_scraped, categories_stored) def test_random_tvshow_episodes(self): ''' Rest episode from a random tvshow in a random category ''' categories = self._apihelper.get_categories() self.assertTrue(categories) category = random.choice(categories) tvshow_items = self._apihelper.get_tvshow_items(category['id']) self.assertTrue(tvshow_items, msg=category['id']) tvshow = random.choice(tvshow_items) if tvshow.path.startswith('plugin://plugin.video.vrt.nu/programs/'): # When random program has episodes episode_items, sort, ascending, content = self._apihelper.get_episode_items(tvshow.path.split('/')[4].replace('.relevant', '')) self.assertTrue(episode_items, msg=tvshow.path.split('/')[4]) self.assertTrue(sort in ['dateadded', 'episode', 'label', 'unsorted']) self.assertTrue(ascending is True or ascending is False) self.assertTrue(content in ['episodes', 'seasons'], "Content for '%s' is '%s'" % (tvshow.title, content)) elif tvshow.path.startswith('plugin://plugin.video.vrt.nu/play/id/'): # When random program is playable item pass else: self.fail('We did not expect this, either we find episodes or it is a playable item') def test_categories(self): ''' Test to ensure our hardcoded categories conforms to scraped categories ''' category_items = self._apihelper.get_category_items() self.assertEqual(len(category_items), 17) def test_featured(self): ''' Test to ensure our hardcoded categories conforms to scraped categories ''' featured_items = self._apihelper.get_featured_items() self.assertEqual(len(featured_items), 7)
class ApiHelperTests(unittest.TestCase): _favorites = favorites.Favorites(addon.kodi) _apihelper = vrtapihelper.VRTApiHelper(addon.kodi, _favorites) def test_get_api_data_single_season(self): title_items, sort, ascending, content = self._apihelper.get_episode_items( program='het-journaal') self.assertTrue(120 <= len(title_items) <= 140, 'We got %s items instead.' % len(title_items)) self.assertEqual(sort, 'dateadded') self.assertFalse(ascending) self.assertEqual(content, 'episodes') def test_get_api_data_multiple_seasons(self): title_items, sort, ascending, content = self._apihelper.get_episode_items( program='thuis') self.assertTrue(len(title_items) < 5) self.assertEqual(sort, 'label') self.assertFalse(ascending) self.assertEqual(content, 'seasons') def test_get_api_data_specific_season(self): title_items, sort, ascending, content = self._apihelper.get_episode_items( program='pano') self.assertEqual(len(title_items), 4) self.assertEqual(sort, 'label') self.assertFalse(ascending) self.assertEqual(content, 'seasons') def test_get_api_data_specific_season_without_broadcastdate(self): title_items, sort, ascending, content = self._apihelper.get_episode_items( program='postbus-x') self.assertEqual(len(title_items), 3) self.assertEqual(sort, 'label') self.assertTrue(ascending) self.assertEqual(content, 'seasons') def test_get_recent_episodes(self): ''' Test items, sort and order ''' episode_items, sort, ascending, content = self._apihelper.get_episode_items( page=1, variety='recent') self.assertEqual(len(episode_items), 50) self.assertEqual(sort, 'dateadded') self.assertFalse(ascending) self.assertEqual(content, 'episodes') def test_get_recent_episodes_page1(self): ''' Test items, sort and order ''' episode_items, sort, ascending, content = self._apihelper.get_episode_items( page=2, variety='recent') self.assertEqual(len(episode_items), 50) self.assertEqual(sort, 'dateadded') self.assertFalse(ascending) self.assertEqual(content, 'episodes') def test_get_recent_episodes_page2(self): ''' Test items, sort and order ''' episode_items, sort, ascending, content = self._apihelper.get_episode_items( page=3, variety='recent') self.assertEqual(len(episode_items), 50) self.assertEqual(sort, 'dateadded') self.assertFalse(ascending) self.assertEqual(content, 'episodes') def test_get_offline_episodes(self): ''' Test items, sort and order ''' episode_items, sort, ascending, content = self._apihelper.get_episode_items( page=1, variety='offline') self.assertTrue(episode_items) self.assertEqual(sort, 'dateadded') self.assertFalse(ascending) self.assertEqual(content, 'episodes') def test_get_tvshows(self): ''' Test items, sort and order ''' category = 'nieuws-en-actua' tvshow_items = self._apihelper.get_tvshow_items(category=category) self.assertTrue(tvshow_items) def test_tvshows(self): ''' Test A-Z tvshow listing and CHANNELS list ''' tvshow_items = self._apihelper.get_tvshow_items(category=None) # Test we get a non-empty A-Z listing back self.assertTrue(tvshow_items) # Test every brand is a known channel studio name bogus_brands = ['lang-zullen-we-lezen', 'VRT'] channel_studios = [c.get('studio') for c in CHANNELS] + bogus_brands for tvshow in tvshow_items: self.assertTrue( tvshow.info_dict['studio'] in channel_studios, '%s | %s | %s' % (tvshow.title, tvshow.info_dict['studio'], channel_studios)) def test_get_latest_episode(self): video = self._apihelper.get_latest_episode(program='het-journaal') self.assertTrue(len(video) == 2) self.assertTrue(video.get('video_id') is not None) self.assertTrue(video.get('publication_id') is not None)
def __init__(self, _kodi): ''' Initialise object ''' self._kodi = _kodi self._favorites = favorites.Favorites(_kodi) self._apihelper = vrtapihelper.VRTApiHelper(_kodi, self._favorites)