From 8a3a47a8afd3e20df280f4b717e0806356947095 Mon Sep 17 00:00:00 2001 From: Luna Date: Sat, 10 Jun 2023 00:50:38 -0300 Subject: [PATCH] inject rating tag into post.tag_string --- main.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 5ca42aa..eeec124 100644 --- a/main.py +++ b/main.py @@ -194,7 +194,21 @@ async def fetch_post(ctx, md5) -> Optional[dict]: if not rows: return None assert len(rows) == 1 - return json.loads(rows[0][0]) + post = json.loads(rows[0][0]) + post_rating = post["rating"] + match post_rating: + case "g": + rating_tag = "general" + case "s": + rating_tag = "sensitive" + case "q": + rating_tag = "questionable" + case "e": + rating_tag = "explicit" + case _: + raise AssertionError("invalid post rating {post_rating!r}") + post["tag_string"] = post["tag_string"] + " " + rating_tag + return post async def insert_post(ctx, post):