瀏覽代碼

vtuner pages: always calculate element count

milaq 5 年之前
父節點
當前提交
4fb591a18b
共有 2 個文件被更改,包括 2 次插入6 次删除
  1. 0 2
      ycast/server.py
  2. 2 4
      ycast/vtuner.py

+ 0 - 2
ycast/server.py

@@ -48,7 +48,6 @@ def get_directories_page(subdir, directories, startitems, enditems):
     if len(directories) == 0:
         page.add(vtuner.Display("No entries found."))
         return page
-    page.set_count(len(directories))
     offset = 0
     limit = len(directories)
     if startitems and enditems:
@@ -69,7 +68,6 @@ def get_stations_page(stations, startitems, enditems):
     if len(stations) == 0:
         page.add(vtuner.Display("No stations found."))
         return page
-    page.set_count(len(stations))
     offset = 0
     limit = len(stations)
     if startitems and enditems:

+ 2 - 4
ycast/vtuner.py

@@ -16,13 +16,11 @@ def strip_https(url):
 class Page:
     def __init__(self):
         self.items = []
-        self.count = -1
+        self.count = 0
 
     def add(self, item):
         self.items.append(item)
-
-    def set_count(self, count):
-        self.count = count
+        self.count = len(self.items)
 
     def to_xml(self):
         xml = etree.Element('ListOfItems')