Finish fixing lint smells in loops directory.
This commit is contained in:
@@ -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,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.
|
||||||
|
|||||||
Reference in New Issue
Block a user