30
30
RawPrivateKeyToStaticPublicKeyInput ,
31
31
)
32
32
from aws_cryptographic_material_providers .mpl .references import IKeyring
33
- from aws_cryptography_primitives .smithygenerated .aws_cryptography_primitives .models import ECDHCurveSpec
34
33
from aws_dbesdk_dynamodb .encrypted .client import EncryptedClient
35
34
from aws_dbesdk_dynamodb .structures .dynamodb import (
36
35
DynamoDbTableEncryptionConfig ,
47
46
EXAMPLE_ECC_PUBLIC_KEY_FILENAME_RECIPIENT = "RawEcdhKeyringExamplePublicKeyRecipient.pem"
48
47
49
48
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 ):
51
50
"""
52
51
Demonstrate using a raw ECDH keyring with static keys.
53
52
@@ -107,7 +106,7 @@ def raw_ecdh_keyring_get_item_put_item(ddb_table_name: str, curve_spec: ECDHCurv
107
106
mat_prov = AwsCryptographicMaterialProviders (config = MaterialProvidersConfig ())
108
107
109
108
keyring_input = CreateRawEcdhKeyringInput (
110
- curve_spec = str ( curve_spec ) ,
109
+ curve_spec = curve_spec ,
111
110
key_agreement_scheme = RawEcdhStaticConfigurationsRawPrivateKeyToStaticPublicKey (
112
111
RawPrivateKeyToStaticPublicKeyInput (
113
112
# 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
123
122
put_get_example_with_keyring (raw_ecdh_keyring , ddb_table_name )
124
123
125
124
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 ):
127
126
"""
128
127
Demonstrate using a raw ECDH keyring with ephemeral keys.
129
128
@@ -160,8 +159,9 @@ def ephemeral_raw_ecdh_keyring_put_item(ddb_table_name: str, curve_spec: ECDHCur
160
159
# encrypt data and CANNOT decrypt messages.
161
160
# The DynamoDb encryption client uses this to encrypt items.
162
161
mat_prov = AwsCryptographicMaterialProviders (config = MaterialProvidersConfig ())
162
+
163
163
keyring_input = CreateRawEcdhKeyringInput (
164
- curve_spec = str ( curve_spec ) ,
164
+ curve_spec = curve_spec ,
165
165
key_agreement_scheme = RawEcdhStaticConfigurationsEphemeralPrivateKeyToStaticPublicKey (
166
166
EphemeralPrivateKeyToStaticPublicKeyInput (recipient_public_key = public_key_bytes )
167
167
),
@@ -175,7 +175,7 @@ def ephemeral_raw_ecdh_keyring_put_item(ddb_table_name: str, curve_spec: ECDHCur
175
175
put_example_with_keyring (raw_ecdh_keyring , ddb_table_name )
176
176
177
177
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 ):
179
179
"""
180
180
Demonstrate using a raw ECDH keyring with discovery.
181
181
@@ -210,7 +210,7 @@ def discovery_raw_ecdh_keyring_get_item(ddb_table_name: str, curve_spec: ECDHCur
210
210
mat_prov = AwsCryptographicMaterialProviders (config = MaterialProvidersConfig ())
211
211
212
212
keyring_input = CreateRawEcdhKeyringInput (
213
- curve_spec = str ( curve_spec ) ,
213
+ curve_spec = curve_spec ,
214
214
key_agreement_scheme = RawEcdhStaticConfigurationsPublicKeyDiscovery (
215
215
PublicKeyDiscoveryInput (recipient_static_private_key = private_key_utf8_encoded )
216
216
),
0 commit comments