+\hdZddlmZddlmZmZmZmZddlZddl m Z ddl m Z m Z erddlmZddlmZdd lmZd Zd d Zy) z#MONGODB-AWS Authentication helpers.) annotations) TYPE_CHECKINGAnyMappingTypeNBinary)ConfigurationErrorOperationFailure)_ReadableBuffer)AsyncConnection)MongoCredentialFcK ddl}ddlm}m}|d|j dkr tdGdd |j} ||j|j|j|jj}|jd}d d |d } |jd | d{} | } tdD]@} |j| d}d | d|d} |jd | d{} | ds@yy#t$r}td|d}~wwxYw7r7-#|j $r)}|dt#|d|j$ddd}~wt&$r |dwxYww)zAuthenticate using MONGODB-AWS.rNzhMONGODB-AWS authentication requires pymongo-auth-aws: install with: python -m pip install 'pymongo[aws]')set_cached_credentialsset_use_cached_credentialsT z@MONGODB-AWS authentication requires MongoDB version 4.4 or laterc$eZdZddZddZddZy))_authenticate_aws..AwsSaslContextctS)z#Return the bson.binary.Binary type.r)selfs `/root/niggaflix-v3/playground/venv/lib/python3.12/site-packages/pymongo/asynchronous/auth_aws.py binary_typez5_authenticate_aws..AwsSaslContext.binary_type6sMc,tj|S)zEncode a dictionary to BSON.)bsonencode)rdocs r bson_encodez5_authenticate_aws..AwsSaslContext.bson_encode:s;;s# #rc,tj|S)zDecode BSON to a dictionary.)rdecode)rdatas r bson_decodez5_authenticate_aws..AwsSaslContext.bson_decode>s;;t$ $rN)returnz Type[Binary])rMapping[str, Any]r#bytes)r!r r#r$)__name__ __module__ __qualname__rrr"rrAwsSaslContextr4s  $ %rr*z MONGODB-AWS) saslStart mechanismpayloadz $external r.conversationId) saslContinuer0r.donez (pymongo-auth-aws version ))pymongo_auth_aws ImportErrorr pymongo_auth_aws.authrrmax_wire_versionr* AwsCredentialusernamepasswordmechanism_propertiesaws_session_tokenstepcommandrangePyMongoAuthAwsErrorr __version__ Exception) credentialsconnr4errr*ctxclient_payload client_first server_firstres_cmdexcs r_authenticate_awsrN s t$ q  !cdd %)88 %"  * *$$$$00BB  $%&]~^ !\\+|DD r A XXc)n5N !"./?"@)C  [#66C6{ W   A  PE7  / /t$e./?/K/K.LA N   t$ sE8D ?E8A-D,5D(6AD,<D*= D,D,E8 D, E8 D% D  D%%E8(D,*D,,E5;$EE55E8)rCrrDr r#None)__doc__ __future__rtypingrrrrr bson.binaryr pymongo.errorsr r bson.typingsr pymongo.asynchronous.poolr pymongo.auth_sharedr_IS_SYNCrNr)rrrYs3*"44 ?,93 Dr