init: profile

This commit is contained in:
starnakin 2023-11-03 22:22:24 +01:00
parent 06e95b9751
commit 4938e05c4a
4 changed files with 34 additions and 2 deletions

View File

@ -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

14
src/Profile.py Normal file
View File

@ -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']

View File

@ -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()

View File

@ -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"
accounts_register: str = accounts + "register"
profiles: str = "profiles"
profiles_page: str = profiles