add support for datalcass fields with no default value

Author: csaba.nemes

sphinx_automodapi/autodoc_enhancements.py

@@ -1,7 +1,7 @@
 """
 Miscellaneous enhancements to help autodoc along.
 """
-
+import dataclasses
 import inspect
 import sys
 import types
@@ -68,7 +68,12 @@ def type_object_attrgetter(obj, attr, *defargs):
                 return base.__dict__[attr]
             break
 
-    return getattr(obj, attr, *defargs)
+    try:
+        return getattr(obj, attr, *defargs)
+    except AttributeError:
+        # for dataclasses, get the attribute from the __dataclass_fields__
+        if dataclasses.is_dataclass(obj):
+            return obj.__dataclass_fields__[attr].name
 
 
 def setup(app):

sphinx_automodapi/automodsumm.py

@@ -88,6 +88,7 @@ class members that are inherited from a base class. This value can be
 import os
 import re
 import io
+import dataclasses
 
 from sphinx.util import logging
 from sphinx.ext.autosummary import Autosummary
@@ -551,11 +552,22 @@ def get_members_class(obj, typ, include_public=[],
                     else:
                         names = getattr(obj, '__dict__').keys()
 
+                    # add dataclass_field names for dataclass classes
+                    if dataclasses.is_dataclass(obj):
+                        dataclass_fieldnames = getattr(obj, '__dataclass_fields__').keys()
+                        names = list(set(list(names) + list(dataclass_fieldnames)))
+
                 for name in names:
                     try:
                         documenter = get_documenter(app, safe_getattr(obj, name), obj)
                     except AttributeError:
-                        continue
+                        # for dataclasses try to get the attribute from the __dataclass_fields__
+                        if dataclasses.is_dataclass(obj):
+                            try:
+                                attr = obj.__dataclass_fields__[name]
+                                documenter = get_documenter(app, attr, obj)
+                            except KeyError:
+                                continue
                     if typ is None or documenter.objtype == typ:
                         items.append(name)
                     elif typ == 'attribute' and documenter.objtype == 'property':