Madeorsk/youtube-dl
1
0
Fork 0
mirror of https://codeberg.org/polarisfm/youtube-dl synced 2024-12-28 00:57:54 +01:00
Code Issues Releases Wiki Activity

Set up Base class for shared methods.

Browse Source 
Reorganised the script adding a PornTrexBaseIE class that will be used
for shared methods. Currently hosting only two methods:
1. _login
2. _real_initialize
This commit is contained in:
nindogo 2019-04-21 23:06:54 +03:00
parent 05b65ea96d
commit 6d0f924cbf
1 changed files with 17 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
youtube_dl/extractor/porntrex.py
View File

@ -10,19 +10,7 @@ from ..utils import (
) )
class PornTrexIE(InfoExtractor): class PornTrexBaseIE(InfoExtractor):
_NETRC_MACHINE = 'porntrex'
_VALID_URL = r'https?://(?:www\.)?porntrex\.com/video/(?P<id>[0-9]+)/'
_TEST = {
'url': 'https://www.porntrex.com/video/350451/naomi-woods-the-list',
'info_dict': {
'id': '350451',
'ext': 'mp4',
'title': 'Naomi Woods - The List in 4k',
'uploader': 'delman',
'description': 'Naomi Woods The List',
}
}
def _login(self): def _login(self):
username, password = self._get_login_info() username, password = self._get_login_info()
@ -53,6 +41,21 @@ class PornTrexIE(InfoExtractor):
def _real_initialize(self): def _real_initialize(self):
self._login() self._login()
class PornTrexIE(PornTrexBaseIE):
_NETRC_MACHINE = 'porntrex'
_VALID_URL = r'https?://(?:www\.)?porntrex\.com/video/(?P<id>[0-9]+)/'
_TEST = {
'url': 'https://www.porntrex.com/video/350451/naomi-woods-the-list',
'info_dict': {
'id': '350451',
'ext': 'mp4',
'title': 'Naomi Woods - The List in 4k',
'uploader': 'delman',
'description': 'Naomi Woods The List',
}
}
def _real_extract(self, url): def _real_extract(self, url):
video_id = self._match_id(url) video_id = self._match_id(url)
webpage = self._download_webpage(url, video_id) webpage = self._download_webpage(url, video_id)
@ -94,7 +97,7 @@ class PornTrexIE(InfoExtractor):
} }
class PornTrexPlayListIE(InfoExtractor): class PornTrexPlayListIE(PornTrexBaseIE):
_NETRC_MACHINE = 'porntrex' _NETRC_MACHINE = 'porntrex'
_VALID_URL = \ _VALID_URL = \
r'https?://(?:www\.)?porntrex\.com/playlists/(?P<id>[0-9]+)/' r'https?://(?:www\.)?porntrex\.com/playlists/(?P<id>[0-9]+)/'
@ -111,35 +114,6 @@ class PornTrexPlayListIE(InfoExtractor):
} }
} }
def _login(self):
username, password = self._get_login_info()
if username is None:
return
login_page = self._download_webpage(
'https://www.porntrex.com/login/', None, 'Downloading login page')
login_form = self._hidden_inputs(login_page)
login_form.update({
'username': username.encode('utf-8'),
'pass': password.encode('utf-8'),
'remember_me': str(1).encode('utf-8'),
})
login_page = self._download_webpage(
'https://www.porntrex.com/ajax-login/', None,
note='Logging in',
data=urlencode_postdata(login_form))
if re.search(r'generic-error hidden', login_page):
raise ExtractorError(
'Unable to login, incorrect username and/or password',
expected=True)
def _real_initialize(self):
self._login()
def _real_extract(self, url): def _real_extract(self, url):
playlist_id = self._match_id(url) playlist_id = self._match_id(url)
webpage = self._download_webpage(url, playlist_id) webpage = self._download_webpage(url, playlist_id)