diff --git a/discussions.py b/discussions.py index 35aac2d..8538070 100644 --- a/discussions.py +++ b/discussions.py @@ -42,27 +42,40 @@ storage = datafile.data fetch_url = "https://services.fandom.com/discussion/{wikiid}/posts?sortDirection=descending&sortKey=creation_date&limit={limit}".format(wikiid=settings["fandom_discussions"]["wiki_id"], limit=settings["fandom_discussions"]["limit"]) -def embed_formatter(post): +def embed_formatter(post, post_type): """Embed formatter for Fandom discussions.""" embed = DiscordMessage("embed", "discussion") - embed["author"]["name"] = post["createdBy"]["name"] - embed["author"]["icon_url"] = post["createdBy"]["avatarUrl"] - embed["author"]["url"] = "{wikiurl}f/u/{creatorId}".format(wikiurl=settings["fandom_discussions"]["wiki_url"], creatorId=post["creatorId"]) - if post["isReply"]: - embed["title"] = _("Replied to \"{title}\"").format(title=post["_embedded"]["thread"][0]["title"]) - embed["url"] = "{wikiurl}f/p/{threadId}/r/{postId}".format(wikiurl=settings["fandom_discussions"]["wiki_url"], threadId=post["threadId"], postId=post["id"]) - else: - embed["title"] = _("Created \"{title}\"").format(title=post["title"]) - embed["url"] = "{wikiurl}f/p/{threadId}".format(wikiurl=settings["fandom_discussions"]["wiki_url"], threadId=post["threadId"]) - if settings["fandom_discussions"]["appearance"]["embed"]["show_content"]: - embed["description"] = post["rawContent"] if len(post["rawContent"]) < 2000 else post["rawContent"][0:2000] + "…" + embed.set_author(post["createdBy"]["name"], "{wikiurl}f/u/{creatorId}".format( + wikiurl=settings["fandom_discussions"]["wiki_url"], creatorId=post["creatorId"]), icon_url=post["createdBy"]["avatarUrl"]) + if post_type == "TEXT": # TODO + if post["isReply"]: + embed["title"] = _("Replied to \"{title}\"").format(title=post["_embedded"]["thread"][0]["title"]) + embed["url"] = "{wikiurl}f/p/{threadId}/r/{postId}".format( + wikiurl=settings["fandom_discussions"]["wiki_url"], threadId=post["threadId"], postId=post["id"]) + else: + embed["title"] = _("Created \"{title}\"").format(title=post["title"]) + embed["url"] = "{wikiurl}f/p/{threadId}".format(wikiurl=settings["fandom_discussions"]["wiki_url"], + threadId=post["threadId"]) + if settings["fandom_discussions"]["appearance"]["embed"]["show_content"]: + embed["description"] = post["rawContent"] if len(post["rawContent"]) < 2000 else post["rawContent"][ + 0:2000] + "…" + elif post_type == "POLL": + poll = post["poll"] + embed["title"] = _("Created a poll titled \"{}\"").format(poll["question"]) + image_type = False + if poll["answers"][0]["image"] is not None: + image_type = True + for num, option in enumerate(poll["answers"]): + embed.add_field(option["text"] if image_type is True else _("Option {}").format(num+1), + option["text"] if image_type is False else _("__[View image]({image_url})__").format(image_url=option["image"]["url"]), + inline=True) embed["footer"]["text"] = post["forumName"] embed["timestamp"] = datetime.datetime.fromtimestamp(post["creationDate"]["epochSecond"], tz=datetime.timezone.utc).isoformat() embed.finish_embed() send_to_discord(embed) -def compact_formatter(post): +def compact_formatter(post, post_type): """Compact formatter for Fandom discussions.""" message = None if not post["isReply"]: @@ -91,12 +104,23 @@ def fetch_discussions(): if request_json: for post in request_json: if int(post["id"]) > storage["discussion_id"]: - formatter(post) + parse_discussion_post(post) if int(post["id"]) > storage["discussion_id"]: storage["discussion_id"] = int(post["id"]) datafile.save_datafile() -def format_discussion_post(modal): +def parse_discussion_post(post): + """Initial post recognition & handling""" + post_type = post.get("funnel", "TEXT") + if post_type == "TEXT": + formatter(post, post_type) + elif post_type == "POLL": + formatter(post, post_type) + else: + discussion_logger.warning("The type of {} is an unknown discussion post type. Please post an issue on the project page to have it added https://gitlab.com/piotrex43/RcGcDw/-/issues.") + + +def format_discussion_text(modal): """This function converts fairly convoluted Fandom jsonModal of a discussion post into Markdown formatted usable thing. Takes string, returns string. Kudos to MarkusRost for allowing me to implement this formatter based on his code in Wiki-Bot.""" description = "" diff --git a/misc.py b/misc.py index cba04c8..dbc6c01 100644 --- a/misc.py +++ b/misc.py @@ -337,9 +337,16 @@ class DiscordMessage(): def __setup_embed(self): self.embed = defaultdict(dict) - self.webhook_object["embeds"] = [self.embed] + if "embeds" not in self.webhook_object: + self.webhook_object["embeds"] = [self.embed] + else: + self.webhook_object["embeds"].append(self.embed) self.embed["color"] = None + def add_embed(self): + self.finish_embed() + self.__setup_embed() + def finish_embed(self): if self.embed["color"] is None: if settings["appearance"]["embed"].get(self.event_type, {"color": None})["color"] is None: @@ -349,9 +356,10 @@ class DiscordMessage(): else: self.embed["color"] = math.floor(self.embed["color"]) - def set_author(self, name, url): + def set_author(self, name, url, icon_url=""): self.embed["author"]["name"] = name self.embed["author"]["url"] = url + self.embed["author"]["icon_url"] = icon_url def add_field(self, name, value, inline=False): if "fields" not in self.embed: