Add @doc docstring option for properties

1 files changed, 4 insertions, 4 deletions

diff --git a/grapher.py b/grapher.py
index aeaa447..a54bcc7 100644
--- a/grapher.py
+++ b/grapher.py
@@ -143,8 +143,10 @@ for doc in docs:
         epydoc.docstringparser.parse_docstring (var_var, None)
         if options.debug:
           print var_name, var_var.name, get_var_type (var_var)
-        var_var_name = str (var_var.name.replace (prefix, ""))
-        vars_dict[var_name].append ((var_var_name, get_var_type (var_var)))
+        var_type = get_var_type (var_var)
+        if all_properties or "@doc" in str (var_var.docstring) or var_type:
+            var_var_name = str (var_var.name.replace (prefix, ""))
+            vars_dict[var_name].append ((var_var_name, get_var_type (var_var)))
       elif type (var_var.value) in (epydoc.apidoc.RoutineDoc,
                                     epydoc.apidoc.StaticMethodDoc):
         if no_method:
@@ -245,8 +247,6 @@ for class_name in classes:
   if is_excluded (class_name):
     continue
   for (var_name, type_name) in vars_dict[class_name]:
-    if not all_properties and (not type_name or is_excluded (type_name)):
-      continue
     if class_name not in var_fields_dict:
         var_fields_dict[class_name] = []
         method_fields_dict[class_name] = []