h NSSKrSSKJs Jr "SS\R 5rSrg)Nc*\rSrSrSrSrSrSrSrg)InMemoryKmsClientzJThis is a mock class implementation of KmsClient, built for testing only. cd[RRU5 URUlg)z%Create an InMemoryKmsClient instance.N)pe KmsClient__init__custom_kms_confmaster_keys_map)selfconfigs Z/home/james-whalen/.local/lib/python3.13/site-packages/pyarrow/tests/parquet/encryption.pyr InMemoryKmsClient.__init__s" d#%55cURURS5nSRX1/5n[R"U5nU$)zXNot a secure cipher - the wrapped key is just the master key concatenated with key bytesutf-8r)r encodejoinbase64 b64encode)r key_bytesmaster_key_identifiermaster_key_bytes wrapped_keyresults rwrap_keyInMemoryKmsClient.wrap_key sK //0EFMM hh 0<= !!+. rcURUn[R"U5nUSSnUSSnX5RS5:XaU$[ SXV5e)z?Not a secure cipher - just extract the key from the wrapped keyNrzIncorrect master key used)r r b64decodedecode ValueError)r rrexpected_master_keydecoded_wrapped_keyr decrypted_keys r unwrap_keyInMemoryKmsClient.unwrap_key)sj#223HI$..{;.s3+BC0 #:#:7#C C 4): :r)r N) __name__ __module__ __qualname____firstlineno____doc__r rr&__static_attributes__rrrrs6  :rrc[US5nURS5nUS:XdeSSS5 g!,(df  g=f)zVerify that the file is encrypted by looking at its first 4 bytes. If it's the magic string PARE then this is a parquet with encrypted footer.rbsPAREN)openread)pathfile magic_strs rverify_file_encryptedr76s7 dD TIIaL G###   s1 ?)rpyarrow.parquet.encryptionparquet encryptionrrrr7r.rrr;s&"'': :@$r