From 71917bd56ce795c288576229207526660275c2c5 Mon Sep 17 00:00:00 2001
From: Lev <me@llistochek.ru>
Date: Tue, 6 Dec 2022 21:38:39 +0700
Subject: [PATCH] =?UTF-8?q?=D0=9E=D0=BF=D1=82=D0=B8=D0=BC=D0=B8=D0=B7?=
 =?UTF-8?q?=D0=B0=D1=86=D0=B8=D1=8F=20=D0=B7=D0=B0=D0=B3=D1=80=D1=83=D0=B7?=
 =?UTF-8?q?=D0=BA=D0=B8=20=D0=B2=D1=81=D0=B5=D1=85=20=D1=82=D1=80=D0=B5?=
 =?UTF-8?q?=D0=BA=D0=BE=D0=B2=20=D0=B8=D1=81=D0=BF=D0=BE=D0=BB=D0=BD=D0=B8?=
 =?UTF-8?q?=D1=82=D0=B5=D0=BB=D1=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/main.py b/main.py
index 9494b79..628c5d9 100644
--- a/main.py
+++ b/main.py
@@ -292,15 +292,15 @@ if __name__ == '__main__':
 
     if args.artist_id is not None:
         artist_info = get_artist_info(session, args.artist_id)
-        albums = [get_full_album_info(session, a.id) for a in artist_info.albums]
-        for album in albums:
-            if args.stick_to_artist and artist_info is not None:
-                if album.artists[0] != artist_info.name:
-                    albums.remove(album)
-                    continue
-            result_tracks.extend(album.tracks)
+        albums_count = 0
+        for album in artist_info.albums:
+            if args.stick_to_artist and album.artists[0] != artist_info.name:
+                continue
+            full_album = get_full_album_info(session, album.id)
+            result_tracks.extend(full_album.tracks)
+            albums_count += 1
         print(f'{artist_info.name}')
-        print(f'Альбомов: {len(albums)}')
+        print(f'Альбомов: {albums_count}')
     elif args.album_id is not None:
         album = get_full_album_info(session, args.album_id)
         result_tracks = album.tracks