catch absence of avatar

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "wiki-postbot"
-version = "0.1.1"
+version = "0.1.2"
 description = "Add posts to the wiki!"
 authors = ["sneakers-the-rat <JLSaunders987@gmail.com>"]
 license = "GPL-3.0"

wiki_postbot/templates/wiki.py

@@ -5,6 +5,7 @@ Templates for representing different kinds of messages on mediawiki
 from wiki_postbot.formats.wiki import WikiPage
 from abc import abstractmethod
 from discord.message import Message
+import warnings
 
 class WikiTemplate(WikiPage):
 
@@ -18,10 +19,17 @@ class TemplateMessage(WikiTemplate):
 
     @classmethod
     def format_discord(self, msg:Message) -> str:
+        # try/catch the avatar URL
+        try:
+            avatar = msg.author.avatar.url
+        except Exception as e:
+            warnings.warn("No avatar found!")
+            avatar = ""
+
         return (
             "{{Message\n"
             f"|Author={msg.author.name}\n"
-            f"|Avatar={msg.author.avatar.url}\n"
+            f"|Avatar={avatar}\n"
             f"|Date Sent={msg.created_at.strftime('%y-%m-%d %H:%M:%S')}\n"
             f"|Channel={msg.channel}\n"
             f"|Text={msg.content}\n"
@@ -30,37 +38,37 @@ class TemplateMessage(WikiTemplate):
             )
 
 
-
-template_message = TemplateMessage.from_source(
-    title="Template:Message",
-    source="""<noinclude>
-<pre>
-{{Message
-|Author=
-|Avatar=
-|Date Sent=
-|Channel=(Optional)
-|Text=
-|Link=
-}}
-</pre>
-</noinclude>
-<includeonly>
-
-{{#subobject:{{{Author}}}-{{{Date Sent}}}
-  |Message topic={{PAGENAME}}
-  |Has author={{{Author}}}
-  |Date sent={{{Date Sent}}}
-  |Has URL={{{Link}}}
-  |Contains text={{{Text}}}
-}}
-<div style="border: 1px solid black; border-radius: 5px;padding:5px"
->
-<div style="display:flex; flex-direction:row; align-items:center; border-bottom:1px solid black; gap:10px; padding-bottom:5px;"><img src={{{Avatar|}}} style="width:30px;border-radius:10px"/><span style="font-weight:bold;">{{{Author}}}</span><span><nowiki>#</nowiki>{{{Channel|}}}</span><span style="font-style:italic;color:#999999">[{{{Link}}} {{{Date Sent}}}]</span></div>
-<div>
-{{{Text}}}
-</div>
-</div>
-</includeonly>
-"""
-)
+#
+# template_message = TemplateMessage.from_source(
+#     title="Template:Message",
+#     source="""<noinclude>
+# <pre>
+# {{Message
+# |Author=
+# |Avatar=
+# |Date Sent=
+# |Channel=(Optional)
+# |Text=
+# |Link=
+# }}
+# </pre>
+# </noinclude>
+# <includeonly>
+#
+# {{#subobject:{{{Author}}}-{{{Date Sent}}}
+#   |Message topic={{PAGENAME}}
+#   |Has author={{{Author}}}
+#   |Date sent={{{Date Sent}}}
+#   |Has URL={{{Link}}}
+#   |Contains text={{{Text}}}
+# }}
+# <div style="border: 1px solid black; border-radius: 5px;padding:5px"
+# >
+# <div style="display:flex; flex-direction:row; align-items:center; border-bottom:1px solid black; gap:10px; padding-bottom:5px;"><img src={{{Avatar|}}} style="width:30px;border-radius:10px"/><span style="font-weight:bold;">{{{Author}}}</span><span><nowiki>#</nowiki>{{{Channel|}}}</span><span style="font-style:italic;color:#999999">[{{{Link}}} {{{Date Sent}}}]</span></div>
+# <div>
+# {{{Text}}}
+# </div>
+# </div>
+# </includeonly>
+# """
+# )