mirror of
https://github.com/soimort/you-get.git
synced 2025-01-23 13:35:16 +03:00
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
#!/usr/bin/env python
|
|
|
|
import unittest
|
|
import urllib
|
|
|
|
from you_get.extractors import (
|
|
imgur,
|
|
magisto,
|
|
youtube,
|
|
bilibili,
|
|
)
|
|
|
|
|
|
class YouGetTests(unittest.TestCase):
|
|
def test_imgur(self):
|
|
imgur.download('http://imgur.com/WVLk5nD', info_only=True)
|
|
imgur.download('http://imgur.com/gallery/WVLk5nD', info_only=True)
|
|
|
|
def test_magisto(self):
|
|
magisto.download(
|
|
'http://www.magisto.com/album/video/f3x9AAQORAkfDnIFDA',
|
|
info_only=True
|
|
)
|
|
|
|
def test_youtube(self):
|
|
youtube.download(
|
|
'http://www.youtube.com/watch?v=pzKerr0JIPA', info_only=True
|
|
)
|
|
youtube.download('http://youtu.be/pzKerr0JIPA', info_only=True)
|
|
youtube.download(
|
|
'http://www.youtube.com/attribution_link?u=/watch?v%3DldAKIzq7bvs%26feature%3Dshare', # noqa
|
|
info_only=True
|
|
)
|
|
|
|
def test_bilibili(self):
|
|
maxRetry = 3
|
|
|
|
# the bilibi is not stable, it got 502 occasionally
|
|
for i in range(maxRetry):
|
|
try:
|
|
bilibili.download(
|
|
'https://www.bilibili.com/video/av16907446/', info_only=True
|
|
)
|
|
bilibili.download(
|
|
'https://www.bilibili.com/video/av13228063/', info_only=True
|
|
)
|
|
return
|
|
except urllib.error.HTTPError as e:
|
|
if e.error == 502 and i + 1 < maxRetry:
|
|
continue
|
|
else:
|
|
raise e
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|