32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
import aiohttp
|
|
|
|
|
|
class UserRequests:
|
|
def __init__(self, API_URL, BOT_TOKEN):
|
|
self.API_URL = API_URL
|
|
self.headers = {"Content-Type": "application/json", "X-BOTAUTH": BOT_TOKEN}
|
|
|
|
async def getUser(self, user_id):
|
|
async with aiohttp.ClientSession(self.API_URL, headers=self.headers) as session:
|
|
# Make sure we have a good connection
|
|
async with session.head("/") as alive:
|
|
if alive.status != 200:
|
|
return None
|
|
|
|
async with session.get(f"/api/user/{user_id}") as data:
|
|
response = await data.json()
|
|
return response
|
|
|
|
async def registerUser(self, user_data: dict):
|
|
async with aiohttp.ClientSession(self.API_URL, headers=self.headers) as session:
|
|
# Make sure we have a good connection
|
|
async with session.head("/") as alive:
|
|
if alive.status != 200:
|
|
return None
|
|
|
|
async with session.post("/api/user/register", json=user_data) as data:
|
|
if data.status != 200:
|
|
return {"status": False}
|
|
response = await data.json()
|
|
return response
|