Skip to content

Commit 44c45da

Browse files
committed
Revert "type-hint"
This reverts commit 9025e76 This reverts commit 5ca6fba
1 parent 5ca6fba commit 44c45da

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

Examples/runtimes/python/DynamoDBEncryption/src/keyring/raw_ecdh_keyring_example.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,6 @@
3030
RawPrivateKeyToStaticPublicKeyInput,
3131
)
3232
from aws_cryptographic_material_providers.mpl.references import IKeyring
33-
from aws_cryptography_primitives.smithygenerated.aws_cryptography_primitives.models import ECDHCurveSpec
3433
from aws_dbesdk_dynamodb.encrypted.client import EncryptedClient
3534
from aws_dbesdk_dynamodb.structures.dynamodb import (
3635
DynamoDbTableEncryptionConfig,
@@ -47,7 +46,7 @@
4746
EXAMPLE_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

Comments
 (0)