[vine:user] Simplify
This commit is contained in:
parent
7b93c2c204
commit
a172b258ac
1 changed files with 21 additions and 26 deletions
|
@ -60,37 +60,32 @@ class VineIE(InfoExtractor):
|
|||
'formats': formats,
|
||||
}
|
||||
|
||||
|
||||
class VineUserIE(InfoExtractor):
|
||||
IE_NAME = 'vine:user'
|
||||
_VALID_URL = r'(?:https?://)?vine\.co/(?P<user>[^/]+)/?(\?.*)?$'
|
||||
_VINE_BASE_URL = "https://vine.co/"
|
||||
|
||||
def _profile_url(self, user):
|
||||
return "%sapi/users/profiles/vanity/%s"%(self._VINE_BASE_URL, user)
|
||||
|
||||
def _timeline_url(self, user_id, page=1):
|
||||
return "%sapi/timelines/users/%s?page=%s"%(self._VINE_BASE_URL, user_id, page)
|
||||
|
||||
def _profile_data(self, user):
|
||||
return self._download_json(self._profile_url(user), user)
|
||||
|
||||
def _timeline_data(self, user):
|
||||
profile_data = self._profile_data(user)
|
||||
user_id = profile_data['data']['userId']
|
||||
timeline_data = []
|
||||
for pagenum in itertools.count(1):
|
||||
timeline_page = self._download_json(self._timeline_url(user_id, pagenum), user)
|
||||
timeline_data.extend(timeline_page['data']['records'])
|
||||
if timeline_page['data']['nextPage'] is None:
|
||||
break
|
||||
return timeline_data
|
||||
|
||||
def _extract_videos(self, user):
|
||||
timeline_data = self._timeline_data(user)
|
||||
entries = [self.url_result(e['permalinkUrl'], 'Vine') for e in timeline_data]
|
||||
return self.playlist_result(entries, user)
|
||||
|
||||
def _real_extract(self, url):
|
||||
mobj = re.match(self._VALID_URL, url)
|
||||
user = mobj.group('user')
|
||||
return self._extract_videos(user)
|
||||
|
||||
profile_url = "%sapi/users/profiles/vanity/%s" % (
|
||||
self._VINE_BASE_URL, user)
|
||||
profile_data = self._download_json(
|
||||
profile_url, user, note='Downloading user profile data')
|
||||
|
||||
user_id = profile_data['data']['userId']
|
||||
timeline_data = []
|
||||
for pagenum in itertools.count(1):
|
||||
timeline_url = "%sapi/timelines/users/%s?page=%s" % (
|
||||
self._VINE_BASE_URL, user_id, pagenum)
|
||||
timeline_page = self._download_json(
|
||||
timeline_url, user, note='Downloading page %d' % pagenum)
|
||||
timeline_data.extend(timeline_page['data']['records'])
|
||||
if timeline_page['data']['nextPage'] is None:
|
||||
break
|
||||
|
||||
entries = [
|
||||
self.url_result(e['permalinkUrl'], 'Vine') for e in timeline_data]
|
||||
return self.playlist_result(entries, user)
|
||||
|
|
Loading…
Reference in a new issue