fix: only use chobit covers when the preview file type is audio #8

bump to version v0.6.3
NobeKanai · Apr 3, 2024 · 9d12b4b · 9d12b4b
1 parent 5f3911b
commit 9d12b4b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/dvtag/doujinvoice.py b/dvtag/doujinvoice.py
@@ -80,7 +80,8 @@ def _get_cover(self):
 
             res = json.loads(session.get(chobit_api).text[9:-1])
 
-            self.work_image = res["works"][0]["thumb"].replace("media.dlsite.com/chobit", "file.chobit.cc", 1)
+            if (work := res["works"][0])["file_type"] == "audio":
+                self.work_image = work["thumb"].replace("media.dlsite.com/chobit", "file.chobit.cc", 1)
 
         except Exception as e:
             logging.warning(f"Cannot fetch cover from chobit for {self.rjid}: {e}")
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = dvtag
-version = 0.6.2
+version = 0.6.3
 author = Nobe Kanai
 author_email = nobekanai@gmail.com
 description = A tool for tagging your doujin voice library.