[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,
|
'formats': formats,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
class VineUserIE(InfoExtractor):
|
class VineUserIE(InfoExtractor):
|
||||||
IE_NAME = 'vine:user'
|
IE_NAME = 'vine:user'
|
||||||
_VALID_URL = r'(?:https?://)?vine\.co/(?P<user>[^/]+)/?(\?.*)?$'
|
_VALID_URL = r'(?:https?://)?vine\.co/(?P<user>[^/]+)/?(\?.*)?$'
|
||||||
_VINE_BASE_URL = "https://vine.co/"
|
_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):
|
def _real_extract(self, url):
|
||||||
mobj = re.match(self._VALID_URL, url)
|
mobj = re.match(self._VALID_URL, url)
|
||||||
user = mobj.group('user')
|
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