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

black format

Browse source
This commit is contained in:
gruebel 2019-11-01 07:14:03 +01:00
commit 78e2714496
5 changed files with 143 additions and 92 deletions
Download patch file Download diff file Expand all files Collapse all files

6
moto/iam/exceptions.py
View file

@ -127,13 +127,11 @@ class InvalidInput(RESTError):
code = 400
def __init__(self, message):
super(InvalidInput, self).__init__(
'InvalidInput', message)
super(InvalidInput, self).__init__("InvalidInput", message)
class NoSuchEntity(RESTError):
code = 404
def __init__(self, message):
super(NoSuchEntity, self).__init__(
'NoSuchEntity', message)
super(NoSuchEntity, self).__init__("NoSuchEntity", message)

99
moto/iam/models.py
View file

@ -654,12 +654,22 @@ class User(BaseModel):
class AccountPasswordPolicy(BaseModel):
def __init__(self, allow_change_password, hard_expiry, max_password_age, minimum_password_length,
password_reuse_prevention, require_lowercase_characters, require_numbers,
require_symbols, require_uppercase_characters):
def __init__(
self,
allow_change_password,
hard_expiry,
max_password_age,
minimum_password_length,
password_reuse_prevention,
require_lowercase_characters,
require_numbers,
require_symbols,
require_uppercase_characters,
):
self._errors = []
self._validate(max_password_age, minimum_password_length, password_reuse_prevention)
self._validate(
max_password_age, minimum_password_length, password_reuse_prevention
)
self.allow_users_to_change_password = allow_change_password
self.hard_expiry = hard_expiry
@ -675,35 +685,41 @@ class AccountPasswordPolicy(BaseModel):
def expire_passwords(self):
return True if self.max_password_age and self.max_password_age > 0 else False
def _validate(self, max_password_age, minimum_password_length, password_reuse_prevention):
def _validate(
self, max_password_age, minimum_password_length, password_reuse_prevention
):
if minimum_password_length > 128:
self._errors.append(self._format_error(
key='minimumPasswordLength',
value=minimum_password_length,
constraint='Member must have value less than or equal to 128'
))
self._errors.append(
self._format_error(
key="minimumPasswordLength",
value=minimum_password_length,
constraint="Member must have value less than or equal to 128",
)
)
if password_reuse_prevention and password_reuse_prevention > 24:
self._errors.append(self._format_error(
key='passwordReusePrevention',
value=password_reuse_prevention,
constraint='Member must have value less than or equal to 24'
))
self._errors.append(
self._format_error(
key="passwordReusePrevention",
value=password_reuse_prevention,
constraint="Member must have value less than or equal to 24",
)
)
if max_password_age and max_password_age > 1095:
self._errors.append(self._format_error(
key='maxPasswordAge',
value=max_password_age,
constraint='Member must have value less than or equal to 1095'
))
self._errors.append(
self._format_error(
key="maxPasswordAge",
value=max_password_age,
constraint="Member must have value less than or equal to 1095",
)
)
self._raise_errors()
def _format_error(self, key, value, constraint):
return 'Value "{value}" at "{key}" failed to satisfy constraint: {constraint}'.format(
constraint=constraint,
key=key,
value=value,
constraint=constraint, key=key, value=value,
)
def _raise_errors(self):
@ -713,9 +729,11 @@ class AccountPasswordPolicy(BaseModel):
errors = "; ".join(self._errors)
self._errors = [] # reset collected errors
raise ValidationError('{count} validation error{plural} detected: {errors}'.format(
count=count, plural=plural, errors=errors,
))
raise ValidationError(
"{count} validation error{plural} detected: {errors}".format(
count=count, plural=plural, errors=errors,
)
)
class IAMBackend(BaseBackend):
@ -1657,9 +1675,18 @@ class IAMBackend(BaseBackend):
def list_open_id_connect_providers(self):
return list(self.open_id_providers.keys())
def update_account_password_policy(self, allow_change_password, hard_expiry, max_password_age, minimum_password_length,
password_reuse_prevention, require_lowercase_characters, require_numbers,
require_symbols, require_uppercase_characters):
def update_account_password_policy(
self,
allow_change_password,
hard_expiry,
max_password_age,
minimum_password_length,
password_reuse_prevention,
require_lowercase_characters,
require_numbers,
require_symbols,
require_uppercase_characters,
):
self.account_password_policy = AccountPasswordPolicy(
allow_change_password,
hard_expiry,
@ -1669,18 +1696,24 @@ class IAMBackend(BaseBackend):
require_lowercase_characters,
require_numbers,
require_symbols,
require_uppercase_characters
require_uppercase_characters,
)
def get_account_password_policy(self):
if not self.account_password_policy:
raise NoSuchEntity('The Password Policy with domain name {} cannot be found.'.format(ACCOUNT_ID))
raise NoSuchEntity(
"The Password Policy with domain name {} cannot be found.".format(
ACCOUNT_ID
)
)
return self.account_password_policy
def delete_account_password_policy(self):
if not self.account_password_policy:
raise NoSuchEntity('The account policy with name PasswordPolicy cannot be found.')
raise NoSuchEntity(
"The account policy with name PasswordPolicy cannot be found."
)
self.account_password_policy = None

37
moto/iam/responses.py
View file

@ -839,20 +839,33 @@ class IamResponse(BaseResponse):
return template.render(open_id_provider_arns=open_id_provider_arns)
def update_account_password_policy(self):
allow_change_password = self._get_bool_param('AllowUsersToChangePassword', False)
hard_expiry = self._get_bool_param('HardExpiry')
max_password_age = self._get_int_param('MaxPasswordAge')
minimum_password_length = self._get_int_param('MinimumPasswordLength', 6)
password_reuse_prevention = self._get_int_param('PasswordReusePrevention')
require_lowercase_characters = self._get_bool_param('RequireLowercaseCharacters', False)
require_numbers = self._get_bool_param('RequireNumbers', False)
require_symbols = self._get_bool_param('RequireSymbols', False)
require_uppercase_characters = self._get_bool_param('RequireUppercaseCharacters', False)
allow_change_password = self._get_bool_param(
"AllowUsersToChangePassword", False
)
hard_expiry = self._get_bool_param("HardExpiry")
max_password_age = self._get_int_param("MaxPasswordAge")
minimum_password_length = self._get_int_param("MinimumPasswordLength", 6)
password_reuse_prevention = self._get_int_param("PasswordReusePrevention")
require_lowercase_characters = self._get_bool_param(
"RequireLowercaseCharacters", False
)
require_numbers = self._get_bool_param("RequireNumbers", False)
require_symbols = self._get_bool_param("RequireSymbols", False)
require_uppercase_characters = self._get_bool_param(
"RequireUppercaseCharacters", False
)
iam_backend.update_account_password_policy(
allow_change_password, hard_expiry, max_password_age, minimum_password_length,
password_reuse_prevention, require_lowercase_characters, require_numbers,
require_symbols, require_uppercase_characters)
allow_change_password,
hard_expiry,
max_password_age,
minimum_password_length,
password_reuse_prevention,
require_lowercase_characters,
require_numbers,
require_symbols,
require_uppercase_characters,
)
template = self.response_template(UPDATE_ACCOUNT_PASSWORD_POLICY_TEMPLATE)
return template.render()