feat(db): get_chart(song_id, rating_class)

This commit is contained in:
283375 2023-06-27 23:25:58 +08:00
parent 2a4c19eb72
commit 47db4ca120

View File

@ -151,6 +151,14 @@ class Database(metaclass=Singleton):
def get_charts_by_package_id(self, package_id: str): def get_charts_by_package_id(self, package_id: str):
return self.__get_table("charts", DbChartRow, "package_id = ?", (package_id,)) 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): def validate_song_id(self, song_id):
with self.conn as conn: with self.conn as conn:
cursor = conn.cursor() cursor = conn.cursor()