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 wiki_postbot.formats.wiki import WikiPage
from abc import abstractmethod
from discord.message import Message
import warnings
if TYPE_CHECKING:
from wiki_postbot.clients.slack import SlackMessage
try:
from discord.message import Message
except ImportError:
Message = None
class WikiTemplate(WikiPage):
@abstractmethod
def format_discord(self, msg:Message) -> str:
def format_discord(self, msg:'Message') -> str:
"""
Format a discord message into a template string
"""
@ -21,7 +25,7 @@ class WikiTemplate(WikiPage):
class TemplateMessage(WikiTemplate):
@classmethod
def format_discord(self, msg:Message) -> str:
def format_discord(self, msg:'Message') -> str:
# try/catch the avatar URL
try:
avatar = msg.author.avatar.url