示例#1
0
文件: arm.py 项目: zackliu/azure-cli
 def set_argument_value(namespace, arg, parts):
     existing_values = getattr(namespace, arg.name, None)
     if existing_values is None:
         existing_values = IterateValue()
         existing_values.append(parts[arg.type.settings['id_part']])
     else:
         if isinstance(existing_values, str):
             if not getattr(arg.type, 'configured_default_applied', None):
                 logger.warning(
                     "Property '%s=%s' being overriden by value '%s' from IDs parameter.",
                     arg.name, existing_values, parts[arg.type.settings['id_part']]
                 )
             existing_values = IterateValue()
         existing_values.append(parts[arg.type.settings['id_part']])
     setattr(namespace, arg.name, existing_values)
示例#2
0
            def set_argument_value(namespace, arg, parts):

                existing_values = getattr(namespace, arg.name, None)
                if existing_values is None:
                    existing_values = IterateValue()
                    existing_values.append(parts.get(arg.type.settings['id_part'], None))
                else:
                    if isinstance(existing_values, str):
                        if not getattr(arg.type, 'configured_default_applied', None):
                            logger.warning(
                                "Property '%s=%s' being overriden by value '%s' from IDs parameter.",
                                arg.name, existing_values, parts[arg.type.settings['id_part']]
                            )
                        existing_values = IterateValue()
                    existing_values.append(parts.get(arg.type.settings['id_part']))
                setattr(namespace, arg.name, existing_values)
示例#3
0
    def parse_ids_arguments(_, command, args):
        namespace = args
        cmd = namespace._cmd  # pylint: disable=protected-access

        # some commands have custom IDs and parsing. This will not work for that.
        if not ids_metadata.get(command, None):
            return

        ids = getattr(namespace, 'ids', getattr(namespace, '_ids', None))
        required_args = [
            cmd.arguments[x] for x in ids_metadata[command]['required']
        ]
        optional_args = [
            cmd.arguments[x] for x in ids_metadata[command]['optional']
        ]
        combined_args = required_args + optional_args

        if not ids:
            # ensure the required parameters are provided if --ids is not
            errors = [
                arg for arg in required_args
                if getattr(namespace, arg.name, None) is None
            ]
            if errors:
                missing_required = ' '.join(
                    (arg.options_list[0] for arg in errors))
                raise CLIError('({} | {}) are required'.format(
                    missing_required, '--ids'))
            return

        # show warning if names are used in conjunction with --ids
        other_values = {
            arg.name: {
                'arg': arg,
                'value': getattr(namespace, arg.name, None)
            }
            for arg in combined_args
        }
        for _, data in other_values.items():
            if data['value'] and not getattr(data['value'], 'is_default',
                                             None):
                logger.warning(
                    "option '%s' will be ignored due to use of '--ids'.",
                    data['arg'].type.settings['options_list'][0])

        # create the empty lists, overwriting any values that may already be there
        for arg in combined_args:
            setattr(namespace, arg.name, IterateValue())

        def assemble_json(ids):
            lcount = 0
            lind = None
            for i, line in enumerate(ids):
                if line == '[':
                    if lcount == 0:
                        lind = i
                    lcount += 1
                elif line == ']':
                    lcount -= 1
                    # final closed set of matching brackets
                    if lcount == 0:
                        left = lind
                        right = i + 1
                        l_comp = ids[:left]
                        m_comp = [''.join(ids[left:right])]
                        r_comp = ids[right:]
                        ids = l_comp + m_comp + r_comp
                        return assemble_json(ids)
            # base case--no more merging required
            return ids

        # reassemble JSON strings from bash
        ids = assemble_json(ids)

        # expand the IDs into the relevant fields
        full_id_list = []
        for val in ids:
            try:
                # support piping values from JSON. Does not require use of --query
                json_vals = json.loads(val)
                if not isinstance(json_vals, list):
                    json_vals = [json_vals]
                for json_val in json_vals:
                    if isinstance(json_val, dict) and 'id' in json_val:
                        full_id_list += [json_val['id']]
            except ValueError:
                # supports piping of --ids to the command when using TSV. Requires use of --query
                full_id_list = full_id_list + val.splitlines()
        if full_id_list:
            setattr(namespace, '_ids', full_id_list)

        from azure.mgmt.core.tools import parse_resource_id, is_valid_resource_id
        for val in full_id_list:
            if not is_valid_resource_id(val):
                raise CLIError('invalid resource ID: {}'.format(val))
            # place the ID parts into the correct property lists
            parts = parse_resource_id(val)
            for arg in combined_args:
                id_part = arg.type.settings.get('id_part')
                id_value = parts.get(id_part, None)
                if id_value is None:
                    argument_name = arg.type.settings.get('options_list')[0]
                    raise CLIError(
                        "Argument {arg_name} cannot be derived from ID {id}. "
                        "Please provide a complete resource ID "
                        "containing all information of '{group_name}' "
                        "arguments. ".format(id=val,
                                             arg_name=argument_name,
                                             group_name=arg.arg_group))
                getattr(namespace, arg.name).append(id_value)

        # support deprecating --ids
        deprecate_info = cmd.arguments['ids'].type.settings.get(
            'deprecate_info')
        if deprecate_info:
            if not hasattr(namespace, '_argument_deprecations'):
                setattr(namespace, '_argument_deprecations', [deprecate_info])
            else:
                namespace._argument_deprecations.append(deprecate_info)  # pylint: disable=protected-access
