From 47db4ca1205b24a0369b56b82550b2972a4fe982 Mon Sep 17 00:00:00 2001 From: 283375 Date: Tue, 27 Jun 2023 23:25:58 +0800 Subject: [PATCH] feat(db): `get_chart(song_id, rating_class)` --- src/arcaea_offline/database.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/arcaea_offline/database.py b/src/arcaea_offline/database.py index 81efec0..b0ace06 100644 --- a/src/arcaea_offline/database.py +++ b/src/arcaea_offline/database.py @@ -151,6 +151,14 @@ class Database(metaclass=Singleton): def get_charts_by_package_id(self, package_id: str): return self.__get_table("charts", DbChartRow, "package_id = ?", (package_id,)) + def get_chart(self, song_id: str, rating_class: int): + return self.__get_table( + "charts", + DbChartRow, + "song_id = ? AND rating_class = ?", + (song_id, rating_class), + )[0] + def validate_song_id(self, song_id): with self.conn as conn: cursor = conn.cursor()