Add support for downloading Music Albums

Author: vanyasem

examples/get_album_audio.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+import collections
+
+import vk_api
+from vk_api.audio import VkAudio
+
+
+def main():
+    """ Пример составления топа исполнителей для альбома вк """
+
+    login, password = 'login', 'password'
+    vk_session = vk_api.VkApi(login, password)
+
+    try:
+        vk_session.auth()
+    except vk_api.AuthError as error_msg:
+        print(error_msg)
+        return
+
+    vkaudio = VkAudio(vk_session)
+
+    artists = collections.Counter()
+
+    offset = 0
+
+    while True:
+        audios = vkaudio.get(album_id='-89975290_74038117', offset=offset)
+
+        if not audios:
+            break
+
+        for audio in audios:
+            artists[audio['artist']] += 1
+
+        offset += len(audios)
+
+    # Составляем рейтинг первых 15
+    print('\nTop 15:')
+    for artist, tracks in artists.most_common(15):
+        print('{} - {} tracks'.format(artist, tracks))
+
+    # Ищем треки самого популярного
+    most_common_artist = artists.most_common(1)[0][0]
+
+    print('\nSearch for', most_common_artist)
+
+    tracks = vkaudio.search(q=most_common_artist)[:10]
+
+    for n, track in enumerate(tracks, 1):
+        print('{}. {} {}'.format(n, track['title'], track['url']))
+
+
+if __name__ == '__main__':
+    main()

vk_api/audio.py

@@ -16,15 +16,27 @@ class VkAudio:
     def __init__(self, vk):
         self._vk = vk
 
-    def get(self, owner_id, offset=0):
+    def get(self, owner_id=None, album_id=None, offset=0):
         """ Получить список аудиозаписей пользователя
 
         :param owner_id: ID владельца (отрицательные значения для групп)
+        :param album_id: ID альбома (отрицательные значения для групп)
         :param offset: смещение
         """
 
+        if owner_id is None and album_id is None:
+            raise TypeError("get() missing 1 required argument: 'album_id' or 'owner_id'")
+        elif owner_id is not None and album_id is not None:
+            raise TypeError("get() too many arguments")
+
+        id = owner_id
+        url = "https://m.vk.com/audios{}"
+        if album_id is not None:
+            id = album_id
+            url = "https://m.vk.com/audio?act=audio_playlist{}"
+
         response = self._vk.http.get(
-            'https://m.vk.com/audios{}'.format(owner_id),
+            url.format(id),
             params={
                 'offset': offset
             },
@@ -34,7 +46,7 @@ def get(self, owner_id, offset=0):
         if not response.text:
             raise AccessDenied(
                 'You don\'t have permissions to browse {}\'s audio'.format(
-                    owner_id
+                    id
                 )
             )
 
@@ -92,8 +104,7 @@ def scrap_data(html):
 
     soup = BeautifulSoup(html, 'html.parser')
     tracks = []
-
-    for audio in soup.find_all('div', {'class': 'audio_item ai_has_btn'}):
+    for audio in soup.find_all('div', {'class': 'audio_item'}):
         ai_artist = audio.select('.ai_artist')
         artist = ai_artist[0].text
         link = audio.select('.ai_body')[0].input['value']