q/yandex-music-downloader
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Добавление тега "жанр" (#123)

Browse source
This commit is contained in:
leowerd 2025-06-14 22:12:03 +08:00 committed by GitHub
parent a84a40f38c
commit 6edae0e037
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
ymd/core.py
View file

@ -13,6 +13,7 @@ from mutagen.flac import FLAC, Picture
from mutagen.id3._frames import ( from mutagen.id3._frames import (
APIC, APIC,
TALB, TALB,
TCON,
TDRC, TDRC,
TIT2, TIT2,
TPE1, TPE1,
@ -161,16 +162,19 @@ def set_tags(
album_cover: Optional[AlbumCover], album_cover: Optional[AlbumCover],
compatibility_level: int, compatibility_level: int,
) -> None: ) -> None:
album = track.albums[0] if track.albums else Album()
track_artists = [a.name for a in track.artists if a.name]
album_artists = [a.name for a in album.artists if a.name]
file_type = CONTAINER_MUTAGEN_MAPPING.get(container) file_type = CONTAINER_MUTAGEN_MAPPING.get(container)
if file_type is None: if file_type is None:
raise ValueError(f"Unknown container: {container}") raise ValueError(f"Unknown container: {container}")
tag = file_type(path) tag = file_type(path)
album = track.albums[0] if track.albums else Album()
album_title = full_title(album) album_title = full_title(album)
track_title = full_title(track) track_title = full_title(track)
track_artists = [a.name for a in track.artists if a.name]
album_artists = [a.name for a in album.artists if a.name]
genre = None
if album.genre:
genre = album.genre
track_number = None track_number = None
disc_number = None disc_number = None
if position := album.track_position: if position := album.track_position:
@ -200,6 +204,8 @@ def set_tags(
tag["TRCK"] = TRCK(encoding=3, text=str(track_number)) tag["TRCK"] = TRCK(encoding=3, text=str(track_number))
if disc_number: if disc_number:
tag["TPOS"] = TPOS(encoding=3, text=str(disc_number)) tag["TPOS"] = TPOS(encoding=3, text=str(disc_number))
if genre:
tag["TCON"] = TCON(encoding=3, text=genre)
if lyrics: if lyrics:
tag["USLT"] = USLT(encoding=3, text=lyrics) tag["USLT"] = USLT(encoding=3, text=lyrics)
@ -234,6 +240,8 @@ def set_tags(
tag["trkn"] = [(track_number, 0)] tag["trkn"] = [(track_number, 0)]
if disc_number: if disc_number:
tag["disk"] = [(disc_number, 0)] tag["disk"] = [(disc_number, 0)]
if genre:
tag["\xa9gen"] = genre
if lyrics: if lyrics:
tag["\xa9lyr"] = lyrics tag["\xa9lyr"] = lyrics
@ -259,6 +267,8 @@ def set_tags(
tag["tracknumber"] = str(track_number) tag["tracknumber"] = str(track_number)
if disc_number: if disc_number:
tag["discnumber"] = str(disc_number) tag["discnumber"] = str(disc_number)
if genre:
tag["genre"] = genre
if lyrics: if lyrics:
tag["lyrics"] = lyrics tag["lyrics"] = lyrics