def disable_rls(args):
if not ask.confirm(
'Are you sure you want to disable row level security on %s?' %
args.entity):
return
io.start('Disabling row level security on entity type %s' %
highlight(args.entity))
ensure_resource_exists(args.entity, ResourceType.ENTITY_TYPE)
security.disable_row_level_security(args.entity)
def _input_script_name():
file_name = ''
while not file_name:
name = mcmd.io.ask.input_('Supply the name of the script:')
if context().get_scripts_folder().joinpath(name).exists():
overwrite = confirm('%s already exists. Overwrite?' % name)
if overwrite:
file_name = name
else:
file_name = name
return file_name
def _make_member_of_group_role(user: User, role: Role):
group = role.group
membership = _get_group_membership(user, group)
if membership:
if membership.role.name == role.name:
io.info('User {} is already {} of group {}'.format(
highlight(user.username), highlight(role.label),
highlight(group.name)))
else:
update = ask.confirm(
'User {} is {} of group {}. Do you want to update his/her role to {}?'
.format(user.username, membership.role.label, group.name,
role.label))
if update:
_update_group_role_membership(user, group, role)
else:
_add_group_role_membership(user, group, role)
def __ask_input(input_: Input, state: _ScriptExecutionState):
if input_.name in state.values:
return
message = input_.message.render(
state.values) if input_.message else input_.name
if input_.type == InputType.TEXT:
value = ask.input_(message, required=True)
elif input_.type == InputType.BOOL:
value = ask.confirm(message)
elif input_.type == InputType.PASS:
value = ask.password(message, required=True)
else:
raise ValueError('Invalid InputType: {}'.format(input_.type))
state.values[input_.name] = value
def _switch_to_new_host(url):
if ask.confirm("Do you want to switch to the new host?"):
io.start("Switching to host {}".format(highlight(url)))
config.set_host(url)