tom/FAAPI
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add logout function

Browse Source
This commit is contained in:
Thomas Wade 2020-04-22 03:58:23 +09:30
parent ebcc589841
commit 3a1be7daf6
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
FAAPI/FAAPI.py
View File

@ -426,7 +426,7 @@ class FAAPI:
Initialises a new FAAPI instance. Initialises a new FAAPI instance.
""" """
self._cookies = None self._cookies = None
self._req = None self._req = requests.session()
self._driver = None self._driver = None
self.logged_in = False self.logged_in = False
self.username = None self.username = None
@ -457,8 +457,7 @@ class FAAPI:
else: else:
self._cookies = cookies self._cookies = cookies
# Set up a Requests session with the cookies we just got # Set up the Requests session with the cookies we just got
self._req = requests.session()
[self._req.cookies.set(cookie['name'], cookie['value']) for cookie in self._cookies] [self._req.cookies.set(cookie['name'], cookie['value']) for cookie in self._cookies]
# Verify login by getting our username # Verify login by getting our username
@ -474,6 +473,19 @@ class FAAPI:
self.logged_in = True self.logged_in = True
return True return True
def logout(self):
"""
Logs this instance out of FurAffinity.
"""
if not self.logged_in:
# No need to do anything if we're already logged out
return
# Reset our state and start a new session
self.logged_in = False
self._cookies = None
self._req = requests.session()
def get_user(self, username: str) -> User: def get_user(self, username: str) -> User:
soup = self.get_soup('{}/user/{}/'.format(FA_BASE_URL, username)) soup = self.get_soup('{}/user/{}/'.format(FA_BASE_URL, username))