class Test(unittest.TestCase):
def setUp(self):
self.cookie_jar = CookieJar()
self.opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(self.cookie_jar))
self.u = VplayUrls()
self.r = VplayRegex()
# login
self.opener.open(self.u.get_login_url(), self.u.get_login_params(USERNAME, PASSWORD))
def test_base_url(self):
response = self.opener.open(self.u.get_base_url())
username = self.r.get_username(response.read())
self.assertEqual(username, USERNAME)
def test_tv_show_pagination(self):
page = random.randint(2, 100) # I assume there are more than 100 pages
response = self.opener.open(self.u.get_tv_shows_url(page))
print "TEST SHOW PAGE %s" % page
data = response.read()
username = self.r.get_username(data)
self.assertEqual(username, USERNAME)
tv_shows = list(self.r.get_tv_shows(data))
self.assertTrue(tv_shows)
for tv_show in tv_shows:
print "%s" % tv_show
self.assertEqual(len(tv_shows), TV_SHOWS_PER_PAGE) # Pagination currently is set to 15
def test_tv_show_search(self):
print "TEST TV SHOW SEARCH"
page = random.randint(2, 10)
query = "Seinfeld"
response = self.opener.open(self.u.get_tv_shows_url(page, query))
data = response.read()
username = self.r.get_username(data)
self.assertEqual(username, USERNAME)
tv_shows = list(self.r.get_tv_shows(data))
self.assertTrue(tv_shows)
for tv_show in tv_shows:
print "%s" % tv_show
def test_tv_shows(self):
# choose a random tv show page
response = self.opener.open(self.u.get_tv_shows_url(random.randint(1, 100)))
data = response.read()
tv_shows = list(self.r.get_tv_shows(data))
self.assertTrue(tv_shows)
# choose a random tv show
tv_show = tv_shows[random.randint(0, len(tv_shows) - 1)]
print "TV Show: %s" % tv_show
self.assertIn("path", tv_show)
self.assertIn("title", tv_show)
self.assertIn("image", tv_show)
response = self.opener.open(self.u.get_tv_seasons_url(tv_show["path"]))
data = response.read()
tv_seasons = list(self.r.get_tv_seasons(data))
self.assertTrue(tv_seasons)
# Select a random season
tv_season = tv_seasons[random.randint(0, len(tv_seasons) - 1)]
print "Random TV Season: %s" % tv_season
self.assertIn("path", tv_season)
self.assertIn("title", tv_season)
response = self.opener.open(self.u.get_tv_seasons_url(tv_season["path"]))
data = response.read()
tv_episodes = list(self.r.get_tv_episodes(data))
self.assertTrue(tv_episodes)
# Select a random episode
tv_episode = tv_episodes[random.randint(0, len(tv_episodes) - 1)]
print "Random TV Episode: %s" % tv_episode
self.assertIn("path", tv_episode)
self.assertIn("full_title", tv_episode)
self.assertIn("image", tv_episode)
self.assertIn("title", tv_episode)
self.assertIn("watched", tv_episode)
episode_url = self.u.get_tv_episode_url(tv_episode["path"])
episode_key = self.r.get_tv_episode_key(episode_url)
self.assertTrue(episode_key)
response = self.opener.open(self.u.get_dino_url(), self.u.get_dino_params(episode_key))
data = response.read()
self.assertTrue(data, "Empty response from dino")
dino = self.r.get_dino(data)
self.assertIn("url", dino)
self.assertIn("thumb", dino)
self.assertIn("subs", dino)
# Test subtitles if present
if dino["subs"]:
languages = json.loads(dino["subs"])
selected_lang = languages[0]
print "Subtitle in %s" % selected_lang
response = self.opener.open(self.u.get_subs_url(), self.u.get_subs_params(episode_key, selected_lang))
data = response.read()
sub_data = self.r.get_sub(data)
self.assertTrue(sub_data)
# Check if data can be decoded
sub_json = json.loads(sub_data)
def test_hdvideos(self):
page = random.randint(1, 50)
response = self.opener.open(self.u.get_hdvideos_url(page))
data = response.read()
videos = list(self.r.get_videos(data))
self.assertTrue(videos)
# select random video
video = videos[random.randint(0, len(videos) - 1)]
print "Video: %s" % video
self.assertIn("path", video)
self.assertIn("duration", video)
self.assertIn("image", video)
self.assertIn("title", video)
def test_video_search(self):
page = random.randint(1, 10)
query = "funny"
response = self.opener.open(self.u.get_search_url(query, page))
data = response.read()
videos = list(self.r.get_videos(data))
print "Videos: %s" % ("\n".join(["%s" % v for v in videos]))
self.assertTrue(videos)
print "Searching for .. found %d videos on page %d" % (len(videos), page)
self.assertEqual(len(videos), SEARCH_RESULTS_PER_PAGE)
# Select random video
video = videos[random.randint(0, len(videos) - 1)]
print "Video: %s" % video
self.assertIn("path", video)
self.assertIn("duration", video)
self.assertIn("image", video)
self.assertIn("title", video)
def test_top50_videos(self):
response = self.opener.open(self.u.get_top50_url())
data = response.read()
videos = list(self.r.get_top50_videos(data))
print "Videos: %s" % ("\n".join(["%s" % v for v in videos]))
self.assertTrue(videos)
self.assertEqual(len(videos), 50)
# Choose random video
video = videos[random.randint(0, len(videos) - 1)]
self.assertIn("path", video)
self.assertIn("image", video)
self.assertIn("title", video)
class VplayApp(mc.Player):
MAX_FAILED_LOGIN_COUNT = 3
def log(self, msg):
mc.LogInfo('VPLAY: %s' % msg)
def notify(self, msg):
self.log('NOTIFICATION %s' % msg)
mc.ShowDialogNotification(msg)
def __init__(self):
mc.Player.__init__(self, True)
self.http = mc.Http()
self.r = VplayRegex()
self.u = VplayUrls()
config = mc.GetApp().GetLocalConfig()
self.username = config.GetValue('username')
self.password = config.GetValue('password')
self.logged_in = False
self.failed_login_count = 0
self.last_played_episode = 0
self.populate_tv_shows = True
try:
platform_func = getattr(mc, 'GetPlatform')
self.platform = platform_func()
except AttributeError:
self.platform = 'None'
# FIXME: See what returns mc.GetPlatform()
# TODO: Update this when I'll recover my boxee box
def is_boxeebox(self):
return False
# return self.platform != 'None'
def get_username(self):
if not self.username:
self.username = mc.ShowDialogKeyboard('Enter Username', '', False)
config = mc.GetApp().GetLocalConfig()
config.SetValue('username', self.username)
return self.username
def get_password(self):
if not self.password:
self.password = mc.ShowDialogKeyboard('Enter Password', '', True)
config = mc.GetApp().GetLocalConfig()
config.SetValue('password', self.password)
return self.password
def _login(self):
# Erase saved credentials
if self.failed_login_count == self.MAX_FAILED_LOGIN_COUNT:
self._logout()
username = self.get_username()
password = self.get_password()
if username and password:
response = self.http.Post(self.u.get_login_url(), self.u.get_login_params(username, password))
logged_username = self.r.get_username(response)
if logged_username:
self.logged_in = True
else:
self.failed_login_count = self.failed_login_count + 1
mc.ShowDialogNotification('Unable to log in.')
self.logged_in = False
self.update_login_status()
return self.logged_in
'''
Reset user credentials from local config so that user can re-introduce them.
'''
def _logout(self):
# Reset saved credentials
config = mc.GetApp().GetLocalConfig()
config.Reset('username')
config.Reset('password')
self.username = ''
self.password = ''
# Reset http object (hopefully it will erase any saved cookies
self.http.Reset()
self.logged_in = False
self.failed_login_count = 0
self.update_login_status()
def toggle_login(self):
if self.logged_in:
self._logout()
else:
self._login()
'''
Updates UI based on self.logged_in
'''
def update_login_status(self):
button_label = 'Login'
if self.logged_in:
button_label = 'Logout %s' % self.username
mc.GetActiveWindow().GetButton(LOGIN_BUTTON_ID).SetLabel(button_label)
'''
Executed at window load, used to update UI elements (set focus, etc)
'''
def on_load(self):
# Update login status, we might already have cookies from previous run
response = self.http.Get(self.u.get_base_url())
username = self.r.get_username(response)
if username:
self.logged_in = True
self.update_login_status()
# Enforce login
if not self.logged_in:
self._login()
# Coming back from player
list = mc.GetActiveWindow().GetList(NAVIGATION_LIST_ID)
items = list.GetItems()
# Load the initial list
if len(items) == 0 and self.populate_tv_shows:
self.populate_tv_shows = False
self.load_tv_shows()
self.log('Last focused item: %s' % self.last_played_episode)
# Play the next episode
if self.GetLastPlayerEvent() == self.EVENT_ENDED:
# do we have a next episode
if self.last_played_episode + 1 < len(items):
list.SetFocusedItem(self.last_played_episode + 1)
current_item = list.GetItem(self.last_played_episode + 1)
self.load_next()
# If current item is nav item, we have to play first item on next page
if current_item.GetProperty('type') in (HD_VIDEOS_PAGE, ):
self.load_next()
else:
# only focus last played episode
list.SetFocusedItem(self.last_played_episode)
else:
# Maybe player is in background or user stopped the player
list.SetFocusedItem(self.last_played_episode)
def load_tv_shows(self):
# clear nav list
items = self.get_tv_shows()
mc.GetActiveWindow().GetList(NAVIGATION_LIST_ID).SetItems(items)
mc.GetActiveWindow().PushState()
'''
Search tv shows
'''
def search_tv_shows(self):
query = mc.ShowDialogKeyboard('Enter search query', '', False)
items = self.get_tv_shows(query)
mc.GetActiveWindow().GetList(NAVIGATION_LIST_ID).SetItems(items)
def load_hd_videos(self):
items = self.get_hd_videos()
mc.GetActiveWindow().GetList(NAVIGATION_LIST_ID).SetItems(items)
def search_videos(self):
query = mc.ShowDialogKeyboard('Enter search query', '', False)
items = self.get_hd_videos(query=query)
mc.GetActiveWindow().GetList(NAVIGATION_LIST_ID).SetItems(items)
def load_top50_videos(self):
items = self.get_top50_videos()
mc.GetActiveWindow().GetList(NAVIGATION_LIST_ID).SetItems(items)
'''
App main controller - handles nav list's clicks
'''
def load_next(self):
list = mc.GetActiveWindow().GetList(NAVIGATION_LIST_ID)
item = list.GetItem(list.GetFocusedItem())
items = None # items to populate list
if item:
item_type = item.GetProperty('type')
if item_type == TV_SHOW:
items = self.get_seasons(item)
elif item_type == TV_SHOW_PAGE:
page = item.GetProperty('page')
items = self.get_tv_shows(page=page)
elif item_type == TV_SEASON:
# must load season's episodes
items = self.get_episodes(item)
elif item_type == TV_EPISODE:
self.last_played_episode = list.GetFocusedItem()
self.play_episode(item)
elif item_type == HD_VIDEOS_PAGE:
query = item.GetProperty('query')
page = item.GetProperty('page')
items = self.get_hd_videos(page, query)
else:
# Load default list
items = self.get_tv_shows()
if items:
# save current window state
mc.GetActiveWindow().PushState()
self.last_played_episode = 0
mc.GetActiveWindow().GetList(NAVIGATION_LIST_ID).SetItems(items)
'''
Inserts Next Page and Prev Page buttons
nav_type one of this values: TV_SHOW_PAGE, HD_VIDEOS_PAGE, SEARCH_PAGE
query Used for Search
'''
def _append_nav(self, items, page, nav_type, query=''):
# append back button
item = mc.ListItem(mc.ListItem.MEDIA_UNKNOWN)
item.SetLabel('Next Page')
item.SetProperty('type', nav_type)
item.SetProperty('page', str(page+1))
item.SetProperty('query', query)
items.append(item)
if page > 1:
item = mc.ListItem(mc.ListItem.MEDIA_UNKNOWN)
item.SetLabel('Prev Page')
item.SetProperty('type', nav_type)
item.SetProperty('page', str(page-1))
if query:
item.SetProperty('query', query)
items.append(item)
'''
Returns a list of tv shows
'''
def get_tv_shows(self, search=None, page='1'):
data = self.http.Get(self.u.get_tv_shows_url(page=int(page), search=search))
page = int(page)
items = mc.ListItems()
for show in self.r.get_tv_shows(data):
show_item = mc.ListItem(mc.ListItem.MEDIA_UNKNOWN)
show_item.SetLabel(show['title'])
show_item.SetPath(show['path'])
show_item.SetTitle(show['title'])
show_item.SetTVShowTitle(show['title'])
show_item.SetThumbnail(show['image'])
# Custom properties
show_item.SetProperty('type', TV_SHOW)
show_item.SetProperty('tv_show', show['title'])
items.append(show_item)
# append navigation
self._append_nav(items, page, TV_SHOW_PAGE)
return items
'''
Returns a list of seasons for tv_show show
'''
def get_seasons(self, tv_show_item):
data = self.http.Get(self.u.get_tv_seasons_url(tv_show_item.GetPath()))
description = self.r.get_tv_show_description(data)
items = mc.ListItems()
for season in self.r.get_tv_seasons(data):
item = mc.ListItem(mc.ListItem.MEDIA_UNKNOWN)
item.SetLabel(season['title'])
item.SetPath(season['path'])
item.SetTitle(season['title'])
item.SetTVShowTitle(tv_show_item.GetTVShowTitle())
item.SetThumbnail(tv_show_item.GetThumbnail())
# Set custom
item.SetProperty('type', TV_SEASON)
item.SetProperty('tv_show', tv_show_item.GetProperty('tv_show'))
item.SetProperty('tv_season', season['title'])
# item.SetProperty('tv_show_thumb', tv_show_item.GetThumbnail())
item.SetProperty('description', description)
items.append(item)
return items
'''
Returns a list of episodes for season
'''
def get_episodes(self, season_item):
data = self.http.Get(self.u.get_tv_seasons_url(season_item.GetPath()))
items = mc.ListItems()
for episode in self.r.get_tv_episodes(data):
watched = ''
if episode['watched']:
watched = '(Watched)'
item = mc.ListItem(mc.ListItem.MEDIA_UNKNOWN)
item.SetLabel('%s %s' % (episode['title'], watched))
item.SetPath(episode['path'])
item.SetTitle(episode['title'])
item.SetTVShowTitle(season_item.GetProperty('tv_show'))
item.SetThumbnail(episode['image'])
# custom properties
item.SetProperty('type', TV_EPISODE)
item.SetProperty('tv_show', season_item.GetProperty('tv_show'))
item.SetProperty('tv_season', season_item.GetProperty('tv_season'))
item.SetProperty('tv_show_image', season_item.GetProperty('tv_show_thumb'))
items.append(item)
return items
def get_hd_videos(self, page='1', query=''):
page = int(page)
data = self.http.Get(self.u.get_hdvideos_url(page, query))
items = mc.ListItems()
for video in self.r.get_videos(data):
item = mc.ListItem(mc.ListItem.MEDIA_UNKNOWN)
item.SetLabel('%s (%s)' % (video['title'], video['duration']))
item.SetPath(video['path'])
item.SetTitle(video['title'])
item.SetTVShowTitle(video['title'])
item.SetThumbnail(video['image'])
item.SetProperty('type', TV_EPISODE)
item.SetProperty('tv_show', 'Videos')
item.SetProperty('tv_season', '')
item.SetProperty('tv_show_thumb', video['image'])
items.append(item)
# Add nav buttons
self._append_nav(items, page, HD_VIDEOS_PAGE, query)
return items
def get_top50_videos(self):
data = self.http.Get(self.u.get_top50_url())
items = mc.ListItems()
for video in self.r.get_top50_videos(data):
item = mc.ListItem(mc.ListItem.MEDIA_UNKNOWN)
item.SetLabel(video['title'])
item.SetPath(video['path'])
item.SetTitle(video['title'])
item.SetTVShowTitle(video['title'])
item.SetThumbnail(video['image'])
item.SetProperty('type', TV_EPISODE)
item.SetProperty('tv_show', 'Top50 Videos')
item.SetProperty('tv_season', '')
item.SetProperty('tv_show_thumb', video['image'])
items.append(item)
return items
'''
Converts Vplay subtitle time into sub format
'''
def convert_time(self, f):
return '%s,0' % str(timedelta(seconds=f))
def _load_subs(self, episode_key, lang='RO'):
sub_raw_data = self.http.Post(self.u.get_subs_url(), self.u.get_subs_params(episode_key, lang))
file_path = None
if sub_raw_data:
file_path = mc.GetTempDir() + lang + '_' + episode_key + '.sub'
try:
sub_data = self.r.get_sub(sub_raw_data)
sub_json = json.loads(sub_data)
file = open(file_path, mode='w+')
count = 1
for json_line in sub_json:
line = '%d \n%s --> %s\n%s\n\n' % (count,
self.convert_time(json_line['f']),
self.convert_time(json_line['t']),
json_line['s'])
line = line.replace('<br>', ' ')
file.write(line.encode('utf-8'))
count = count + 1
file.close()
except Exception, ex:
file_path = None
self.log('Error saving sub: %s' % ex)
return file_path