mirror of
https://codeberg.org/polarisfm/youtube-dl
synced 2024-11-22 16:44:32 +01:00
added VimeoSearchIE
This commit is contained in:
parent
a115e07594
commit
c2e6381597
@ -1334,6 +1334,7 @@ from .vimeo import (
|
|||||||
VimeoUserIE,
|
VimeoUserIE,
|
||||||
VimeoWatchLaterIE,
|
VimeoWatchLaterIE,
|
||||||
VHXEmbedIE,
|
VHXEmbedIE,
|
||||||
|
VimeoSearchIE,
|
||||||
)
|
)
|
||||||
from .vimple import VimpleIE
|
from .vimple import VimpleIE
|
||||||
from .vine import (
|
from .vine import (
|
||||||
|
@ -7,7 +7,7 @@ import json
|
|||||||
import re
|
import re
|
||||||
import itertools
|
import itertools
|
||||||
|
|
||||||
from .common import InfoExtractor
|
from .common import InfoExtractor, SearchInfoExtractor
|
||||||
from ..compat import (
|
from ..compat import (
|
||||||
compat_kwargs,
|
compat_kwargs,
|
||||||
compat_HTTPError,
|
compat_HTTPError,
|
||||||
@ -1126,3 +1126,47 @@ class VHXEmbedIE(VimeoBaseInfoExtractor):
|
|||||||
info = self._parse_config(config, video_id)
|
info = self._parse_config(config, video_id)
|
||||||
self._vimeo_sort_formats(info['formats'])
|
self._vimeo_sort_formats(info['formats'])
|
||||||
return info
|
return info
|
||||||
|
|
||||||
|
|
||||||
|
class VimeoSearchIE(SearchInfoExtractor):
|
||||||
|
IE_DESC = 'vimeo.com search'
|
||||||
|
_MAX_RESULTS = float('inf')
|
||||||
|
IE_NAME = 'vimeo:search'
|
||||||
|
_SEARCH_KEY = 'vimeosearch'
|
||||||
|
_TESTS = []
|
||||||
|
|
||||||
|
def _get_n_results(self, query, n):
|
||||||
|
"""Get a specified number of results for a query"""
|
||||||
|
|
||||||
|
video_ids = []
|
||||||
|
|
||||||
|
for pagenum in itertools.count(1):
|
||||||
|
if len(video_ids) >= n:
|
||||||
|
break
|
||||||
|
|
||||||
|
webpage = self._download_webpage(
|
||||||
|
'https://vimeo.com/search/page:%i?' % pagenum,
|
||||||
|
'vimeosearch:' + query,
|
||||||
|
query={'q': query})
|
||||||
|
|
||||||
|
for match in re.finditer(r'vimeo.com\\\/([0-9]{9})', webpage):
|
||||||
|
if len(video_ids) >= n:
|
||||||
|
break
|
||||||
|
|
||||||
|
video_id = match.group(1)
|
||||||
|
|
||||||
|
# video ids may appear multiple time so only add new ones
|
||||||
|
if video_id not in video_ids:
|
||||||
|
video_ids.append(video_id)
|
||||||
|
|
||||||
|
entries = []
|
||||||
|
|
||||||
|
for video_id in video_ids:
|
||||||
|
entries.append({'_type': 'url', 'url': 'https://vimeo.com/' + video_id})
|
||||||
|
|
||||||
|
return {
|
||||||
|
'_type': 'playlist',
|
||||||
|
'id': query,
|
||||||
|
'title': query,
|
||||||
|
'entries': entries
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user