mirrors/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-02 06:27:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

add login error handling

Browse source 
Authored by: bashonly
This commit is contained in:
bashonly 2024-10-29 18:37:47 -05:00
parent 61007bb684
commit 4918dfd46b
No known key found for this signature in database
GPG key ID: 783F096F253D15B0
1 changed files with 17 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
yt_dlp/extractor/gamedevtv.py
View file

@ -1,7 +1,9 @@
import json import json
from .common import InfoExtractor from .common import InfoExtractor
from ..networking.exceptions import HTTPError
from ..utils import ( from ..utils import (
ExtractorError,
clean_html, clean_html,
int_or_none, int_or_none,
join_nonempty, join_nonempty,
@ -15,7 +17,6 @@
class GameDevTVDashboardIE(InfoExtractor): class GameDevTVDashboardIE(InfoExtractor):
_VALID_URL = r'https?://(?:www\.)?gamedev\.tv/dashboard/courses/(?P<id>\d+)' _VALID_URL = r'https?://(?:www\.)?gamedev\.tv/dashboard/courses/(?P<id>\d+)'
_NETRC_MACHINE = 'gamedevtv' _NETRC_MACHINE = 'gamedevtv'
_API_HEADERS = {}
_TESTS = [{ _TESTS = [{
'url': 'https://www.gamedev.tv/dashboard/courses/25', 'url': 'https://www.gamedev.tv/dashboard/courses/25',
'info_dict': { 'info_dict': {
@ -31,8 +32,10 @@ class GameDevTVDashboardIE(InfoExtractor):
}, },
'playlist_count': 100, 'playlist_count': 100,
}] }]
_API_HEADERS = {}
def _perform_login(self, username, password): def _perform_login(self, username, password):
try:
response = self._download_json( response = self._download_json(
'https://api.gamedev.tv/api/students/login', None, 'Logging in', 'https://api.gamedev.tv/api/students/login', None, 'Logging in',
headers={'Content-Type': 'application/json'}, headers={'Content-Type': 'application/json'},
@ -41,6 +44,11 @@ def _perform_login(self, username, password):
'password': password, 'password': password,
'cart_items': [], 'cart_items': [],
}).encode()) }).encode())
except ExtractorError as e:
if isinstance(e.cause, HTTPError) and e.cause.status == 401:
raise ExtractorError('Invalid username/password', expected=True)
raise
self._API_HEADERS['Authorization'] = f'{response["token_type"]} {response["access_token"]}' self._API_HEADERS['Authorization'] = f'{response["token_type"]} {response["access_token"]}'
def _real_initialize(self): def _real_initialize(self):