diff --git a/youtube_dl/extractor/cognito.py b/youtube_dl/extractor/cognito.py index 441e77e57..70cca2c7c 100755 --- a/youtube_dl/extractor/cognito.py +++ b/youtube_dl/extractor/cognito.py @@ -1,12 +1,18 @@ from .common import InfoExtractor - -import boto3 -from warrant import Cognito -from warrant.aws_srp import AWSSRP +from ..utils import ExtractorError class CognitoBaseIE(InfoExtractor): + def _cognito_login(self, auth_data): + + try: + import boto3 + from warrant import Cognito + from warrant.aws_srp import AWSSRP + except ImportError: + raise ExtractorError('%s depends on boto3 and warrant.' % self.IE_NAME) + region = auth_data['PoolId'].split('_')[0] client = boto3.client( 'cognito-idp',