`{track_volume_num_optional}` and `{track_volume_num_optional_CD}` added for an optional sub CD folder structure. Fixes #151

1 files changed, 8 insertions, 0 deletions

diff --git a/tidal_dl_ng/helper/path.py b/tidal_dl_ng/helper/path.py
index 4be06e7..3d72cdf 100644
--- a/tidal_dl_ng/helper/path.py
+++ b/tidal_dl_ng/helper/path.py
@@ -155,6 +155,14 @@ def format_str_media(name: str, media: Track | Album | Playlist | UserPlaylist |
             case "track_volume_num":
                 if isinstance(media, Track | Video):
                     result = str(media.volume_num)
+            case "track_volume_num_optional":
+                if isinstance(media, Track | Video):
+                    num_volumes: int = media.album.num_volumes if hasattr(media, "album") else 1
+                    result = "" if media.volume_num is num_volumes else str(media.volume_num)
+            case "track_volume_num_optional_CD":
+                if isinstance(media, Track | Video):
+                    num_volumes: int = media.album.num_volumes if hasattr(media, "album") else 1
+                    result = "" if media.volume_num is num_volumes else f"CD{media.volume_num!s}"
     except Exception as e:
         # TODO: Implement better exception logging.
         print(e)