mirror of
https://github.com/NovaOSS/nova-cord.git
synced 2024-11-25 21:33:58 +01:00
125 lines
3.9 KiB
Python
125 lines
3.9 KiB
Python
"""Account system functionality."""
|
|
|
|
import os
|
|
import json
|
|
import requests
|
|
|
|
import embedder
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
async def request_user_by_discord_id(discord_id):
|
|
return requests.get(
|
|
url=f'http://localhost:2333/users?discord_id={discord_id}',
|
|
timeout=3,
|
|
headers={
|
|
'Content-Type': 'application/json',
|
|
'Authorization': os.getenv('CORE_API_KEY')
|
|
}
|
|
)
|
|
|
|
async def get_account(interaction):
|
|
try:
|
|
user_response = await request_user_by_discord_id(interaction.user.id)
|
|
|
|
except Exception as exc:
|
|
await embedder.error(interaction, """Sorry, there was an error while checking if you have an account.
|
|
Please report this issue to the staff!""", ephemeral=True)
|
|
raise exc
|
|
|
|
if user_response.status_code == 404:
|
|
await embedder.error(interaction, """You don't have an account yet!""", ephemeral=True)
|
|
return
|
|
|
|
return user_response.json()
|
|
|
|
async def get_info(interaction):
|
|
account = await get_account(interaction)
|
|
|
|
await embedder.info(interaction, f"""### Your account
|
|
This is all we have stored about your API account in our database.
|
|
Feel free to request a removal of your account by contacting the staff.
|
|
|
|
||```json
|
|
{json.dumps(account, indent=4)}
|
|
```||
|
|
(Click to reveal)
|
|
|
|
Learn more about how to use our API at **https://nova-oss.com**.
|
|
""", ephemeral=True)
|
|
|
|
async def get_credits(interaction):
|
|
account = await get_account(interaction)
|
|
if not account:
|
|
return
|
|
|
|
amount_credits = account["credits"]
|
|
|
|
await embedder.info(interaction, f"""### Your credits
|
|
Amount: **{amount_credits if amount_credits < 1000000 else '∞'}**
|
|
""", ephemeral=True)
|
|
|
|
async def get_credits_of(interaction, user):
|
|
if not interaction.user.guild_permissions.administrator:
|
|
await embedder.error(interaction, """Sorry, you don't have the permission to do that.""", ephemeral=True)
|
|
return
|
|
|
|
try:
|
|
userinfo = await request_user_by_discord_id(user.id)
|
|
|
|
except Exception as exc:
|
|
await embedder.error(interaction, """Sorry, there was an error while checking if you have an account.
|
|
Please report this issue to the staff!""", ephemeral=True)
|
|
raise exc
|
|
|
|
if userinfo.status_code == 404:
|
|
await embedder.error(interaction, """You don't have an account yet!""", ephemeral=True)
|
|
return
|
|
|
|
account = userinfo.json()
|
|
amount_credits = account["credits"]
|
|
|
|
await embedder.info(interaction, f"""### Credits of {user.name}
|
|
Amount: **{amount_credits if amount_credits < 1000000 else '∞'}**
|
|
""", ephemeral=True)
|
|
|
|
async def set_credits(interaction, user, amount):
|
|
if not interaction.user.guild_permissions.administrator:
|
|
await embedder.error(interaction, """Sorry, you don't have the permission to do that.""", ephemeral=True)
|
|
return
|
|
|
|
try:
|
|
userinfo = await request_user_by_discord_id(user.id)
|
|
|
|
except Exception as exc:
|
|
await embedder.error(interaction, """Sorry, there was an error while checking if you have an account.
|
|
Please report this issue to the staff!""", ephemeral=True)
|
|
raise exc
|
|
|
|
if userinfo.status_code == 404:
|
|
await embedder.error(interaction, """You don't have an account yet!""", ephemeral=True)
|
|
return
|
|
|
|
account = userinfo.json()
|
|
account["credits"] = amount
|
|
|
|
try:
|
|
requests.put(
|
|
url=f'http://localhost:2333/users?discord_id={account["auth"]["discord"]}',
|
|
timeout=3,
|
|
headers={
|
|
'Content-Type': 'application/json',
|
|
'Authorization': os.getenv('CORE_API_KEY')
|
|
},
|
|
data=json.dumps(account)
|
|
)
|
|
|
|
except Exception as exc:
|
|
await embedder.error(interaction, """Sorry, there was an error while setting your credits.
|
|
Please report this issue to the staff!""", ephemeral=True)
|
|
raise exc
|
|
|
|
await embedder.ok(interaction, f"""Successfully set the credits of {user.name} to **{amount}**.""", ephemeral=True)
|