[addanime] try to download HQ before normal
[youtube-dl] / youtube_dl / extractor / addanime.py
index 46db8262fef261cc1d73d831c359111060cca8d8..adbda194a10b32b69e408f4a1b56b5db68a6b883 100644 (file)
@@ -1,4 +1,3 @@
-import ast
 import re
 
 from .common import InfoExtractor
@@ -18,8 +17,8 @@ class AddAnimeIE(InfoExtractor):
     IE_NAME = u'AddAnime'
     _TEST = {
         u'url': u'http://www.add-anime.net/watch_video.php?v=24MR3YO5SAS9',
-        u'file': u'24MR3YO5SAS9.flv',
-        u'md5': u'1036a0e0cd307b95bd8a8c3a5c8cfaf1',
+        u'file': u'24MR3YO5SAS9.mp4',
+        u'md5': u'3f8e232ad52163c87fa23897e736cb2c',
         u'info_dict': {
             u"description": u"One Piece 606",
             u"title": u"One Piece 606"
@@ -61,8 +60,12 @@ class AddAnimeIE(InfoExtractor):
                 note=u'Confirming after redirect')
             webpage = self._download_webpage(url, video_id)
 
-        video_url = self._search_regex(r"var normal_video_file = '(.*?)';",
+        video_url = self._search_regex(r"var hq_video_file = '(.*?)';",
                                        webpage, u'video file URL')
+        if not video_url:  # if there's no hq_video_file, get normal_video_file
+            video_url = self._search_regex(r"var normal_video_file = '(.*?)';",
+                                           webpage, u'video file URL')
+        video_extension = video_url[-3:]  # mp4 or flv ?
         video_title = self._og_search_title(webpage)
         video_description = self._og_search_description(webpage)
 
@@ -70,7 +73,7 @@ class AddAnimeIE(InfoExtractor):
             '_type': 'video',
             'id':  video_id,
             'url': video_url,
-            'ext': 'flv',
+            'ext': video_extension,
             'title': video_title,
             'description': video_description
         }