chore: attempt to be more informative for missing attributes

JohnVillalovos · JohnVillalovos · commit 644ef294b38c · 2021-11-23T14:25:21.000-08:00
A commonly reported issue from users on Gitter is that they get an AttributeError for an attribute that should be present. This is often caused due to the fact that they used the `list()` method to retrieve the object and objects retrieved this way often only have a subset of the full data. Add more details in the AttributeError message that explains the situation to users. This will hopefully allow them to resolve the issue. Update the FAQ in the docs to add a section discussing the issue. Closes #1138
diff --git a/docs/faq.rst b/docs/faq.rst
@@ -16,6 +16,18 @@ I cannot edit the merge request / issue I've just retrieved
     See the :ref:`merge requests example <merge_requests_examples>` and the
     :ref:`issues examples <issues_examples>`.
 
+.. _attribute_error_list:
+
+I get an ``AttributeError`` when accessing attributes of an object retrieved via a ``list()`` call.
+    Fetching a list of objects, doesn’t always include all attributes in the
+    objects. To retrieve an object with all attributes use a ``get()`` call.
+
+    Example with projects::
+
+        for projects in gl.projects.list():
+            # Retrieve project object with all attributes
+            project = gl.projects.get(project.id)
+
 How can I clone the repository of a project?
     python-gitlab doesn't provide an API to clone a project. You have to use a
     git library or call the ``git`` command.
diff --git a/gitlab/base.py b/gitlab/base.py
@@ -32,6 +32,11 @@
 ]
 
 
+_URL_ATTRIBUTE_ERROR = (
+    "https://python-gitlab.readthedocs.io/en/stable/faq.html#attribute-error-list"
+)
+
+
 class RESTObject(object):
     """Represents an object built from server data.
 
@@ -45,13 +50,20 @@ class RESTObject(object):
 
     _id_attr: Optional[str] = "id"
     _attrs: Dict[str, Any]
+    _list_created_object: bool  # Indicates if object was created from a list() action
     _module: ModuleType
     _parent_attrs: Dict[str, Any]
     _short_print_attr: Optional[str] = None
     _updated_attrs: Dict[str, Any]
     manager: "RESTManager"
 
-    def __init__(self, manager: "RESTManager", attrs: Dict[str, Any]) -> None:
+    def __init__(
+        self,
+        manager: "RESTManager",
+        attrs: Dict[str, Any],
+        *,
+        list_created_object: bool = False,
+    ) -> None:
         if not isinstance(attrs, dict):
             raise GitlabParsingError(
                 "Attempted to initialize RESTObject with a non-dictionary value: "
@@ -64,6 +76,7 @@ def __init__(self, manager: "RESTManager", attrs: Dict[str, Any]) -> None:
                 "_attrs": attrs,
                 "_updated_attrs": {},
                 "_module": importlib.import_module(self.__module__),
+                "_list_created_object": list_created_object,
             }
         )
         self.__dict__["_parent_attrs"] = self.manager.parent_attrs
@@ -107,7 +120,16 @@ def __getattr__(self, name: str) -> Any:
                 try:
                     return self.__dict__["_parent_attrs"][name]
                 except KeyError:
-                    raise AttributeError(name)
+                    message = name
+                    if self._list_created_object:
+                        message = (
+                            f"Unable to find attribute {name!r}. This object "
+                            f"({self.__class__}) was created via a list() call and "
+                            f"only a subset of the data may be present. To ensure all "
+                            f"data is present get the object using a get(object.id) "
+                            f"call. See {_URL_ATTRIBUTE_ERROR} for details."
+                        )
+                    raise AttributeError(message)
 
     def __setattr__(self, name: str, value: Any) -> None:
         self.__dict__["_updated_attrs"][name] = value
@@ -229,7 +251,7 @@ def __next__(self) -> RESTObject:
 
     def next(self) -> RESTObject:
         data = self._list.next()
-        return self._obj_cls(self.manager, data)
+        return self._obj_cls(self.manager, data, list_created_object=True)
 
     @property
     def current_page(self) -> int:
diff --git a/gitlab/mixins.py b/gitlab/mixins.py
@@ -240,7 +240,7 @@ def list(self, **kwargs: Any) -> Union[base.RESTObjectList, List[base.RESTObject
             assert self._obj_cls is not None
         obj = self.gitlab.http_list(path, **data)
         if isinstance(obj, list):
-            return [self._obj_cls(self, item) for item in obj]
+            return [self._obj_cls(self, item, list_created_object=True) for item in obj]
         else:
             return base.RESTObjectList(self, self._obj_cls, obj)
 
diff --git a/tests/unit/test_base.py b/tests/unit/test_base.py
@@ -90,6 +90,28 @@ def test_instantiate_non_dict(self, fake_gitlab, fake_manager):
         with pytest.raises(gitlab.exceptions.GitlabParsingError):
             FakeObject(fake_manager, ["a", "list", "fails"])
 
+    def test_missing_attribute_does_not_raise_custom(self, fake_gitlab, fake_manager):
+        """Ensure a missing attribute does not raise our custom error message
+        if the RESTObject was not created from a list"""
+        obj = FakeObject(manager=fake_manager, attrs={"foo": "bar"})
+        with pytest.raises(AttributeError) as excinfo:
+            obj.missing_attribute
+        assert str(excinfo.value) == "missing_attribute"
+        assert "was created via a list()" not in str(excinfo.value)
+        assert base._URL_ATTRIBUTE_ERROR not in str(excinfo.value)
+
+    def test_missing_attribute_from_list_raises_custom(self, fake_gitlab, fake_manager):
+        """Ensure a missing attribute raises our custom error message if the
+        RESTObject was created from a list"""
+        obj = FakeObject(
+            manager=fake_manager, attrs={"foo": "bar"}, list_created_object=True
+        )
+        with pytest.raises(AttributeError) as excinfo:
+            obj.missing_attribute
+        assert "missing_attribute" in str(excinfo.value)
+        assert "was created via a list()" in str(excinfo.value)
+        assert base._URL_ATTRIBUTE_ERROR in str(excinfo.value)
+
     def test_picklability(self, fake_manager):
         obj = FakeObject(fake_manager, {"foo": "bar"})
         original_obj_module = obj._module
