Fix - Set-Parameter Command fails with 'NoneType' (#5)

lucasvieirasilva · Sep 14, 2020 · fb447fa · fb447fa
1 parent 54e002a
commit fb447fa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/aws_secrets/cli/set_parameter.py b/aws_secrets/cli/set_parameter.py
@@ -3,13 +3,13 @@
 import boto3
 import base64
 from botocore.exceptions import ClientError
-from aws_secrets.miscellaneous import kms
+from aws_secrets.miscellaneous.kms import encrypt
 from aws_secrets.miscellaneous import session
 
 
 @click.command(name='set-parameter')
 @click.option('-e', '--env-file', type=click.Path(), required=True)
-@click.option('-n', '--name', required=True)
+@click.option('-n', '--name', prompt=True, required=True)
 @click.option('-t', '--type',
               required=True, type=click.Choice(['String', 'SecureString'], case_sensitive=True),
               default='SecureString')
@@ -51,9 +51,8 @@ def set_parameter(env_file, name, type, kms, profile, region):
 
     if parameter['type'] == 'SecureString':
         kms_arn = str(yaml_data['kms']['arn'])
-
         print('Encrypting the value')
-        encrypted_value = kms.encrypt(session.session(), value, kms_arn)
+        encrypted_value = encrypt(session.session(), value, kms_arn)
         parameter['value'] = encrypted_value.decode('utf-8')
     else:
         print('Put new value to the parameter')

diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 
 setup(
     name="aws-ssm-secrets-cli",
-    version='0.0.14',
+    version='0.0.15',
     license='MIT',
     description = 'AWS Secret CLI for manage SSM SecureString and SecretsManager',
     long_description = open("README.md", "r").read(),