IndiumRevolt/main.py
2023-06-19 17:13:16 -04:00

39 lines
1.1 KiB
Python

# pylint: disable=missing-module-docstring
import asyncio
import os
from dotenv import load_dotenv
import aiohttp
import revolt
from revolt.ext import commands
load_dotenv()
token = os.getenv('TOKEN')
api_url = os.getenv('API_URL')
prefix = os.getenv('PREFIX')
class Client(commands.CommandsClient):
# This class contains all of the commands the bot uses.
async def get_prefix(self, message: revolt.Message):
return prefix
@commands.command()
async def ping(self, ctx: commands.Context):
# Checks if the bot is running.
await ctx.send("Pong!")
@commands.command()
async def avatar(self, ctx: commands.Context, member: revolt.Member):
# Checks a user's avatar.
if not isinstance(member, revolt.Member):
await ctx.send("Please provide a member argument!")
return
avatar = member.avatar.url
await ctx.send(f"{avatar}")
async def main():
# This function logs into the bot user.
async with aiohttp.ClientSession() as session:
client = Client(session, token, api_url=api_url)
await client.start()
asyncio.run(main())