Skip to content

Commit

Permalink
fix: expect latest item but cached ones got
Browse files Browse the repository at this point in the history
  • Loading branch information
geniucker-dev committed May 18, 2024
1 parent 917e333 commit a79b8b1
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions zjuintl_assistant/assistant.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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


Expand Down Expand Up @@ -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


Expand Down

0 comments on commit a79b8b1

Please sign in to comment.