Use an auth token for extended ratelimit (#96)

* Use an auth token for extended ratelimit

Daily jams are failing for users in production due to ratelimit errors.
The intent is to pass troi's auth token to the patch in prod so that
the ratelimit can be bypassed.

Author: amCap1712

troi/core.py

@@ -1,5 +1,4 @@
 #!/usr/bin/env python3
-
 import sys
 from typing import Dict
 

troi/listenbrainz/feedback.py

@@ -14,9 +14,10 @@ class ListensFeedbackLookup(Element):
         :param user_name: The ListenBrainz user_name for whom to fetch feedback information.
     """
 
-    def __init__(self, user_name):
+    def __init__(self, user_name, auth_token=None):
         super().__init__()
         self.user_name = user_name
+        self.auth_token = auth_token
 
     @staticmethod
     def inputs():
@@ -36,6 +37,8 @@ def read(self, inputs):
             mbids.add(r.mbid)
         mbids = list(mbids)
 
+        headers = {"Authorization": f"Token {self.auth_token}"} if self.auth_token else {}
+
         feedback_map = {}
         batch_size = 50
         for idx in range(0, len(mbids), batch_size):
@@ -44,14 +47,14 @@ def read(self, inputs):
             while True:
                 response = requests.get(
                     f"https://api.listenbrainz.org/1/feedback/user/{self.user_name}/get-feedback-for-recordings",
-                    params={"recording_mbids": ",".join(recording_mbids)}
+                    params={"recording_mbids": ",".join(recording_mbids)},
+                    headers=headers
                 )
                 if response.status_code == 429:
                     sleep(2)
                     continue
 
                 break
-
             response.raise_for_status()
             data = response.json()["feedback"]
             if len(data) == 0:

troi/listenbrainz/listens.py

@@ -16,13 +16,15 @@ class RecentListensTimestampLookup(Element):
         Timestamps are stored in the listenbrainz dict, with key name "latest_listened_at".
 
         :param user_name: The ListenBrainz user for whome to fetch recent listen timestamps.
+        :param auth_token: a ListenBrainz auth token
         :param days: The number of days to check.
     """
 
-    def __init__(self, user_name, days: int):
+    def __init__(self, user_name, days: int, auth_token=None):
         super().__init__()
         self.user_name = user_name
         self.days = days
+        self.auth_token = auth_token
         self.index: Optional[dict[str, int]] = None
 
     @staticmethod
@@ -42,9 +44,11 @@ def _fetch_recent_listens_index(self):
         min_dt = datetime.now() + timedelta(days=-self.days)
         min_ts = int(min_dt.timestamp())
         while True:
+            headers = {"Authorization": f"Token {self.auth_token}"} if self.auth_token else {}
             response = requests.get(
                 f"https://api.listenbrainz.org/1/user/{self.user_name}/listens",
-                params={"min_ts": min_ts, "count": 100}
+                params={"min_ts": min_ts, "count": 100},
+                headers=headers
             )
             if response.status_code == 429:
                 sleep(2)

troi/listenbrainz/recs.py

@@ -19,9 +19,11 @@ class UserRecordingRecommendationsElement(Element):
 
     MAX_RECORDINGS_TO_FETCH = 2000
 
-    def __init__(self, user_name, artist_type, count=25, offset=0):
+    def __init__(self, user_name, artist_type, count=25, offset=0, auth_token=None):
         super().__init__()
         self.client = pylistenbrainz.ListenBrainz()
+        if auth_token:
+            self.client.set_auth_token(auth_token)
         self.user_name = user_name
         self.count = count
         self.offset = offset

troi/patches/daily_jams.py

@@ -57,9 +57,9 @@ def create(self, inputs):
         if jam_date is None:
             jam_date = datetime.utcnow().strftime("%Y-%m-%d %a")
 
-        recs = troi.listenbrainz.recs.UserRecordingRecommendationsElement(user_name, "raw", count=1000)
+        recs = troi.listenbrainz.recs.UserRecordingRecommendationsElement(user_name, "raw", count=1000, auth_token=inputs.get("token"))
 
-        recent_listens_lookup = troi.listenbrainz.listens.RecentListensTimestampLookup(user_name, days=2)
+        recent_listens_lookup = troi.listenbrainz.listens.RecentListensTimestampLookup(user_name, days=2, auth_token=inputs.get("token"))
         recent_listens_lookup.set_sources(recs)
 
         # Remove tracks that have not been listened to before.
@@ -69,7 +69,7 @@ def create(self, inputs):
         latest_filter = troi.filters.LatestListenedAtFilterElement(DAYS_OF_RECENT_LISTENS_TO_EXCLUDE)
         latest_filter.set_sources(never_listened)
 
-        feedback_lookup = troi.listenbrainz.feedback.ListensFeedbackLookup(user_name)
+        feedback_lookup = troi.listenbrainz.feedback.ListensFeedbackLookup(user_name, auth_token=inputs.get("token"))
         feedback_lookup.set_sources(latest_filter)
 
         recs_lookup = troi.musicbrainz.recording_lookup.RecordingLookupElement()