Add ability to set properties in subsections

This change adds new parameters to the `aws configure set`` command to
specify a sub-section for setting a property. These parameters are
analogous to the existing `--profile` parameter. A parameter will be
added for each sub-section type and take a value of the subsection name.
Following is the generic pattern for the `aws configure set` command:

```
aws configure set
  --<sub-section-type> <sub-section-name> \
  <property> <value>
```

For example, the following command should set the property `sso_region`
to the value `us-west-2` in the `sso-session` sub-section named
`my-sso-session`:

```
aws configure set
  --sso-session my-sso-session \
  sso_region us-west-2
```

Following is an example setting a nested property in a sub-section:

```
aws configure set \
  --<sub-section-type> <sub-section-name> \
  <nested-section>.<property> value
```

The only sub-section types allowed are `services` and `sso-session`.

Author: kdaily

awscli/customizations/configure/__init__.py

@@ -10,12 +10,22 @@
 # distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
+import re
 import string
 
 from awscli.compat import shlex
+from awscli.customizations.exceptions import ParamValidationError
+from awscli.customizations.utils import validate_mutually_exclusive
 
 NOT_SET = '<not set>'
 PREDEFINED_SECTION_NAMES = 'plugins'
+# A map between the command line parameter name and the Python object name
+# For allowed sub-section types
+SUBSECTION_TYPE_ALLOWLIST = {
+    'sso-session': 'sso_session',
+    'services': 'services',
+}
+
 _WHITESPACE = ' \t'
 
 
@@ -53,3 +63,27 @@ def get_section_header(section_type, section_name):
     if any(c in _WHITESPACE for c in section_name):
         section_name = shlex.quote(section_name)
     return f'{section_type} {section_name}'
+
+
+def get_subsection_from_args(args):
+    # Validate mutual exclusivity of sub-section type parameters
+    groups = [[param] for param in SUBSECTION_TYPE_ALLOWLIST.values()]
+    validate_mutually_exclusive(args, *groups)
+
+    subsection_name = None
+    subsection_type = None
+
+    for section_type, param_name in SUBSECTION_TYPE_ALLOWLIST.items():
+        if hasattr(args, param_name):
+            param_value = getattr(args, param_name)
+            if param_value is not None:
+                if not re.match(r"^\w", param_value):
+                    raise ParamValidationError(
+                        f"aws: [ERROR]: Invalid value for --{section_type}."
+                    )
+                subsection_name = param_value
+                subsection_type = section_type
+                break
+
+    return (subsection_type, subsection_name)
+

awscli/customizations/configure/set.py

@@ -15,7 +15,12 @@
 from awscli.customizations.commands import BasicCommand
 from awscli.customizations.configure.writer import ConfigFileWriter
 
-from . import PREDEFINED_SECTION_NAMES, profile_to_section
+from . import (
+    PREDEFINED_SECTION_NAMES,
+    get_section_header,
+    get_subsection_from_args,
+    profile_to_section
+)
 
 
 class ConfigureSetCommand(BasicCommand):
@@ -41,6 +46,20 @@ class ConfigureSetCommand(BasicCommand):
             'cli_type_name': 'string',
             'positional_arg': True,
         },
+        {
+            'name': 'sso-session',
+            'help_text': 'The name of the SSO session sub-section to configure.',
+            'action': 'store',
+            'cli_type_name': 'string',
+            'group_name': 'subsection',
+        },
+        {
+            'name': 'services',
+            'help_text': 'The name of the services sub-section to configure.',
+            'action': 'store',
+            'cli_type_name': 'string',
+            'group_name': 'subsection',
+        },
     ]
     # Any variables specified in this list will be written to
     # the ~/.aws/credentials file instead of ~/.aws/config.
@@ -60,10 +79,37 @@ def _get_config_file(self, path):
         config_path = self._session.get_config_variable(path)
         return os.path.expanduser(config_path)
 
+    def _set_subsection_property(self, section_type, section_name, varname, value):
+        if '.' in varname:
+            parts = varname.split('.')
+            if len(parts) > 2:
+                return 0
+
+            varname = parts[0]
+            value = {parts[1]: value}
+        
+        # Build update dict
+        updated_config = {
+            '__section__': get_section_header(section_type, section_name), 
+            varname: value
+        }
+
+        # Write to config file
+        config_filename = self._get_config_file('config_file')
+        self._config_writer.update_config(updated_config, config_filename)
+        
+        return 0
+    
     def _run_main(self, args, parsed_globals):
         varname = args.varname
         value = args.value
         profile = 'default'
+
+        section_type, section_name = get_subsection_from_args(args)
+        if section_type is not None:
+            return self._set_subsection_property(section_type, section_name, varname, value)
+
+        # Not in a sub-section, continue with previous profile logic.
         # Before handing things off to the config writer,
         # we need to find out three things:
         # 1. What section we're writing to (profile).

tests/functional/configure/test_configure.py

@@ -299,6 +299,80 @@ def test_get_nested_attribute(self):
         )
         self.assertEqual(stdout, "")
 
+    def test_set_with_subsection_no_name_provided(self):
+        _, stderr, _ = self.run_cmd(
+                [
+                    "configure",
+                    "set",
+                    "--sso-session",
+                    '',
+                    "space",
+                    "test",
+                ],
+                expected_rc=252
+            )
+        self.assertIn("Invalid value for --sso-session", stderr)
+
+    def test_set_deeply_nested_property_in_subsection_does_nothing(self):
+        self.set_config_file_contents(
+            "[services my-services]\n" "ec2 =\n" "    endpoint_url = localhost\n"
+        )
+
+        self.run_cmd(
+            [
+                "configure",
+                "set",
+                "--services",
+                'my-services',
+                "s3.express.endpoint_url",
+                "localhost",
+            ],
+            expected_rc=0
+        )
+        self.assertEqual(
+            "[services my-services]\n" "ec2 =\n" "    endpoint_url = localhost\n",
+            self.get_config_file_contents(),
+        )
+
+    def test_set_with_two_subsections_specified_results_in_error(self):
+        _, stderr, _ = self.run_cmd(
+                [
+                    "configure",
+                    "set",
+                    "--sso-session",
+                    'my-sso',
+                    "--services",
+                    'my-services',
+                    "space",
+                    "test",
+                ],
+                expected_rc=252
+            )
+        self.assertIn(
+            "cannot be specified when one of the following keys are also specified:",
+            stderr
+        )
+
+    def test_set_updates_existing_property_in_subsection(self):
+        self.set_config_file_contents(
+            "[sso-session my-sso-session]\n" "sso_region = us-west-2\n"
+        )
+
+        self.run_cmd(
+            [
+                "configure",
+                "set",
+                "--sso-session",
+                'my-sso-session',
+                "sso_region",
+                "eu-central-1",
+            ],
+            expected_rc=0
+        )
+        self.assertEqual(
+            "[sso-session my-sso-session]\n" "sso_region = eu-central-1\n",
+            self.get_config_file_contents(),
+        )
 
 class TestConfigureHasArgTable(unittest.TestCase):
     def test_configure_command_has_arg_table(self):

tests/unit/customizations/configure/test_set.py

@@ -205,3 +205,28 @@ def test_configure_set_with_profile_with_tab_dotted(self):
             {'__section__': "profile 'some\tprofile'", 'region': 'us-west-2'},
             'myconfigfile',
         )
+
+    def test_set_top_level_property_in_subsection(self):
+        set_command = ConfigureSetCommand(self.session, self.config_writer)
+        set_command(
+            args=['sso_region', 'us-west-2', '--sso-session', 'my-session'],
+            parsed_globals=None,
+        )
+        self.config_writer.update_config.assert_called_with(
+            {'__section__': 'sso-session my-session', 'sso_region': 'us-west-2'},
+            'myconfigfile',
+        )
+
+    def test_set_nested_property_in_subsection(self):
+        set_command = ConfigureSetCommand(self.session, self.config_writer)
+        set_command(
+            args=['--services', 'my-services', 's3.endpoint_url', 'http://localhost:4566'],
+            parsed_globals=None,
+        )
+        self.config_writer.update_config.assert_called_with(
+            {
+                '__section__': 'services my-services',
+                's3': {'endpoint_url': 'http://localhost:4566'},
+            },
+            'myconfigfile',
+        )