michel/ft_transcendence
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: post method, csrf token and return right valu

Browse Source
This commit is contained in:
starnakin 2023-11-03 21:51:57 +01:00
parent 2e680d94b6
commit f14279b956
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/Client.py
View File

@ -13,18 +13,15 @@ class Client:
return (not self.token is None) return (not self.token is None)
def login(self, username: str, password: str): def login(self, username: str, password: str):
response: Response = self._post(urls.accounts_login_url, {'username': username, 'password': password}) response: Response = self._post(urls.accounts_login, {'username': username, 'password': password})
return response.content return response.content
def register(self, username: str, password: str): def _post(self, uri: str, data: dict = {}):
response: Response = self._post(urls.accounts_register_url, {'username': username, 'password': password})
return response
def _post(self, uri: str, data: dict):
url: str = self.url + uri url: str = self.url + uri
if self.token is None: if self.token is None:
response: Response = self.session.get(url) response: Response = self.session.get(url)
self.token = response.cookies['csrftoken'] self.token = response.cookies.get('csrftoken')
data.update({'csrfmiddlewaretoken': self.token}) data.update({'csrfmiddlewaretoken': self.token})
response: Response = self.session.post(url, data = data, headers = dict(Referer=url)) response: Response = self.session.post(url, data = data, headers = dict(Referer=url))
return response.content self.token = response.cookies.get('csrftoken')
return response