catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add update_access_key endpoint (#1423)

Browse source
This commit is contained in:
Brian Pandola 2018-01-10 15:29:08 -08:00 • committed by Jack Danger
commit 350cf9257e
3 changed files with 38 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
moto/iam/models.py
View file

@ -349,6 +349,14 @@ class User(BaseModel):
raise IAMNotFoundException(
"Key {0} not found".format(access_key_id))
def update_access_key(self, access_key_id, status):
for key in self.access_keys:
if key.access_key_id == access_key_id:
key.status = status
break
else:
raise IAMNotFoundException("The Access Key with id {0} cannot be found".format(access_key_id))
def get_cfn_attribute(self, attribute_name):
from moto.cloudformation.exceptions import UnformattedGetAttTemplateException
if attribute_name == 'Arn':
@ -817,6 +825,10 @@ class IAMBackend(BaseBackend):
key = user.create_access_key()
return key
def update_access_key(self, user_name, access_key_id, status):
user = self.get_user(user_name)
user.update_access_key(access_key_id, status)
def get_all_access_keys(self, user_name, marker=None, max_items=None):
user = self.get_user(user_name)
keys = user.get_all_access_keys()

8
moto/iam/responses.py
View file

@ -440,6 +440,14 @@ class IamResponse(BaseResponse):
template = self.response_template(CREATE_ACCESS_KEY_TEMPLATE)
return template.render(key=key)
def update_access_key(self):
user_name = self._get_param('UserName')
access_key_id = self._get_param('AccessKeyId')
status = self._get_param('Status')
iam_backend.update_access_key(user_name, access_key_id, status)
template = self.response_template(GENERIC_EMPTY_TEMPLATE)
return template.render(name='UpdateAccessKey')
def list_access_keys(self):
user_name = self._get_param('UserName')