implement RelatedPaginator watched/progress generically;

Author: pannal

lib/windows/pagination.py

@@ -1,5 +1,6 @@
 # coding=utf-8
 import kodigui
+from lib import util
 
 
 class MLCPaginator(object):
@@ -189,4 +190,12 @@ def createListItem(self, rel):
             rel.title or '',
             thumbnailImage=rel.defaultThumb.asTranscodedImageURL(*self.parentWindow.RELATED_DIM),
             data_source=rel
-        )
+        )
+
+    def prepareListItem(self, data, mli):
+        if data.type in ('season', 'show'):
+            if not mli.dataSource.isWatched:
+                mli.setProperty('unwatched.count', str(mli.dataSource.unViewedLeafCount))
+        else:
+            mli.setProperty('unwatched', not mli.dataSource.isWatched and '1' or '')
+            mli.setProperty('progress', util.getProgressImage(mli.dataSource))

lib/windows/preplay.py

@@ -27,10 +27,6 @@ class RelatedPaginator(pagination.BaseRelatedPaginator):
     def getData(self, offset, amount):
         return self.parentWindow.video.getRelated(offset=offset, limit=amount)
 
-    def prepareListItem(self, data, mli):
-        mli.setProperty('unwatched', not mli.dataSource.isWatched and '1' or '')
-        mli.setProperty('progress', util.getProgressImage(mli.dataSource))
-
 
 class PrePlayWindow(kodigui.ControlledWindow, windowutils.UtilMixin):
     xmlFile = 'script-plex-pre_play.xml'

lib/windows/subitems.py

@@ -29,14 +29,6 @@ class RelatedPaginator(pagination.BaseRelatedPaginator):
     def getData(self, offset, amount):
         return self.parentWindow.mediaItem.getRelated(offset=offset, limit=amount)
 
-    def prepareListItem(self, data, mli):
-        if data.type in ('show', 'season'):
-            if not mli.dataSource.isWatched:
-                mli.setProperty('unwatched.count', str(mli.dataSource.unViewedLeafCount))
-        else:
-            mli.setProperty('unwatched', not mli.dataSource.isWatched and '1' or '')
-            mli.setProperty('progress', util.getProgressImage(mli.dataSource))
-
 
 class ShowWindow(kodigui.ControlledWindow, windowutils.UtilMixin):
     xmlFile = 'script-plex-seasons.xml'

lib/windows/videoplayer.py

@@ -30,14 +30,6 @@ def readyForPaging(self):
     def getData(self, offset, amount):
         return (self.parentWindow.prev or self.parentWindow.next).getRelated(offset=offset, limit=amount)
 
-    def prepareListItem(self, data, mli):
-        if self.parentWindow.prev and self.parentWindow.prev.type == 'episode':
-            if not mli.dataSource.isWatched:
-                mli.setProperty('unwatched.count', str(mli.dataSource.unViewedLeafCount))
-        else:
-            mli.setProperty('unwatched', not mli.dataSource.isWatched and '1' or '')
-        mli.setProperty('progress', util.getProgressImage(mli.dataSource))
-
 
 class OnDeckPaginator(pagination.MLCPaginator):
     def readyForPaging(self):