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

Поддержка версии альбома

Browse source
This commit is contained in:
Lev 2022-12-11 18:12:28 +07:00
parent c294b43800
commit 422b0891f6
No known key found for this signature in database
GPG key ID: 21C6C2C9C0A4460D
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
main.py
View file

@ -41,14 +41,15 @@ class PlaylistId:
class BasicAlbumInfo: class BasicAlbumInfo:
id: str id: str
title: str title: str
version: Optional[str]
year: int year: int
artists: list[str] artists: list[str]
@staticmethod @staticmethod
def from_json(json: dict): def from_json(json: dict):
artists = [a['name'] for a in json['artists']] artists = [a['name'] for a in json['artists']]
return BasicAlbumInfo(id=json['id'], title=json['title'], year=json['year'], return BasicAlbumInfo(id=json['id'], title=json['title'], version=json.get('version', None),
artists=artists) year=json['year'], artists=artists)
@dataclass @dataclass
@ -312,8 +313,12 @@ if __name__ == '__main__':
print(f'Треков: {len(result_tracks)}') print(f'Треков: {len(result_tracks)}')
for track in result_tracks: for track in result_tracks:
if args.add_version and track.version is not None: if args.add_version:
if track.version is not None:
track.title = f'{track.title} ({track.version})' track.title = f'{track.title} ({track.version})'
album = track.album
if album.version is not None:
album.title = f'{album.title} ({track.album.version})'
save_path = prepare_track_path(args.dir, args.path_pattern, track) save_path = prepare_track_path(args.dir, args.path_pattern, track)
if args.skip_existing and save_path.is_file(): if args.skip_existing and save_path.is_file():