[gaana] Add new extractor

youtube_dl/extractor/extractors.py

@@ -398,6 +398,9 @@
 from .funk import FunkIE
 from .fusion import FusionIE
 from .fxnetworks import FXNetworksIE
+from .gaana import (
+    GaanaIE,
+)
 from .gaia import GaiaIE
 from .gameinformer import GameInformerIE
 from .gameone import (

youtube_dl/extractor/gaana.py

@@ -0,0 +1,194 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+import re
+import hashlib
+from ..aes import aes_cbc_decrypt
+from ..compat import (
+    compat_b64decode,
+)
+
+from .common import InfoExtractor
+from ..utils import (
+    bytes_to_intlist,
+    intlist_to_bytes,
+    int_or_none,
+)
+
+
+class GaanaBaseIE(InfoExtractor):
+    _BASE_URL = 'https://gaana.com'
+    _API_URL = 'https://apiv2.gaana.com/track/stream'
+    _COOKIE = ''
+
+    def _Decrypt(self, data):
+
+        key = b'g@1n!(f1#r.0$)&%'
+        iv = b'asd!@#!@#@!12312'
+        stream_url = intlist_to_bytes(aes_cbc_decrypt(
+            bytes_to_intlist(compat_b64decode(data)),
+            bytes_to_intlist(key),
+            bytes_to_intlist(iv))).decode()
+        # unpad
+        s = stream_url[:-ord(stream_url[len(stream_url) - 1:])]
+        return s
+
+    def _Create_ht(self, track_id):
+        if not self._COOKIE:
+            self._COOKIE = self._get_cookies(self._BASE_URL)['PHPSESSID'].value
+
+        mess = track_id + '|' + self._COOKIE + '|03:40:31 sec'
+        ht = hashlib.md5(mess.encode()).hexdigest() + self._COOKIE[3:9] + '='
+        return ht
+
+    def _create_entry(self, raw_data, video_id):
+        video_data = raw_data.get('path')
+        title = raw_data.get('title')
+        thumbnail = raw_data.get('albumartwork').replace("_175x175_", "_480x480_")
+        duration = raw_data.get('duration')
+        artist = raw_data.get('artist')
+
+        def _format_artist(art):
+            r_sample = r'#..(\d+)#..(\w+)[^|,]*'
+            res = re.sub(r_sample, '', art)
+            return re.sub(r',', ', ', res)
+
+        if artist:
+            artist = _format_artist(artist)
+
+        formats = []
+        if isinstance(video_data, dict):
+            for value in video_data.keys():
+                content = video_data.get(value)
+                for k in content:
+                    format_url = self._Decrypt(k.get('message'))
+
+                    if value == 'auto':
+                        format_id = 'normal'
+                    else:
+                        format_id = value
+
+                    info = {
+                        'url': format_url,
+                        'format_id': format_id,
+                        'ext': 'mp4',
+                        'abr': int_or_none(k.get('bitRate')),
+                        'format_note': 'mp4-aac'
+                    }
+
+                    if format_id == 'normal':
+                        formats.insert(0, info)
+                    else:
+                        formats.append(info)
+
+        else:
+            track_id = raw_data.get('track_ids')
+            ht = self._Create_ht(track_id)
+
+            for g in ('normal', 'medium', 'high'):
+                js = self._download_json(self._API_URL, title, headers={
+                    'async': '1',
+                    'method': 'POST'}, query={
+                    'ht': ht,
+                    'request_type': 'web',
+                    'track_id': track_id,
+                    'quality': g
+                })
+                format_url = js.get('stream_path')
+
+                formats.append({
+                    'url': format_url,
+                    'format_id': g,
+                    'ext': 'mp4',
+                    'abr': int_or_none(js.get('bit_rate')),
+                    'format_note': 'mp4-aac'
+                })
+
+        return {
+            'id': video_id,
+            'title': title,
+            'duration': int_or_none(duration),
+            'formats': formats,
+            'album': raw_data.get('albumtitle'),
+            'thumbnail': thumbnail,
+            'artist': artist,
+            'release_date': raw_data.get('release_date'),
+            'language': raw_data.get('language')
+        }
+
+
+class GaanaIE(GaanaBaseIE):
+    IE_NAME = 'gaana'
+    _VALID_URL = r'https?://(?:www\.)?gaana\.com/(?P<idtype>(song|album|artist|playlist))/(?P<id>[^/#?]+)'
+    _TESTS = [{
+        'url': 'https://gaana.com/song/chamma-chamma-5',
+        'md5': 'c94cc6896b24bc2b5bb3e5f496b0c809',
+        'info_dict': {
+            'id': '24725286',
+            'ext': 'mp4',
+            'title': 'Chamma Chamma',
+            'thumbnail': r'https://a10.gaanacdn.com/images/song/86/24725286/crop_480x480_1544754220.jpg',
+            'duration': 196,
+            'album': 'Fraud Saiyaan',
+            'artist': 'Neha Kakkar, Romi, Arun, Ikka',
+            'language': 'Hindi',
+            'release_date': 'Dec 14, 2018'
+        }
+    }, {
+        'url': 'https://gaana.com/song/saaun-di-jhadi-1',
+        'md5': 'f484453440d9c4edd2d320bd2ce7a069',
+        'info_dict': {
+            'id': '1742426',
+            'ext': 'mp4',
+            'title': 'Saaun Di Jhadi',
+            'thumbnail': r'https://a10.gaanacdn.com/images/albums/25/151425/crop_480x480_151425.jpg',
+            'duration': 301,
+            'album': 'Virsa - Romance',
+            'artist': None,
+            'language': 'Punjabi',
+            'release_date': 'Jan 20, 2014'
+        }
+    }, {
+        'url': 'https://gaana.com/album/simmba',
+        'playlist_count': 7,
+        'info_dict': {
+            'id': '2291554',
+            'title': 'Simmba'
+        }
+    }, {
+        'url': 'https://gaana.com/artist/dhvani-bhanushali',
+        'playlist_min_count': 26,
+        'info_dict': {
+            'id': '1380321',
+            'title': 'Dhvani Bhanushali'
+        }
+    }, {
+        'url': 'https://gaana.com/playlist/gaana-dj-jatt-di-playlist',
+        'playlist_min_count': 19,
+        'info_dict': {
+            'id': '8700656',
+            'title': 'Jatt Di Playlist'
+        }
+    }]
+
+    def _real_extract(self, url):
+        r_match = re.match(self._VALID_URL, url)
+
+        video_id = r_match.group('id')
+        type_id = r_match.group('idtype')
+        self.IE_NAME = 'gaana:' + type_id
+
+        self._set_cookie(self._BASE_URL, 'PHPSESSID', 'val')
+        webpage = self._download_webpage(url, video_id)
+
+        matchobj = re.findall(r'class="parentnode sourcelist_\d+">(.*?)</span>', webpage)
+        matchobj = [self._parse_json(g, video_id) for g in matchobj]
+        entries = []
+        if len(matchobj) > 1:
+            for g in matchobj:
+                entries.append(self._create_entry(g, g['id']))
+            data = re.findall(r'id="parent-row-(?:album|artist|playlist)\d+">(.*?)</span>', webpage)
+            data = self._parse_json(data[0], video_id)
+            return self.playlist_result(entries, data['id'], data['title'])
+        else:
+            return self._create_entry(matchobj[0], matchobj[0]['id'])