z-Ph LddlZddlmcmZGddejZdZdS)Nc$eZdZdZdZdZdZdS)InMemoryKmsClientzRThis is a mock class implementation of KmsClient, built for testing only. c\tj||j|_dS)z%Create an InMemoryKmsClient instance.N)pe KmsClient__init__custom_kms_confmaster_keys_map)selfconfigs `/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyarrow/tests/parquet/encryption.pyrzInMemoryKmsClient.__init__s) d###%5c|j|d}d||g}tj|}|S)z`Not 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 r wrap_keyzInMemoryKmsClient.wrap_key sR /0EFMM hh 0)<== !+.. rc|j|}tj|}|dd}|dd}||dkr|St d||)zGNot 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_keyzInMemoryKmsClient.unwrap_key)s{#23HI$.{;;.ss3+BCC0 #3#:#:7#C#C C C 4)=:: :rN)__name__ __module__ __qualname____doc__rrr#rr rrsK666  : : : : :rrct|d5}|d}|dksJ ddddS#1swxYwYdS)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 r verify_file_encryptedr16s dD  $TIIaLL G#####$$$$$$$$$$$$$$$$$$s=AA)rpyarrow.parquet.encryptionparquet encryptionrrrr1r(rr r5sq" '''''''''::::: :::@$$$$$r