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

dont import discord message for type checking

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