Fixed double slashes in image urls

MujyKun · Aug 16, 2021 · b5076fe · b5076fe
1 parent 5c53a92
commit b5076fe
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/IreneUtility/models/idol.py b/IreneUtility/models/idol.py
@@ -50,20 +50,23 @@ async def send_images_to_host(self):
         file_name = f"{self.id}_IDOL.png"
         if self.thumbnail:
             file_loc = f"{base_util.ex.keys.idol_avatar_location}{file_name}"
-            if 'images.irenebot.com' not in self.thumbnail:
+            if base_util.ex.keys.image_host not in self.thumbnail:
                 await base_util.ex.download_image(self.thumbnail, file_loc)
-            if base_util.ex.check_file_exists(file_loc):
-                image_url = f"{base_util.ex.keys.image_host}/avatar/{file_name}"
-                await base_util.ex.sql.s_groupmembers.set_member_thumbnail(self.id, image_url)
-                self.thumbnail = image_url
+                if base_util.ex.check_file_exists(file_loc):
+                    image_url = f"{base_util.ex.keys.image_host}avatar/{file_name}"
+                    await base_util.ex.sql.s_groupmembers.set_member_thumbnail(self.id, image_url)
+                    self.thumbnail = image_url
+            self.thumbnail = self.thumbnail.replace("//", "/")
+
         if self.banner:
             file_loc = f"{base_util.ex.keys.idol_banner_location}{file_name}"
-            if 'images.irenebot.com' not in self.banner:
+            if base_util.ex.keys.image_host not in self.banner:
                 await base_util.ex.download_image(self.banner, file_loc)
-            image_url = f"https://images.irenebot.com/banner/{file_name}"
-            if base_util.ex.check_file_exists(file_loc):
-                await base_util.ex.sql.s_groupmembers.set_member_banner(self.id, image_url)
-                self.banner = image_url
+                image_url = f"{base_util.ex.keys.image_host}banner/{file_name}"
+                if base_util.ex.check_file_exists(file_loc):
+                    await base_util.ex.sql.s_groupmembers.set_member_banner(self.id, image_url)
+                    self.banner = image_url
+            self.banner = self.banner.replace("//", "/")
 
     def set_attribute(self, column, content):
         """Sets the attribute for a column in the DB.