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

---
 setup.py                   | 2 +-
 tpblite/models/torrents.py | 7 +++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index 4df49d6..300c7a0 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 from setuptools import setup, find_packages
 
 setup(name='tpblite',
-        version='0.1.0',
+        version='0.1.1',
         description='The Unofficial Pirate Bay Lightweight Python API',
         author='Matt Lyon',
         author_email='matthewlyon18@gmail.com',
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