Compare commits

..

No commits in common. "a13feba991eb63b7380472a8cde6a904283d393f" and "22ff3d58d3208ab8493a63425b401f2ed6c0ca40" have entirely different histories.

2 changed files with 7 additions and 9 deletions

View File

@ -9,13 +9,12 @@ class Client:
def __init__(self, url: str): def __init__(self, url: str):
self.url: str = url self.url: str = url
self.token: str = None self.token: str = None
self.csrf_token: str = None
self.session: Session = Session() self.session: Session = Session()
self.accounts: Accounts = Accounts(self) self.accounts: Accounts = Accounts(self)
self.profiles: Profiles = Profiles(self) self.profiles: Profiles = Profiles(self)
def is_authentificate(self): def is_authentificate(self):
return (self.token is not 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, {'username': username, 'password': password}) response: Response = self._post(urls.accounts_login, {'username': username, 'password': password})
@ -23,16 +22,16 @@ class Client:
def _post(self, uri: str, data: dict = {}): def _post(self, uri: str, data: dict = {}):
url: str = self.url + uri url: str = self.url + uri
if self.csrf_token is None: if self.token is None:
response: Response = self.session.get(url) response: Response = self.session.get(url)
self.csrf_token = response.cookies.get('csrftoken') self.token = response.cookies.get('csrftoken')
data.update({'csrfmiddlewaretoken': self.csrf_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))
self.csrf_token = response.cookies.get('csrftoken') self.token = response.cookies.get('csrftoken')
return response return response
def _get(self, uri: str): def _get(self, uri: str):
url: str = self.url + uri url: str = self.url + uri
response: Response = self.session.get(url) response: Response = self.session.get(url)
self.csrf_token = response.cookies.get('csrftoken') self.token = response.cookies.get('csrftoken')
return response return response

View File

@ -1,7 +1,6 @@
def test(func: callable, params, expected_value, title: str, description = None): def test(func: callable, params, expected_value, title: str, description = None):
print(title, end=" ") print(title, end=" ")
value = func(*params) if (func(*params) == expected_value):
if (value == expected_value):
print("[OK]") print("[OK]")
return return
print ("[ERROR]") print ("[ERROR]")