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)