Added iter for DB

This commit is contained in:
2022-09-02 09:47:18 +03:00
parent 646008e4d8
commit 99bf0769e5
7 changed files with 46 additions and 26 deletions

View File

@@ -3,7 +3,7 @@ import disnake
from disnake import Option, OptionType, Colour
from disnake.ext import commands
from lib import read_db
from lib import DB_Reader
from lib import logger
@@ -23,15 +23,20 @@ class General(commands.Cog):
]
)
async def info(self, inter, user=None):
user_audio = None
default_audio = None
user = user or inter.author
user_audio = await read_db(inter.guild.id, user.id, column='usertracks')
default_audio = await read_db(inter.guild.id, user.id, column='defaulttracks')
_user = DB_Reader(inter.guild.id)
for r in _user:
if r.userid == user.id:
user_audio = r.usertracks
default_audio = r.defaulttracks
rolelist = [r.mention for r in user.roles if r != inter.guild.default_role]
if rolelist:
roles = "\n".join(rolelist)
else:
roles = "Not added any role"
if user_audio:
audios = "" + "\n".join(sorted(user_audio.split(", ")))
else:

View File

@@ -4,6 +4,7 @@ import disnake
import psutil
from disnake.ext import commands
from __init__ import version_info as _ve
from lib import determine_prefix, determine_time
from lib import logger
@@ -15,7 +16,6 @@ class Bot_info(commands.Cog, name='Bot Info'):
@commands.Cog.listener() # this is a decorator for events/listeners
async def on_ready(self):
logger.info(f'Cog {__name__.split(".")[1]} is ready!.')
logger.info(f'{[names.name for names in self.bot.users]}')
@commands.slash_command(name="info_bot",
description='Shows general info about bot') # this is for making a command
@@ -32,7 +32,8 @@ class Bot_info(commands.Cog, name='Bot Info'):
f'Prefix: `{determine_prefix(self.bot, inter)}\n`'
f"Max audio duration: {determine_time(inter)} sec\n"
)
emb.add_field(name="Bot owner", value=f"<@{self.bot.owner_id}>")
emb.add_field(name="Bot info:", value=f"Bot owner: <@{self.bot.owner_id}>\n"
f"Bot version: {_ve.major}.{_ve.minor}.{_ve.micro}-{_ve.releaselevel}")
emb.set_footer(text="Information requested by: {}".format(inter.author.display_name))
await inter.response.send_message(embed=emb, ephemeral=True)