25 lines
737 B
Python
25 lines
737 B
Python
|
from .common import InfoExtractor
|
||
|
|
|
||
|
|
import boto3
|
||
|
|
from warrant import Cognito
|
||
|
|
from warrant.aws_srp import AWSSRP
|
||
|
|
|
||
|
|
class CognitoBaseIE(InfoExtractor):
|
||
|
|
|
||
|
|
def _cognito_login(self, auth_data):
|
||
|
|
region = auth_data['PoolId'].split('_')[0]
|
||
|
|
client = boto3.client(
|
||
|
|
'cognito-idp',
|
||
|
|
region_name = region,
|
||
|
|
aws_access_key_id = 'SomeNonsenseValue',
|
||
|
|
aws_secret_access_key = 'YetAnotherNonsenseValue'
|
||
|
|
)
|
||
|
|
aws = AWSSRP(
|
||
|
|
username = auth_data['Username'],
|
||
|
|
password = auth_data['Password'],
|
||
|
|
pool_id = auth_data['PoolId'],
|
||
|
|
client_id = auth_data['ClientId'],
|
||
|
|
client=client
|
||
|
|
)
|
||
|
|
return aws.authenticate_user()
|