Merge pull request #66 from neigrok/master

Add VkAudio

Author: python273

examples/get_all_audio.py

@@ -0,0 +1,53 @@
+# -*- 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(owner_id=-99463083, 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()

requirements.txt

@@ -1,2 +1,3 @@
+beautifulsoup4
 requests
 enum34

vk_api/audio.py

@@ -0,0 +1,73 @@
+# -*- coding: utf-8 -*-
+from bs4 import BeautifulSoup
+
+from .exceptions import AccessDenied
+
+
+class VKAudio:
+    def __init__(self, vk):
+        self._vk = vk
+
+    def get(self, owner_id, offset=0):
+        """ Получение списка аудиозаписей пользователя
+
+        :param owner_id: ID владельца (отрицательные значения для групп)
+        :param offset: смещение
+        """
+
+        response = self._vk.http.get(
+            'https://m.vk.com/audios{}'.format(owner_id),
+            params={
+                'offset': offset
+            },
+            allow_redirects=False
+        )
+
+        if not response.text:
+            raise AccessDenied(
+                'You don\'t have permissions to browse {}\'s audio'.format(
+                    owner_id
+                )
+            )
+
+        return scrap_data(response.text)
+
+    def search(self, q='', offset=0):
+        """ Поиск аудиозаписей
+
+        :param q: запрос
+        :param offset: смещение
+        """
+
+        response = self._vk.http.get(
+            'https://m.vk.com/audio',
+            params={
+                'act': 'search',
+                'q': q,
+                'offset': offset
+            }
+        )
+
+        return scrap_data(response.text)
+
+
+def scrap_data(html):
+    """ Парсинг списка аудиозаписей """
+
+    soup = BeautifulSoup(html, 'html.parser')
+    tracks = []
+
+    for audio in soup.find_all('div', {'class': 'audio_item ai_has_btn'}):
+        ai_artist = audio.select('.ai_artist')
+        artist = ai_artist[0].text
+        link = audio.select('.ai_body')[0].input['value']
+
+        tracks.append({
+            'artist': artist,
+            'title': audio.select('.ai_title')[0].text,
+            'dur': audio.select('.ai_dur')[0]['data-dur'],
+            'id': audio['id'],
+            'url': link
+        })
+
+    return tracks

vk_api/exceptions.py

@@ -15,6 +15,10 @@ class VkApiError(Exception):
     pass
 
 
+class AccessDenied(VkApiError):
+    pass
+
+
 class AuthorizationError(VkApiError):  # todo: delete
     pass
 

vk_api/vk_api.py

@@ -1,4 +1,4 @@
-# -*- coding: utf-8 -*-
+# -*- coding: utf-8 -*-
 """
 @author: Kirill Python
 @contact: https://vk.com/python273