1 files changed, 105 insertions, 0 deletions
diff --git a/plugin.video.mediathekview/classes/filmui.py b/plugin.video.mediathekview/classes/filmui.py
new file mode 100644
index 0000000..eb2cfa8
--- /dev/null
+++ b/plugin.video.mediathekview/classes/filmui.py
@@ -0,0 +1,105 @@
+# -*- coding: utf-8 -*-
+# Copyright 2017 Leo Moll and Dominik Schlösser
+#
+
+# -- Imports ------------------------------------------------
+import xbmcaddon, xbmcplugin, xbmcgui
+
+from classes.film import Film
+from classes.settings import Settings
+
+# -- Classes ------------------------------------------------
+class FilmUI( Film ):
+	def __init__( self, plugin, sortmethods = [ xbmcplugin.SORT_METHOD_TITLE, xbmcplugin.SORT_METHOD_DATE, xbmcplugin.SORT_METHOD_DURATION, xbmcplugin.SORT_METHOD_SIZE ] ):
+		self.plugin			= plugin
+		self.handle			= plugin.addon_handle
+		self.settings		= Settings()
+		self.sortmethods	= sortmethods
+		self.showshows		= False
+		self.showchannels	= False
+
+	def Begin( self, showshows, showchannels ):
+		self.showshows		= showshows
+		self.showchannels	= showchannels
+		# xbmcplugin.setContent( self.handle, 'tvshows' )
+		for method in self.sortmethods:
+			xbmcplugin.addSortMethod( self.handle, method )
+
+	def Add( self, alttitle = None ):
+		# get the best url
+		videourl = self.url_video_hd if ( self.url_video_hd != "" and self.settings.preferhd ) else self.url_video if self.url_video != "" else self.url_video_sd
+		videohds = " (HD)" if ( self.url_video_hd != "" and self.settings.preferhd ) else ""
+		# exit if no url supplied
+		if videourl == "":
+			return
+
+		if alttitle is not None:
+			resultingtitle = alttitle
+		else:
+			if self.showshows:
+				resultingtitle = self.show + ': ' + self.title
+			else:
+				resultingtitle = self.title
+			if self.showchannels:
+				resultingtitle += ' [' + self.channel + ']'
+
+		infoLabels = {
+			'title' : resultingtitle + videohds,
+			'sorttitle' : resultingtitle,
+			'tvshowtitle' : self.show,
+			'plot' : self.description
+		}
+
+		if self.size > 0:
+			infoLabels['size'] = self.size * 1024 * 1024
+
+		if self.seconds > 0:
+			infoLabels['duration'] = self.seconds
+
+		if self.aired is not None:
+			airedstring = '%s' % self.aired
+			infoLabels['date']		= airedstring[8:10] + '-' + airedstring[5:7] + '-' + airedstring[:4]
+			infoLabels['aired']		= airedstring
+			infoLabels['dateadded']	= airedstring
+			
+		li = xbmcgui.ListItem( resultingtitle, self.description )
+		li.setInfo( type = 'video', infoLabels = infoLabels )
+		li.setProperty( 'IsPlayable', 'true' )
+
+		# create context menu
+		contextmenu = []
+		if self.size > 0:
+			# Download video
+			contextmenu.append( (
+				self.plugin.language( 30921 ),
+				'RunPlugin({})'.format( self.plugin.build_url( { 'mode': "download", 'id': self.id, 'quality': 1 } ) )
+			) )
+			if self.url_video_hd:
+				# Download SD video
+				contextmenu.append( (
+					self.plugin.language( 30923 ),
+					'RunPlugin({})'.format( self.plugin.build_url( { 'mode': "download", 'id': self.id, 'quality': 2 } ) )
+				) )
+			if self.url_video_sd:
+				# Download SD video
+				contextmenu.append( (
+					self.plugin.language( 30922 ),
+					'RunPlugin({})'.format( self.plugin.build_url( { 'mode': "download", 'id': self.id, 'quality': 0 } ) )
+				) )
+		# Add to queue
+		# TODO: Enable later
+#		contextmenu.append( (
+#			self.plugin.language( 30924 ),
+#			'RunPlugin({})'.format( self.plugin.build_url( { 'mode': "enqueue", 'id': self.id } ) )
+#		) )
+		li.addContextMenuItems( contextmenu )
+
+		xbmcplugin.addDirectoryItem(
+			handle		= self.handle,
+			url			= videourl,
+			listitem	= li,
+			isFolder	= False
+		)
+
+	def End( self ):
+		xbmcplugin.endOfDirectory( self.handle, cacheToDisc = False )