jonny/wiki-postbot
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity

dont import discord message for type checking

Browse source
This commit is contained in:
sneakers-the-rat 2023-10-16 16:37:33 -07:00
parent dace014779
commit ae4d610754
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
wiki_postbot/templates/wiki.py
View file

@ -4,16 +4,20 @@ Templates for representing different kinds of messages on mediawiki
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
from wiki_postbot.formats.wiki import WikiPage from wiki_postbot.formats.wiki import WikiPage
from abc import abstractmethod from abc import abstractmethod
from discord.message import Message
import warnings import warnings
if TYPE_CHECKING: if TYPE_CHECKING:
from wiki_postbot.clients.slack import SlackMessage from wiki_postbot.clients.slack import SlackMessage
try:
from discord.message import Message
except ImportError:
Message = None
class WikiTemplate(WikiPage): class WikiTemplate(WikiPage):
@abstractmethod @abstractmethod
def format_discord(self, msg:Message) -> str: def format_discord(self, msg:'Message') -> str:
""" """
Format a discord message into a template string Format a discord message into a template string
""" """
@ -21,7 +25,7 @@ class WikiTemplate(WikiPage):
class TemplateMessage(WikiTemplate): class TemplateMessage(WikiTemplate):
@classmethod @classmethod
def format_discord(self, msg:Message) -> str: def format_discord(self, msg:'Message') -> str:
# try/catch the avatar URL # try/catch the avatar URL
try: try:
avatar = msg.author.avatar.url avatar = msg.author.avatar.url