From a79b8b16908d3141aea73c1153e42258ed409ca3 Mon Sep 17 00:00:00 2001 From: geniucker-dev Date: Sat, 18 May 2024 23:07:02 +0800 Subject: [PATCH] fix: expect latest item but cached ones got --- zjuintl_assistant/assistant.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/zjuintl_assistant/assistant.py b/zjuintl_assistant/assistant.py index a5b97c5..80a0255 100644 --- a/zjuintl_assistant/assistant.py +++ b/zjuintl_assistant/assistant.py @@ -61,6 +61,20 @@ def get_cookie_jar(self, key: str = None, base: str = None) -> requests.cookies. return self.__cookie_jars[key] + def remove_cookie_jar(self, key: str = None): + """ + Remove cookie jar + """ + + if key is None: + key = inspect.currentframe().f_back.f_code.co_name + if key in self.__cookie_jars: + logger.debug(f"Removing {key} cookie jar") + self.__cookie_jars.pop(key) + else: + logger.debug(f"{key} cookie jar not found") + + def logout(self): """ Logout by clearing cookies and set related flags to False @@ -292,6 +306,9 @@ def get_bb_grades(self, count) -> list[data_classes.BB_Grade]: logger.debug("Get BB grades success") + # remove cookie jar + self.remove_cookie_jar() + return result @@ -364,6 +381,9 @@ def get_bb_announcements(self, count: int, full: bool = False) -> list[data_clas logger.debug("Get BB announcements success") + # remove cookie jar + self.remove_cookie_jar() + return result