Adds ability to use "set_fields" with parameter names

Author: DrPaulSharp

RATapi/classlist.py

@@ -296,11 +296,15 @@ def extend(self, other: Sequence[T]) -> None:
         self._check_unique_name_fields(other)
         self.data.extend(other)
 
-    def set_fields(self, index: int, **kwargs) -> None:
+    def set_fields(self, index: Union[int, slice, str, T], **kwargs) -> None:
         """Assign the values of an existing object's attributes using keyword arguments."""
         self._validate_name_field(kwargs)
         pydantic_object = False
 
+        # Find index if name or object is supplied
+        if isinstance(index, (str, self._class_handle)):
+            index = self.index(index)
+
         if importlib.util.find_spec("pydantic"):
             # Pydantic is installed, so set up a context manager that will
             # suppress custom validation errors until all fields have been set.

tests/test_classlist.py

@@ -729,6 +729,7 @@ def test_extend_empty_classlist(extended_list: Sequence, one_name_class_list: Cl
     assert isinstance(extended_list[-1], class_list._class_handle)
 
 
+@pytest.mark.parametrize("index", [0, "Alice"])
 @pytest.mark.parametrize(
     ["new_values", "expected_classlist"],
     [
@@ -739,10 +740,12 @@ def test_extend_empty_classlist(extended_list: Sequence, one_name_class_list: Cl
         ),
     ],
 )
-def test_set_fields(two_name_class_list: ClassList, new_values: dict[str, Any], expected_classlist: ClassList) -> None:
+def test_set_fields(
+    two_name_class_list: ClassList, index: Union[int, str], new_values: dict[str, Any], expected_classlist: ClassList
+) -> None:
     """We should be able to set field values in an element of a ClassList using keyword arguments."""
     class_list = two_name_class_list
-    class_list.set_fields(0, **new_values)
+    class_list.set_fields(index, **new_values)
     assert class_list == expected_classlist