Finish fixing lint smells in loops directory.

This commit is contained in:
2025-03-03 23:36:07 -05:00
parent 9cf05f44d0
commit f43dccb12d
2 changed files with 9 additions and 9 deletions
+6 -7
View File
@@ -13,23 +13,22 @@ class BotPresence(commands.Cog, name="Bot Presence"):
self.bot = bot self.bot = bot
@commands.Cog.listener() @commands.Cog.listener()
async def on_guild_join(self, guild: nextcord.Guild): # pylint: disable=unused-argument async def on_guild_join(self, guild: nextcord.Guild) -> None: # noqa: ARG002
"""Listener for Guild Join. Updated presence on guild join.""" """Listener for Guild Join. Updated presence on guild join."""
await self.change_presence() await self.change_presence()
@commands.Cog.listener() @commands.Cog.listener()
async def on_ready(self): async def on_ready(self) -> None:
"""Starts the loop when bot is ready.""" """Starts the loop when bot is ready."""
await self.bot.wait_until_ready() await self.bot.wait_until_ready()
# pylint doesn't recognize that periodic_change_presense is subclassed from tasks.loop. self.periodic_change_presence.start()
self.periodic_change_presence.start() # pylint: disable=no-member
@tasks.loop(minutes=2) @tasks.loop(minutes=2)
async def periodic_change_presence(self): async def periodic_change_presence(self) -> None:
"""Loops every 2 minutes and changes presence.""" """Loops every 2 minutes and changes presence."""
await self.change_presence() await self.change_presence()
async def change_presence(self): async def change_presence(self) -> None:
"""Randomly chooses a presence and updates bot's presence.""" """Randomly chooses a presence and updates bot's presence."""
statuses = [ statuses = [
"Now with slash commands!", "Now with slash commands!",
@@ -39,7 +38,7 @@ class BotPresence(commands.Cog, name="Bot Presence"):
"That's a lot of damage!", "That's a lot of damage!",
] ]
status = random.choice(statuses) status = random.choice(statuses) # noqa: S311
status += f" | {len(self.bot.guilds)} servers" status += f" | {len(self.bot.guilds)} servers"
+3 -2
View File
@@ -3,6 +3,7 @@
from nextcord.ext import commands, tasks from nextcord.ext import commands, tasks
from sqlalchemy import select from sqlalchemy import select
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from sqlmodel import col
from database.db import engine from database.db import engine
from database.models import Guild from database.models import Guild
@@ -11,7 +12,7 @@ from database.models import Guild
class GuildData(commands.Cog, name="Guild Data"): class GuildData(commands.Cog, name="Guild Data"):
"""Class to contain loop.""" """Class to contain loop."""
def __init__(self, bot: commands.bot) -> None: def __init__(self, bot: commands.Bot) -> None:
self.bot = bot self.bot = bot
@commands.Cog.listener() @commands.Cog.listener()
@@ -30,7 +31,7 @@ class GuildData(commands.Cog, name="Guild Data"):
with Session(engine) as session: with Session(engine) as session:
for guild in self.bot.guilds: for guild in self.bot.guilds:
# update given guild name for guild id # update given guild name for guild id
stmt = select(Guild).where(Guild.id == guild.id) stmt = select(Guild).where(col(Guild.id == guild.id))
guild_data = session.scalar(stmt) guild_data = session.scalar(stmt)
# just in case checking if the guild doesn't exist. # just in case checking if the guild doesn't exist.