render_chart_official now support int as difficulty

zhanbao2000 · Apr 13, 2023 · 3fc6250 · 3fc6250
1 parent 54ac880
commit 3fc6250
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/__init__.py b/__init__.py
@@ -1,3 +1,5 @@
+from typing import Union
+
 from .model import DifficultyInt
 from .render import Render
 from .resource import (
@@ -11,7 +13,7 @@
 )
 
 
-async def render_chart_official(song_id: int, difficulty: DifficultyInt) -> Render:
+async def render_chart_official(song_id: int, difficulty: Union[DifficultyInt, int]) -> Render:
     chart = await get_chart_official(song_id, difficulty)
     song = await get_song_official(song_id)
     jacket = await get_song_jacket(get_song_jacket_url_official(song_id, song.jacketImage[0]))

diff --git a/resource.py b/resource.py
@@ -36,7 +36,7 @@ class FontResourceMangaer(object):
     font_a_otf_shingopro_medium_2 = assets / 'A-OTF-ShinGoPro-Medium-2.otf'
 
 
-async def get_chart_official(song_id: int, difficulty: int) -> Chart:
+async def get_chart_official(song_id: int, difficulty: DifficultyInt) -> Chart:
     async with get_client() as client:
         response = await client.get(f'https://bestdori.com/api/charts/{song_id}/{difficulty_literal[difficulty]}.json')
         response.raise_for_status()