From c522656e488d8d573ff63a973172d7e3429f9476 Mon Sep 17 00:00:00 2001
From: Matt Lyon <matthewlyon18@gmail.com>
Date: Fri, 6 Dec 2019 21:46:57 +1100
Subject: fixed row removal issue

---
 tpblite/models/torrents.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'tpblite')

diff --git a/tpblite/models/torrents.py b/tpblite/models/torrents.py
index 79c05f1..32b6b5f 100644
--- a/tpblite/models/torrents.py
+++ b/tpblite/models/torrents.py
@@ -4,6 +4,7 @@ from bs4 import BeautifulSoup
 
 from .utils import Query
 
+#TODO: implement a pretty print for Torrents object
 
 def fileSizeStrToInt(size_str):
     '''Converts file size given in *iB format to bytes integer'''
@@ -101,10 +102,12 @@ class Torrents(object):
         
     def __getRows(self, soup):
         rows = soup.body.find_all('tr')
-        # remove first and last entries
+        # remove first entry (header)
         if len(rows) > 1:
             del rows[0]
-            del rows[-1]
+            if len(rows) == 31:
+                # last row is bottom of page
+                del rows[-1]
             return rows
         else:
             print('No torrents found!')
-- 
cgit v1.2.3