pplist: add maxPage setter/getter to PPSMovie

1 files changed, 11 insertions, 0 deletions

diff --git a/totem/plugin/ppslist.py b/totem/plugin/ppslist.py
index d68ce10..7a7769c 100644
--- a/totem/plugin/ppslist.py
+++ b/totem/plugin/ppslist.py
@@ -186,6 +186,7 @@ class PPSMovie:
         self.cn = 1
         self.playerurl = ''
         self.baseurl = ''
+        self.max_page = 1
         self.files = {}
         self.pixbuf = None
         self.parent = None
@@ -254,6 +255,12 @@ class PPSMovie:
         for f in self.files[page_id]:
             f.parent = self
 
+    def setMaxPage(self, max_page):
+        self.max_page = max_page
+
+    def getMaxPage(self):
+        return self.max_page
+
 def download(url, max_retry = 3, interval = 5):
     if not url:
         return ''
@@ -516,6 +523,10 @@ class PPSList:
 
     def parseMoviePlayList(self, cls, movie, s, page_id = 0):
         result = parseMoviePlayList(s)
+        if result:
+            max_page = result[1]
+            if max_page > movie.getMaxPage():
+                movie.setMaxPage(max_page)
         return result
 
     def updateMovie(self, cls, movie, file_list, page_id = 0):