xevion/contest-assistant
1
0
Fork 0
mirror of https://github.com/Xevion/contest-assistant.git synced 2025-12-12 01:09:38 -06:00
Code Issues Packages Projects Releases Wiki Activity

Split ContestCog into ContestCommandsCog and ContestEventsCog

Browse Source 
expected_react_deletions and expected_msg_deletions were moved into bot
.py
There may be more edits included in this that I cannot see.
Added docstrings to get_message and fetch_message
This commit is contained in:
Xevion
2021-02-18 01:35:33 -06:00
parent 39b0dc9a96
commit 1b777a3689
3 changed files with 192 additions and 161 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
bot/bot.py
View File

@@ -1,7 +1,7 @@
import logging
from contextlib import contextmanager
from datetime import datetime
from typing import ContextManager, List, Optional
from typing import ContextManager, List, Optional, Tuple
import discord
from discord.ext import commands
@@ -23,6 +23,9 @@ class ContestBot(commands.Bot):
self.engine = engine
self.Session = sessionmaker(bind=engine)
self.expected_msg_deletions: List[int] = []
self.expected_react_deletions: List[Tuple[int, int]] = []
@contextmanager
def get_session(self, autocommit=True, autoclose=True, rollback=True) -> ContextManager[Session]:
"""Provides automatic commit and closing of Session with exception rollback."""
@@ -108,10 +111,12 @@ class ContestBot(commands.Bot):
await message.add_reaction(self.get_emoji(constants.Emoji.UPVOTE))
def get_message(self, channel_id: int, message_id: int) -> discord.PartialMessage:
"""Get a PartialMessage object given raw integer IDs."""
channel: discord.TextChannel = self.get_channel(channel_id)
return channel.get_partial_message(message_id)
async def fetch_message(self, channel_id: int, message_id: int) -> discord.Message:
"""Fetch a full Message object given raw integer IDs."""
channel: discord.TextChannel = self.get_channel(channel_id)
return await channel.fetch_message(message_id)