From 4918dfd46b738b55b969e8b2b320900b4cd7b846 Mon Sep 17 00:00:00 2001 From: bashonly Date: Tue, 29 Oct 2024 18:37:47 -0500 Subject: [PATCH] add login error handling Authored by: bashonly --- yt_dlp/extractor/gamedevtv.py | 26 +++++++++++++++++--------- 1 file changed, 17 insertions(+), 9 deletions(-) diff --git a/yt_dlp/extractor/gamedevtv.py b/yt_dlp/extractor/gamedevtv.py index a6d744e3e4..1fe1b96978 100644 --- a/yt_dlp/extractor/gamedevtv.py +++ b/yt_dlp/extractor/gamedevtv.py @@ -1,7 +1,9 @@ import json from .common import InfoExtractor +from ..networking.exceptions import HTTPError from ..utils import ( + ExtractorError, clean_html, int_or_none, join_nonempty, @@ -15,7 +17,6 @@ class GameDevTVDashboardIE(InfoExtractor): _VALID_URL = r'https?://(?:www\.)?gamedev\.tv/dashboard/courses/(?P\d+)' _NETRC_MACHINE = 'gamedevtv' - _API_HEADERS = {} _TESTS = [{ 'url': 'https://www.gamedev.tv/dashboard/courses/25', 'info_dict': { @@ -31,16 +32,23 @@ class GameDevTVDashboardIE(InfoExtractor): }, 'playlist_count': 100, }] + _API_HEADERS = {} def _perform_login(self, username, password): - response = self._download_json( - 'https://api.gamedev.tv/api/students/login', None, 'Logging in', - headers={'Content-Type': 'application/json'}, - data=json.dumps({ - 'email': username, - 'password': password, - 'cart_items': [], - }).encode()) + try: + response = self._download_json( + 'https://api.gamedev.tv/api/students/login', None, 'Logging in', + headers={'Content-Type': 'application/json'}, + data=json.dumps({ + 'email': username, + 'password': password, + 'cart_items': [], + }).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"]}' def _real_initialize(self):