From 6edae0e0379b0b449956155605b82200f143a39e Mon Sep 17 00:00:00 2001
From: leowerd <mail@werdes.ru>
Date: Sat, 14 Jun 2025 22:12:03 +0800
Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?=
 =?UTF-8?q?=D0=BD=D0=B8=D0=B5=20=D1=82=D0=B5=D0=B3=D0=B0=20"=D0=B6=D0=B0?=
 =?UTF-8?q?=D0=BD=D1=80"=20(#123)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ymd/core.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/ymd/core.py b/ymd/core.py
index 3f36d9e..8c32a4f 100644
--- a/ymd/core.py
+++ b/ymd/core.py
@@ -13,6 +13,7 @@ from mutagen.flac import FLAC, Picture
 from mutagen.id3._frames import (
     APIC,
     TALB,
+    TCON,
     TDRC,
     TIT2,
     TPE1,
@@ -161,16 +162,19 @@ def set_tags(
     album_cover: Optional[AlbumCover],
     compatibility_level: int,
 ) -> 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)
     if file_type is None:
         raise ValueError(f"Unknown container: {container}")
+
     tag = file_type(path)
+    album = track.albums[0] if track.albums else Album()
     album_title = full_title(album)
     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
     disc_number = None
     if position := album.track_position:
@@ -200,6 +204,8 @@ def set_tags(
             tag["TRCK"] = TRCK(encoding=3, text=str(track_number))
         if disc_number:
             tag["TPOS"] = TPOS(encoding=3, text=str(disc_number))
+        if genre:
+            tag["TCON"] = TCON(encoding=3, text=genre)
 
         if lyrics:
             tag["USLT"] = USLT(encoding=3, text=lyrics)
@@ -234,6 +240,8 @@ def set_tags(
             tag["trkn"] = [(track_number, 0)]
         if disc_number:
             tag["disk"] = [(disc_number, 0)]
+        if genre:
+            tag["\xa9gen"] = genre
 
         if lyrics:
             tag["\xa9lyr"] = lyrics
@@ -259,6 +267,8 @@ def set_tags(
             tag["tracknumber"] = str(track_number)
         if disc_number:
             tag["discnumber"] = str(disc_number)
+        if genre:
+            tag["genre"] = genre
 
         if lyrics:
             tag["lyrics"] = lyrics