diff --git a/src/Client.py b/src/Client.py index baca40e..21b4466 100644 --- a/src/Client.py +++ b/src/Client.py @@ -1,6 +1,7 @@ import requests from requests import Response, Request, Session +import Profile import Accounts import urls @@ -11,6 +12,7 @@ class Client: self.token: str = None self.session: Session = Session() self.accounts: Account = Accounts.Account(self) + self.profile = Profile.Profile def is_authentificate(self): return (not self.token is None) @@ -27,4 +29,10 @@ class Client: data.update({'csrfmiddlewaretoken': self.token}) response: Response = self.session.post(url, data = data, headers = dict(Referer=url)) self.token = response.cookies.get('csrftoken') + return response + + def _get(self, uri: str): + url: str = self.url + uri + response: Response = self.session.get(url) + self.token = response.cookies.get('csrftoken') return response \ No newline at end of file diff --git a/src/Profile.py b/src/Profile.py new file mode 100644 index 0000000..16d15d5 --- /dev/null +++ b/src/Profile.py @@ -0,0 +1,14 @@ +import urls +import Client + +from requests import Response + +class Profile: + + def __init__(self, client, user_id: int): + response: Response = client._get(urls.profiles_page + str(user_id)) + if (response.content == b'Profile Not Found'): + return None + content: dict = eval(response.content) + self.username = content['username'] + self.password = content['password'] \ No newline at end of file diff --git a/src/tests.py b/src/tests.py index f867f0a..5cbc4f2 100644 --- a/src/tests.py +++ b/src/tests.py @@ -35,6 +35,9 @@ def test_accounts_delete(client): print() +def test_profile_get(client): + pass + def tests(): username = uuid4() @@ -46,6 +49,8 @@ def tests(): test_accounts_register(client, username, password) test_accounts_login(client, username, password) test_accounts_delete(client) + + test_profile_get(client) if __name__ == "__main__": tests() \ No newline at end of file diff --git a/src/urls.py b/src/urls.py index 379524a..537ad46 100644 --- a/src/urls.py +++ b/src/urls.py @@ -1,4 +1,9 @@ -accounts: str = "api/accounts/" +api: str = "api/" + +accounts: str = api + "accounts/" accounts_login: str = accounts + "login" accounts_delete: str = accounts + "delete" -accounts_register: str = accounts + "register" \ No newline at end of file +accounts_register: str = accounts + "register" + +profiles: str = "profiles" +profiles_page: str = profiles \ No newline at end of file