fix vk audio

feeds/modules/vk_music.py

@@ -11,31 +11,10 @@ from vk_api import VkApi
 from vk_api.audio import VkAudio
 
 from feeds.models import FeedModuleConfig
+from feeds.modules.utils import get_vk_audio
 from feeds.utils import DatabaseConfig, captcha_handler
 
 
-def get_vk_photo(attachment):
-    for size in (2560, 1280, 807, 604, 130, 75):
-        if f'photo_{size}' in attachment:
-            return attachment[f'photo_{size}']
-    return None
-
-
-def get_file(url, session=None):
-    if session is None:
-        session = requests
-
-    fname = '?'.join(url.split('?')[:-1])
-    extension = os.path.basename(fname).split('.')[-1]
-    f = tempfile.NamedTemporaryFile(suffix=f'.{extension}' if extension else None)
-    r = session.get(url, stream=True)
-    for chunk in r.iter_content(1024 * 1024):
-        if chunk:
-            f.write(chunk)
-    f.seek(0)
-    return f
-
-
 class VKMusicFeedModuleConfig(FeedModuleConfig):
     owner_id = models.IntegerField()
 
@@ -63,7 +42,7 @@ class VKMusicFeedModuleConfig(FeedModuleConfig):
             else:
                 break
         for track in lst:
-            f = get_file(track['url'], vk_session.http)
+            f = get_vk_audio(track['url'])
             bot.send_audio(chat_id, f, duration=track['duration'], performer=track['artist'],
                            title=track['title'], timeout=40)
             yield track['id']