Add -ura switch for songs which have ura charts

This commit is contained in:
Cainan 2024-06-27 12:50:20 +01:00
parent dbee0bc540
commit 104e657a34

View File

@ -113,7 +113,7 @@ def generate_image(draw, text, font, rotated_font, size, position, alignment, st
else:
draw.text(text_position, text, font=font, fill=fill, stroke_width=stroke_width, stroke_fill=stroke_fill)
def create_images(data, id, genreNo, font_path, rotated_font_path):
def create_images(data, id, genreNo, font_path, rotated_font_path, append_ura=False):
font_size_extra_large = 46.06875
font_size_large = 40.60875
font_size_medium = 27.3
@ -154,6 +154,10 @@ def create_images(data, id, genreNo, font_path, rotated_font_path):
japanese_text = convert_full_width(japanese_text)
japanese_sub_text = convert_full_width(japanese_sub_text) if japanese_sub_text else ''
# Append "─" character if -ura argument is provided
if append_ura:
japanese_text += ""
# Check if texts were found
if not japanese_text:
print(f"Error: No Japanese text found for song_{id}")
@ -313,19 +317,24 @@ def create_images(data, id, genreNo, font_path, rotated_font_path):
if __name__ == "__main__":
if len(sys.argv) != 3:
print("Usage: generate.py <id> <genreNo>")
if len(sys.argv) < 3 or len(sys.argv) > 4:
print("Usage: generate.py <id> <genreNo> [-ura]")
sys.exit(1)
id = sys.argv[1]
try:
genreNo = int(sys.argv[2])
if (genreNo < 0 or genreNo >= 8):
if genreNo < 0 or genreNo >= 8:
raise ValueError
except ValueError:
print("Error: genreNo must be an integer between 0 and 7")
sys.exit(1)
if len(sys.argv) == 4 and sys.argv[3] == "-ura":
append_ura = True
else:
append_ura = False
wordlist_path = 'resources/wordlist.json'
font_path = 'resources/DFPKanTeiRyu-XB.ttf'
rotated_font_path = 'resources/KozGoPr6NRegular.otf'
@ -345,4 +354,4 @@ if __name__ == "__main__":
with open(wordlist_path, 'r', encoding='utf-8') as f:
data = json.load(f)
create_images(data, id, genreNo, font_path, rotated_font_path)
create_images(data, id, genreNo, font_path, rotated_font_path, append_ura)