diff --git a/RDramaAPIInterface.py b/RDramaAPIInterface.py index 13c9a5e..e2373b3 100644 --- a/RDramaAPIInterface.py +++ b/RDramaAPIInterface.py @@ -3,7 +3,7 @@ import requests from bs4 import BeautifulSoup import traceback import backoff -from functools import cache +from functools import lru_cache class TimeOutException(Exception): pass @@ -16,11 +16,13 @@ class RDramaAPIInterface: self.headers={"Authorization": authorization_token} self.site = site self.protocol = "https" if https else "http" + self.user_information = None + self.badge_refresh = False def is_chudded(self): return "Chud" in self.get_my_badges_simple() - @cache + @lru_cache(maxsize=None) def get_my_badges_simple(self): return [i['name'] for i in self.get_my_badges()] @@ -31,7 +33,7 @@ class RDramaAPIInterface: url=f"{self.protocol}://{self.site}/submit" return self.post(url, data={'title' : title, 'url': submission_url, 'body': body}) - @cache + @lru_cache(maxsize=None) def get_me(self): self.get_front_page() url=f"{self.protocol}://{self.site}/@me" @@ -124,6 +126,7 @@ class RDramaAPIInterface: def get_front_page(self): url=f"{self.protocol}://{self.site}" + self.badge_refresh = True return self.get(url) def get_hole(self, hole: str): @@ -136,7 +139,8 @@ class RDramaAPIInterface: def get_user_information(self, id): url=f"{self.protocol}://{self.site}/{id}/info" - return self.get(url) + self.user_information = self.get(url) + return result ''' I have no clue what this is supposed to do, lol. '''