From 019c5ad1b55356612b347e55359e6fa413280121 Mon Sep 17 00:00:00 2001 From: Luna Date: Sun, 18 Sep 2022 16:20:33 -0300 Subject: [PATCH] fix e621 api compliance bugs --- e621_api_cloner.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/e621_api_cloner.py b/e621_api_cloner.py index 79b8c09..a19edf6 100644 --- a/e621_api_cloner.py +++ b/e621_api_cloner.py @@ -84,6 +84,7 @@ class Post: } post.pop("up_score") post.pop("down_score") + post["file"] = {"md5": post.pop("md5")} post["flags"] = { "pending": post.pop("is_pending"), "flagged": post.pop("is_flagged"), @@ -109,7 +110,7 @@ class TagCategory(enum.IntEnum): self.ARTIST: "artist", self.COPYRIGHT: "copyright", self.CHARACTER: "character", - self.METADATA: "metadata", + self.METADATA: "meta", self.DEPRECATED: "deprecated", self.SPECIES: "species", self.LORE: "lore", @@ -164,15 +165,13 @@ async def posts_json(): post = Post(**rows[0]) post_json = post.to_json() - post_json["tags"] = {} + post_json["tags"] = {category.to_string(): [] for category in TagCategory} for tag in post.tag_string.split(" "): tag_rows = await app.db.execute_fetchall( "select category from tags where name = ?", (tag,) ) category = TagCategory(tag_rows[0][0]) category_str = category.to_string() - if category_str not in post_json["tags"]: - post_json["tags"][category_str] = [] post_json["tags"][category_str].append(tag) return {"posts": [post_json]}