From 2e06dc231fbc67fc30909ee2c8c9a60997922b06 Mon Sep 17 00:00:00 2001
From: Thomas Wade <gigatech55@gmail.com>
Date: Tue, 25 Sep 2018 17:21:23 +0930
Subject: [PATCH] Enforce limits

---
 scrape.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scrape.py b/scrape.py
index eea4681..a1f4591 100755
--- a/scrape.py
+++ b/scrape.py
@@ -132,7 +132,7 @@ def Get_Tracks_Info(url, limit = 100):
 
     tracks = {}
 
-    track_elems = soup.find_all('li', class_='soundList__item')
+    track_elems = soup.find_all('li', class_='soundList__item')[:limit]
     for track_elem in track_elems:
         # Initialise a track object to hold our data
         track = Track()
@@ -222,7 +222,7 @@ def Get_Follows(url, limit = 100):
 
     users = []
 
-    badge_elems = soup.find_all(class_='badgeList__item')
+    badge_elems = soup.find_all(class_='badgeList__item')[:limit]
     for badge_elem in badge_elems:
         username_elem = badge_elem.find('a', class_='userBadgeListItem__heading')
         if username_elem: