Refactor VkAudio

Author: python273

examples/get_all_audio.py

@@ -1,6 +1,8 @@
 # -*- coding: utf-8 -*-
 import collections
+
 import vk_api
+from vk_api.audio import VKAudio
 
 
 def main():
@@ -15,41 +17,37 @@ def main():
         print(error_msg)
         return
 
-    vkaudio = vk_session.get_audio()
-
-    """
-        VkApi.audio.get и VkApi.audio.search делает запросы к странице m.vk.com
-        и извлекает информацию из html. Необходимый параметр для метода search - 'q',
-        а для get - 'owner_id'. Параметр offset принимает стандартное значение - 0.
-        Здесь мы получим все аудиозаписи пользователя в цикле с использованием параметра offset
-        и составим рейтинг самых популярных исполнителей для данного профиля.
-        Ниже мы выведем список из 10 песен самого популярного
-    """
+    vkaudio = VKAudio(vk_session)
 
-    count = 0
     artists = collections.Counter()
+
+    offset = 0
+
     while True:
-        audios = vkaudio.get(owner_id=-99463083, offset=count)
-        print('offset=', count)
+        audios = vkaudio.get(owner_id=-99463083, offset=offset)
+
+        if not audios:
+            break
+
         for audio in audios:
             artists[audio['artist']] += 1
 
-        count += len(audios)
+        offset += len(audios)
 
-        if len(audios) == 0:
-            break
-
-    # состаляем рейтинг первых 15
+    # Составляем рейтинг первых 15
     print('\nTop 15:')
     for artist, tracks in artists.most_common(15):
-        print('{} - {} tracks added'.format(artist, tracks))
-
-    # ищем треки самого популярного
-    most_common = artists.most_common(1)[0][0]
-    print('\nSearch for ', most_common)
-    tracks = vkaudio.search(q=most_common)
-    for n, track in enumerate(tracks[:10]):
-        print('{}. {}\nURL: {}'.format(n+1, track['title'], track['url']))
+        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

@@ -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,7 +15,7 @@ class VkApiError(Exception):
     pass
 
 
-class AccessDenied (VkApiError):
+class AccessDenied(VkApiError):
     pass
 
 

vk_api/vk_api.py

@@ -14,7 +14,6 @@
 import requests
 
 import jconfig
-from vk_api.vk_audio import VKAudio
 from .exceptions import *
 from .utils import code_from_number, search_re, clean_string
 
@@ -378,9 +377,6 @@ def auth_handler(self):
     def get_api(self):
         return VkApiMethod(self)
 
-    def get_audio(self):
-        return VKAudio(self)
-
     def method(self, method, values=None, captcha_sid=None, captcha_key=None, raw=False):
         """ Использование методов API