remove parameter max_results from all capabilities
This commit is contained in:
parent
c108ca3fc5
commit
b99d599aa9
30 changed files with 84 additions and 80 deletions
|
|
@ -43,7 +43,7 @@ class ArretSurImagesBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
def create_default_browser(self):
|
def create_default_browser(self):
|
||||||
return self.create_browser(self.config['login'].get(), self.config['password'].get())
|
return self.create_browser(self.config['login'].get(), self.config['password'].get())
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.search_videos(pattern)
|
return self.browser.search_videos(pattern)
|
||||||
# raise UserError('Search does not work on ASI website, use ls latest command')
|
# raise UserError('Search does not work on ASI website, use ls latest command')
|
||||||
|
|
|
||||||
|
|
@ -63,7 +63,7 @@ class ArteBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
else:
|
else:
|
||||||
return self.browser.get_video(_id)
|
return self.browser.get_video(_id)
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.search_videos(pattern)
|
return self.browser.search_videos(pattern)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -48,7 +48,7 @@ class CanalplusBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
def create_default_browser(self):
|
def create_default_browser(self):
|
||||||
return self.create_browser(quality=self.config['quality'].get())
|
return self.create_browser(quality=self.config['quality'].get())
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.search_videos(pattern)
|
return self.browser.search_videos(pattern)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -32,7 +32,7 @@ class CappedBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.get_video(_id)
|
return self.browser.get_video(_id)
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=None, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=None):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.search_videos(pattern)
|
return self.browser.search_videos(pattern)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -46,7 +46,7 @@ class DailymotionBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
|
|
||||||
SORTBY = ['relevance', 'rated', 'visited', None]
|
SORTBY = ['relevance', 'rated', 'visited', None]
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.search_videos(pattern, self.SORTBY[sortby])
|
return self.browser.search_videos(pattern, self.SORTBY[sortby])
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -54,7 +54,7 @@ class EHentaiBackend(BaseBackend, ICapGallery, ICapCollection):
|
||||||
password = None
|
password = None
|
||||||
return self.create_browser(self.config['domain'].get(), username, password)
|
return self.create_browser(self.config['domain'].get(), username, password)
|
||||||
|
|
||||||
def search_gallery(self, pattern, sortby=None, max_results=None):
|
def search_gallery(self, pattern, sortby=None):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.search_gallery(pattern)
|
return self.browser.search_gallery(pattern)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -47,7 +47,7 @@ class EuroparlBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
|
|
||||||
SORTBY = ['relevance', 'rating', 'views', 'time']
|
SORTBY = ['relevance', 'rating', 'views', 'time']
|
||||||
|
|
||||||
# def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
# def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
# with self.browser:
|
# with self.browser:
|
||||||
# return self.browser.search_videos(pattern, self.SORTBY[sortby])
|
# return self.browser.search_videos(pattern, self.SORTBY[sortby])
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -44,7 +44,7 @@ class PluzzBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.get_video(_id)
|
return self.browser.get_video(_id)
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.search_videos(pattern)
|
return self.browser.search_videos(pattern)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -66,7 +66,7 @@ class GDCVaultBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
|
|
||||||
SORTBY = ['relevance', 'rating', 'views', 'time']
|
SORTBY = ['relevance', 'rating', 'views', 'time']
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.search_videos(pattern, self.SORTBY[sortby])
|
return self.browser.search_videos(pattern, self.SORTBY[sortby])
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -42,10 +42,9 @@ class GroovesharkBackend(BaseBackend, ICapVideo):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
video.thumbnail.data = self.browser.readurl(video.thumbnail.url)
|
video.thumbnail.data = self.browser.readurl(video.thumbnail.url)
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=10):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
for video in self.browser.search_videos(pattern, max_results):
|
return self.browser.search_videos(pattern)
|
||||||
yield video
|
|
||||||
|
|
||||||
def get_video(self, _id):
|
def get_video(self, _id):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
|
|
|
||||||
|
|
@ -64,7 +64,7 @@ class GroovesharkBrowser(BaseBrowser):
|
||||||
def home(self):
|
def home(self):
|
||||||
self.get_communication_token()
|
self.get_communication_token()
|
||||||
|
|
||||||
def search_videos(self, pattern, max_results):
|
def search_videos(self, pattern):
|
||||||
method = 'getResultsFromSearch'
|
method = 'getResultsFromSearch'
|
||||||
|
|
||||||
parameters = {}
|
parameters = {}
|
||||||
|
|
@ -75,16 +75,14 @@ class GroovesharkBrowser(BaseBrowser):
|
||||||
|
|
||||||
response = self.API_post(method, parameters, self.create_token(method))
|
response = self.API_post(method, parameters, self.create_token(method))
|
||||||
|
|
||||||
songs = self.create_video_from_songs_result(response['result']['result']['Songs'], max_results)
|
songs = self.create_video_from_songs_result(response['result']['result']['Songs'])
|
||||||
#playlists = self.create_video_from_playlist_result(response['result']['result']['Playlists'])
|
#playlists = self.create_video_from_playlist_result(response['result']['result']['Playlists'])
|
||||||
#albums = self.create_video_from_albums_result(response['result']['result']['Albums'])
|
#albums = self.create_video_from_albums_result(response['result']['result']['Albums'])
|
||||||
|
|
||||||
return songs
|
return songs
|
||||||
|
|
||||||
def create_video_from_songs_result(self, songs, max_results):
|
def create_video_from_songs_result(self, songs):
|
||||||
self.VIDEOS_FROM_SONG_RESULTS = []
|
self.VIDEOS_FROM_SONG_RESULTS = []
|
||||||
if max_results:
|
|
||||||
songs = songs[0:max_results]
|
|
||||||
|
|
||||||
for song in songs:
|
for song in songs:
|
||||||
video = BaseVideo(song['SongID'])
|
video = BaseVideo(song['SongID'])
|
||||||
|
|
@ -99,7 +97,8 @@ class GroovesharkBrowser(BaseBrowser):
|
||||||
except ValueError:
|
except ValueError:
|
||||||
video.date = NotAvailable
|
video.date = NotAvailable
|
||||||
self.VIDEOS_FROM_SONG_RESULTS.append(video)
|
self.VIDEOS_FROM_SONG_RESULTS.append(video)
|
||||||
return self.VIDEOS_FROM_SONG_RESULTS
|
|
||||||
|
yield video
|
||||||
|
|
||||||
def create_video_from_playlist_result(self, playlists):
|
def create_video_from_playlist_result(self, playlists):
|
||||||
videos = []
|
videos = []
|
||||||
|
|
|
||||||
|
|
@ -42,7 +42,7 @@ class InaBackend(BaseBackend, ICapVideo):
|
||||||
def get_video(self, _id):
|
def get_video(self, _id):
|
||||||
return self.browser.get_video(_id)
|
return self.browser.get_video(_id)
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.search_videos(pattern)
|
return self.browser.search_videos(pattern)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -65,9 +65,8 @@ class MediawikiBackend(BaseBackend, ICapContent):
|
||||||
content.content = data
|
content.content = data
|
||||||
return content
|
return content
|
||||||
|
|
||||||
def iter_revisions(self, _id, max_results=10):
|
def iter_revisions(self, _id):
|
||||||
for rev in self.browser.iter_wiki_revisions(_id, max_results):
|
return self.browser.iter_wiki_revisions(_id)
|
||||||
yield rev
|
|
||||||
|
|
||||||
def push_content(self, content, message=None, minor=False):
|
def push_content(self, content, message=None, minor=False):
|
||||||
self.browser.set_wiki_source(content, message, minor)
|
self.browser.set_wiki_source(content, message, minor)
|
||||||
|
|
|
||||||
|
|
@ -148,30 +148,43 @@ class MediawikiBrowser(BaseBrowser):
|
||||||
data['lgtoken'] = result['login']['token']
|
data['lgtoken'] = result['login']['token']
|
||||||
self.API_post(data)
|
self.API_post(data)
|
||||||
|
|
||||||
def iter_wiki_revisions(self, page, nb_entries):
|
def iter_wiki_revisions(self, page):
|
||||||
"""
|
"""
|
||||||
Yield 'Revision' objects for the last <nb_entries> revisions of the specified page.
|
Yield 'Revision' objects.
|
||||||
"""
|
"""
|
||||||
if len(self.username) > 0 and not self.is_logged():
|
if len(self.username) > 0 and not self.is_logged():
|
||||||
self.login()
|
self.login()
|
||||||
data = {'action': 'query',
|
|
||||||
'titles': page,
|
|
||||||
'prop': 'revisions',
|
|
||||||
'rvprop': 'ids|timestamp|comment|user|flags',
|
|
||||||
'rvlimit': str(nb_entries),
|
|
||||||
}
|
|
||||||
|
|
||||||
result = self.API_get(data)
|
MAX_RESULTS = 50
|
||||||
pageid = str(result['query']['pages'].keys()[0])
|
results = MAX_RESULTS
|
||||||
|
last_id = None
|
||||||
|
|
||||||
if pageid != "-1":
|
while results == MAX_RESULTS:
|
||||||
for rev in result['query']['pages'][pageid]['revisions']:
|
data = {'action': 'query',
|
||||||
rev_content = Revision(str(rev['revid']))
|
'titles': page,
|
||||||
rev_content.comment = rev['comment']
|
'prop': 'revisions',
|
||||||
rev_content.author = rev['user']
|
'rvprop': 'ids|timestamp|comment|user|flags',
|
||||||
rev_content.timestamp = datetime.datetime.strptime(rev['timestamp'], '%Y-%m-%dT%H:%M:%SZ')
|
'rvlimit': str(MAX_RESULTS),
|
||||||
rev_content.minor = 'minor' in rev
|
}
|
||||||
yield rev_content
|
|
||||||
|
if last_id is not None:
|
||||||
|
data['rvstartid'] = last_id
|
||||||
|
|
||||||
|
result = self.API_get(data)
|
||||||
|
pageid = str(result['query']['pages'].keys()[0])
|
||||||
|
|
||||||
|
results = 0
|
||||||
|
if pageid != "-1":
|
||||||
|
for rev in result['query']['pages'][pageid]['revisions']:
|
||||||
|
rev_content = Revision(str(rev['revid']))
|
||||||
|
rev_content.comment = rev['comment']
|
||||||
|
rev_content.author = rev['user']
|
||||||
|
rev_content.timestamp = datetime.datetime.strptime(rev['timestamp'], '%Y-%m-%dT%H:%M:%SZ')
|
||||||
|
rev_content.minor = 'minor' in rev
|
||||||
|
yield rev_content
|
||||||
|
|
||||||
|
last_id = rev_content.id
|
||||||
|
results += 1
|
||||||
|
|
||||||
def home(self):
|
def home(self):
|
||||||
# We don't need to change location, we're using the JSON API here.
|
# We don't need to change location, we're using the JSON API here.
|
||||||
|
|
|
||||||
|
|
@ -56,7 +56,7 @@ class NolifeTVBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
video = self.browser.get_video(_id)
|
video = self.browser.get_video(_id)
|
||||||
return video
|
return video
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.search_videos(pattern)
|
return self.browser.search_videos(pattern)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -44,7 +44,7 @@ class TricTracTVBackend(BaseBackend, ICapVideo):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.get_video(_id)
|
return self.browser.get_video(_id)
|
||||||
|
|
||||||
def search_videos(self, pattern=None, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern=None, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
with self.browser:
|
with self.browser:
|
||||||
return self.browser.search_videos(pattern)
|
return self.browser.search_videos(pattern)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -47,7 +47,7 @@ class VimeoBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
|
|
||||||
SORTBY = ['relevance', 'rating', 'views', 'time']
|
SORTBY = ['relevance', 'rating', 'views', 'time']
|
||||||
|
|
||||||
# def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
# def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
# with self.browser:
|
# with self.browser:
|
||||||
# return self.browser.search_videos(pattern, self.SORTBY[sortby])
|
# return self.browser.search_videos(pattern, self.SORTBY[sortby])
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -45,7 +45,7 @@ class YoujizzBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
video = self.browser.get_video(_id)
|
video = self.browser.get_video(_id)
|
||||||
return video
|
return video
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
if not nsfw:
|
if not nsfw:
|
||||||
return set()
|
return set()
|
||||||
with self.browser:
|
with self.browser:
|
||||||
|
|
|
||||||
|
|
@ -46,7 +46,7 @@ class YoupornBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
|
|
||||||
SORTBY = ['relevance', 'rating', 'views', 'time']
|
SORTBY = ['relevance', 'rating', 'views', 'time']
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
if not nsfw:
|
if not nsfw:
|
||||||
return set()
|
return set()
|
||||||
with self.browser:
|
with self.browser:
|
||||||
|
|
|
||||||
|
|
@ -116,7 +116,7 @@ class YoutubeBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
video.set_empty_fields(NotAvailable)
|
video.set_empty_fields(NotAvailable)
|
||||||
return video
|
return video
|
||||||
|
|
||||||
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False, max_results=None):
|
def search_videos(self, pattern, sortby=ICapVideo.SEARCH_RELEVANCE, nsfw=False):
|
||||||
YOUTUBE_MAX_RESULTS = 50
|
YOUTUBE_MAX_RESULTS = 50
|
||||||
YOUTUBE_MAX_START_INDEX = 1000
|
YOUTUBE_MAX_START_INDEX = 1000
|
||||||
yt_service = gdata.youtube.service.YouTubeService()
|
yt_service = gdata.youtube.service.YouTubeService()
|
||||||
|
|
@ -133,27 +133,22 @@ class YoutubeBackend(BaseBackend, ICapVideo, ICapCollection):
|
||||||
query.orderby = ('relevance', 'rating', 'viewCount', 'published')[sortby]
|
query.orderby = ('relevance', 'rating', 'viewCount', 'published')[sortby]
|
||||||
query.racy = 'include' if nsfw else 'exclude'
|
query.racy = 'include' if nsfw else 'exclude'
|
||||||
|
|
||||||
if max_results is None or max_results > YOUTUBE_MAX_RESULTS:
|
query.max_results = YOUTUBE_MAX_RESULTS
|
||||||
query_max_results = YOUTUBE_MAX_RESULTS
|
|
||||||
else:
|
|
||||||
query_max_results = max_results
|
|
||||||
query.max_results = query_max_results
|
|
||||||
|
|
||||||
if start_index > YOUTUBE_MAX_START_INDEX:
|
if start_index > YOUTUBE_MAX_START_INDEX:
|
||||||
return
|
return
|
||||||
query.start_index = start_index
|
query.start_index = start_index
|
||||||
start_index += query_max_results
|
start_index += YOUTUBE_MAX_RESULTS
|
||||||
|
|
||||||
feed = yt_service.YouTubeQuery(query)
|
feed = yt_service.YouTubeQuery(query)
|
||||||
for entry in feed.entry:
|
for entry in feed.entry:
|
||||||
yield self._entry2video(entry)
|
yield self._entry2video(entry)
|
||||||
nb_yielded += 1
|
nb_yielded += 1
|
||||||
if nb_yielded == max_results:
|
|
||||||
return
|
|
||||||
|
|
||||||
if nb_yielded < query_max_results:
|
if nb_yielded < YOUTUBE_MAX_RESULTS:
|
||||||
return
|
return
|
||||||
|
|
||||||
|
|
||||||
def latest_videos(self):
|
def latest_videos(self):
|
||||||
return self.search_videos(None, ICapVideo.SEARCH_DATE)
|
return self.search_videos(None, ICapVideo.SEARCH_DATE)
|
||||||
|
|
||||||
|
|
|
||||||
Some files were not shown because too many files have changed in this diff Show more
Loading…
Reference in a new issue