1 files changed, 41 insertions, 0 deletions
diff --git a/plugin.video.mediathekview/resources/lib/channelui.py b/plugin.video.mediathekview/resources/lib/channelui.py
new file mode 100644
index 0000000..36743e8
--- /dev/null
+++ b/plugin.video.mediathekview/resources/lib/channelui.py
@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+# Copyright 2017 Leo Moll and Dominik Schlösser
+#
+
+# -- Imports ------------------------------------------------
+import sys, urllib
+import xbmcplugin, xbmcgui
+
+from resources.lib.channel import Channel
+
+# -- Classes ------------------------------------------------
+class ChannelUI( Channel ):
+	def __init__( self, handle, sortmethods = None, nextdir = 'initial' ):
+		self.base_url		= sys.argv[0]
+		self.nextdir		= nextdir
+		self.handle			= handle
+		self.sortmethods	= sortmethods if sortmethods is not None else [ xbmcplugin.SORT_METHOD_TITLE ]
+		self.count			= 0
+
+	def Begin( self ):
+		for method in self.sortmethods:
+			xbmcplugin.addSortMethod( self.handle, method )
+
+	def Add( self, altname = None ):
+		resultingname = self.channel if self.count == 0 else '%s (%d)' % ( self.channel, self.count, )
+		li = xbmcgui.ListItem( label = resultingname if altname is None else altname )
+		xbmcplugin.addDirectoryItem(
+			handle	= self.handle,
+			url		= self.build_url( {
+				'mode': self.nextdir,
+				'channel': self.id
+			} ),
+			listitem = li,
+			isFolder = True
+		)
+
+	def End( self ):
+		xbmcplugin.endOfDirectory( self.handle )
+
+	def build_url( self, query ):
+		return self.base_url + '?' + urllib.urlencode( query )