示例#4
0
    def parse_ids_arguments(_, command, args):

        namespace = args
        cmd = namespace._cmd  # pylint: disable=protected-access

        # some commands have custom IDs and parsing. This will not work for that.
        if not ids_metadata.get(command, None):
            return

        ids = getattr(namespace, 'ids', getattr(namespace, '_ids', None))
        required_args = [
            cmd.arguments[x] for x in ids_metadata[command]['required']
        ]
        optional_args = [
            cmd.arguments[x] for x in ids_metadata[command]['optional']
        ]
        combined_args = required_args + optional_args

        if not ids:
            # ensure the required parameters are provided if --ids is not
            errors = [
                arg for arg in required_args
                if getattr(namespace, arg.name, None) is None
            ]
            if errors:
                missing_required = ' '.join(
                    (arg.options_list[0] for arg in errors))
                raise ValueError('({} | {}) are required'.format(
                    missing_required, '--ids'))
            return

        # show warning if names are used in conjunction with --ids
        other_values = {
            arg.name: {
                'arg': arg,
                'value': getattr(namespace, arg.name, None)
            }
            for arg in combined_args
        }
        for _, data in other_values.items():
            if data['value'] and not getattr(data['value'], 'is_default',
                                             None):
                logger.warning(
                    "option '%s' will be ignored due to use of '--ids'.",
                    data['arg'].type.settings['options_list'][0])

        # create the empty lists, overwriting any values that may already be there
        for arg in combined_args:
            setattr(namespace, arg.name, IterateValue())

        # expand the IDs into the relevant fields
        full_id_list = []
        for val in ids:
            try:
                # support piping values from JSON. Does not require use of --query
                json_vals = json.loads(val)
                if not isinstance(json_vals, list):
                    json_vals = [json_vals]
                for json_val in json_vals:
                    if 'id' in json_val:
                        full_id_list += [json_val['id']]
            except ValueError:
                # supports piping of --ids to the command when using TSV. Requires use of --query
                full_id_list = full_id_list + val.splitlines()

        for val in full_id_list:
            if not is_valid_resource_id(val):
                raise CLIError('invalid resource ID: {}'.format(val))
            # place the ID parts into the correct property lists
            parts = parse_resource_id(val)
            for arg in combined_args:
                getattr(namespace,
                        arg.name).append(parts[arg.type.settings['id_part']])

        # support deprecating --ids
        deprecate_info = cmd.arguments['ids'].type.settings.get(
            'deprecate_info')
        if deprecate_info:
            if not hasattr(namespace, '_argument_deprecations'):
                setattr(namespace, '_argument_deprecations', [deprecate_info])
            else:
                namespace._argument_deprecations.append(deprecate_info)  # pylint: disable=protected-access