Split mino into central & us variants (#14)

osuAkatsuki · Jul 1, 2024 · 1c1e96f · 1c1e96f
1 parent 84ad926
commit 1c1e96f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/app/adapters/osu_mirrors/__init__.py b/app/adapters/osu_mirrors/__init__.py
@@ -2,8 +2,8 @@
 from datetime import datetime
 
 from app.adapters.osu_mirrors.backends import AbstractBeatmapMirror
-from app.adapters.osu_mirrors.backends import BeatmapMirrorResponse
-from app.adapters.osu_mirrors.backends.mino import MinoMirror
+from app.adapters.osu_mirrors.backends.mino import MinoCentralMirror
+from app.adapters.osu_mirrors.backends.mino import MinoUSMirror
 from app.adapters.osu_mirrors.backends.nerinyan import NerinyanMirror
 from app.adapters.osu_mirrors.backends.osu_direct import OsuDirectMirror
 from app.adapters.osu_mirrors.selectors.dynamic_round_robin import (
@@ -15,7 +15,8 @@
 ZIP_FILE_HEADER = b"PK\x03\x04"
 
 BEATMAP_MIRRORS: list[AbstractBeatmapMirror] = [
-    MinoMirror(),
+    MinoCentralMirror(),
+    MinoUSMirror(),
     NerinyanMirror(),
     OsuDirectMirror(),
     # GatariMirror(),  # Disabled as ratelimit is very low

diff --git a/app/adapters/osu_mirrors/backends/mino.py b/app/adapters/osu_mirrors/backends/mino.py
@@ -11,7 +11,7 @@
 
 class MinoMirror(AbstractBeatmapMirror):
     name = "mino"
-    base_url = "https://central.catboy.best"
+    base_url = "https://catboy.best"
     supported_resources = {MirrorResource.OSZ_FILE, MirrorResource.BACKGROUND_IMAGE}
 
     @override
@@ -87,3 +87,13 @@ async def fetch_beatmap_background_image(
                 status_code=response.status_code if response else None,
                 error_message=str(exc),
             )
+
+
+class MinoCentralMirror(MinoMirror):
+    name = "mino-central"
+    base_url = "https://central.catboy.best"
+
+
+class MinoUSMirror(MinoMirror):
+    name = "mino-us"
+    base_url = "https://us.catboy.best"