diff --git a/source/Track.py b/source/Track.py index d53c9ce..067457d 100644 --- a/source/Track.py +++ b/source/Track.py @@ -148,7 +148,12 @@ class Track: if self.score: if 0 < self.score <= 3: - star_text = "★" * self.score + "☆" * (3 - self.score) + "!" * getattr(self, "warning", 0) + star_text = "★" * self.score + "☆" * (3 - self.score) + if 0 < self.warning <= 3: + star_text += "!" * self.warning + if self.warning == 4: + star_text += "d" + star_text = trackname_color[star_text] + " " if self.since_version == highlight_version: diff --git a/source/definition.py b/source/definition.py index 7ec7811..9eed811 100644 --- a/source/definition.py +++ b/source/definition.py @@ -91,6 +91,9 @@ trackname_color = { "★★★!!": "\\\\c{YOR6}★★★\\\\c{off}", "★★☆!!": "\\\\c{YOR6}★★☆\\\\c{off}", "★☆☆!!": "\\\\c{YOR6}★☆☆\\\\c{off}", + "★★★d": "\\\\c{BLUE}★★★\\\\c{off}", + "★★☆d": "\\\\c{BLUE}★★☆\\\\c{off}", + "★☆☆d": "\\\\c{BLUE}★☆☆\\\\c{off}", } region_id_to_name = {