3030 RawPrivateKeyToStaticPublicKeyInput ,
3131)
3232from aws_cryptographic_material_providers .mpl .references import IKeyring
33- from aws_cryptography_primitives .smithygenerated .aws_cryptography_primitives .models import ECDHCurveSpec
3433from aws_dbesdk_dynamodb .encrypted .client import EncryptedClient
3534from aws_dbesdk_dynamodb .structures .dynamodb import (
3635 DynamoDbTableEncryptionConfig ,
4746EXAMPLE_ECC_PUBLIC_KEY_FILENAME_RECIPIENT = "RawEcdhKeyringExamplePublicKeyRecipient.pem"
4847
4948
50- def raw_ecdh_keyring_get_item_put_item (ddb_table_name : str , curve_spec : ECDHCurveSpec ):
49+ def raw_ecdh_keyring_get_item_put_item (ddb_table_name : str , curve_spec : str ):
5150 """
5251 Demonstrate using a raw ECDH keyring with static keys.
5352
@@ -107,7 +106,7 @@ def raw_ecdh_keyring_get_item_put_item(ddb_table_name: str, curve_spec: ECDHCurv
107106 mat_prov = AwsCryptographicMaterialProviders (config = MaterialProvidersConfig ())
108107
109108 keyring_input = CreateRawEcdhKeyringInput (
110- curve_spec = str ( curve_spec ) ,
109+ curve_spec = curve_spec ,
111110 key_agreement_scheme = RawEcdhStaticConfigurationsRawPrivateKeyToStaticPublicKey (
112111 RawPrivateKeyToStaticPublicKeyInput (
113112 # Must be a UTF8 PEM-encoded private key
@@ -123,7 +122,7 @@ def raw_ecdh_keyring_get_item_put_item(ddb_table_name: str, curve_spec: ECDHCurv
123122 put_get_example_with_keyring (raw_ecdh_keyring , ddb_table_name )
124123
125124
126- def ephemeral_raw_ecdh_keyring_put_item (ddb_table_name : str , curve_spec : ECDHCurveSpec ):
125+ def ephemeral_raw_ecdh_keyring_put_item (ddb_table_name : str , curve_spec : str ):
127126 """
128127 Demonstrate using a raw ECDH keyring with ephemeral keys.
129128
@@ -160,8 +159,9 @@ def ephemeral_raw_ecdh_keyring_put_item(ddb_table_name: str, curve_spec: ECDHCur
160159 # encrypt data and CANNOT decrypt messages.
161160 # The DynamoDb encryption client uses this to encrypt items.
162161 mat_prov = AwsCryptographicMaterialProviders (config = MaterialProvidersConfig ())
162+
163163 keyring_input = CreateRawEcdhKeyringInput (
164- curve_spec = str ( curve_spec ) ,
164+ curve_spec = curve_spec ,
165165 key_agreement_scheme = RawEcdhStaticConfigurationsEphemeralPrivateKeyToStaticPublicKey (
166166 EphemeralPrivateKeyToStaticPublicKeyInput (recipient_public_key = public_key_bytes )
167167 ),
@@ -175,7 +175,7 @@ def ephemeral_raw_ecdh_keyring_put_item(ddb_table_name: str, curve_spec: ECDHCur
175175 put_example_with_keyring (raw_ecdh_keyring , ddb_table_name )
176176
177177
178- def discovery_raw_ecdh_keyring_get_item (ddb_table_name : str , curve_spec : ECDHCurveSpec ):
178+ def discovery_raw_ecdh_keyring_get_item (ddb_table_name : str , curve_spec : str ):
179179 """
180180 Demonstrate using a raw ECDH keyring with discovery.
181181
@@ -210,7 +210,7 @@ def discovery_raw_ecdh_keyring_get_item(ddb_table_name: str, curve_spec: ECDHCur
210210 mat_prov = AwsCryptographicMaterialProviders (config = MaterialProvidersConfig ())
211211
212212 keyring_input = CreateRawEcdhKeyringInput (
213- curve_spec = str ( curve_spec ) ,
213+ curve_spec = curve_spec ,
214214 key_agreement_scheme = RawEcdhStaticConfigurationsPublicKeyDiscovery (
215215 PublicKeyDiscoveryInput (recipient_static_private_key = private_key_utf8_encoded )
216216 ),
0 commit comments