Merge branch 'dev'

AbhijeetKrishnan · Mar 24, 2024 · b7d8356 · b7d8356
2 parents 4f3d60b + 75743a7
commit b7d8356
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/frame_service/wavu/utils.py b/src/frame_service/wavu/utils.py
@@ -12,7 +12,7 @@
 from framedb.const import CharacterName
 
 WAVU_API_URL = "https://wavu.wiki/w/api.php"
-WAVU_VIDEO_LINK = "https://wavu.wiki/t/Special:Redirect/file/"
+WAVU_FILE_LINK = "https://wavu.wiki/t/Special:Redirect/file/"
 
 """Available fields for the Move table in the Wavu DB"""
 FIELDS = [
@@ -128,16 +128,19 @@ def _convert_json_move(move_json: Any) -> WavuMove:
     else:
         alt = ()
 
-    if "image" in move_json:
+    if "image" in move_json and move_json["image"]:
         image = _process_links(move_json["image"])
+        image_name = image.split(":")[-1]
+        image_name = image_name.replace(" ", "_")
+        image = WAVU_FILE_LINK + image_name
     else:
         image = ""
 
     if "video" in move_json and move_json["video"]:
         video = _process_links(move_json["video"])
         video_name = video.split(":")[-1]
         video_name = video_name.replace(" ", "_")
-        video = WAVU_VIDEO_LINK + video_name
+        video = WAVU_FILE_LINK + video_name
     else:
         video = ""