Add feature to Graph model : disable_abstract_fields

FloChehab · Sep 9, 2018 · 30c1a80 · 30c1a80
1 parent 7d262c3
commit 30c1a80
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/django_extensions/management/commands/graph_models.py b/django_extensions/management/commands/graph_models.py
@@ -58,6 +58,12 @@ def __init__(self, *args, **kwargs):
                 'dest': 'disable_fields',
                 'help': 'Do not show the class member fields',
             },
+            '--disable-abstract-fields': {
+                'action': 'store_true',
+                'default': False,
+                'dest': 'disable_abstract_fields',
+                'help': 'Do not show the class member fields that were inherited',
+            },
             '--group-models -g': {
                 'action': 'store_true',
                 'default': False,

diff --git a/django_extensions/management/modelviz.py b/django_extensions/management/modelviz.py
@@ -60,6 +60,7 @@ def __init__(self, app_labels, **kwargs):
         self.graphs = []
         self.cli_options = kwargs.get('cli_options', None)
         self.disable_fields = kwargs.get('disable_fields', False)
+        self.disable_abstract_fields = kwargs.get('disable_abstract_fields', False)
         self.include_models = parse_file_or_list(
             kwargs.get('include_models', "")
         )
@@ -103,6 +104,7 @@ def get_graph_data(self, as_json=False):
             'created_at': now.strftime("%Y-%m-%d %H:%M"),
             'cli_options': self.cli_options,
             'disable_fields': self.disable_fields,
+            'disable_abstract_fields': self.disable_abstract_fields,
             'use_subgraph': self.use_subgraph,
         }
 

diff --git a/django_extensions/templates/django_extensions/graph_models/label.dot b/django_extensions/templates/django_extensions/graph_models/label.dot
@@ -18,11 +18,13 @@
     {{ model.label }}{% if model.abstracts %}<BR/>&lt;<FONT FACE="Helvetica Italic">{{ model.abstracts|join:"," }}</FONT>&gt;{% endif %}
     </FONT></TD></TR>
   {% if not disable_fields %}{% for field in model.fields %}
+  {% if not disable_abstract_fields or not field.abstract %}
     <TR><TD ALIGN="LEFT" BORDER="0">
     <FONT {% if not field.primary_key and field.blank %}COLOR="#7B7B7B" {% endif %}FACE="Helvetica {% if field.abstract %}Italic{% endif %}{% if field.relation or field.primary_key %}Bold{% endif %}">{{ field.label }}</FONT>
     </TD><TD ALIGN="LEFT">
     <FONT {% if not field.primary_key and field.blank %}COLOR="#7B7B7B" {% endif %}FACE="Helvetica {% if field.abstract %}Italic{% endif %}{% if field.relation or field.primary_key %}Bold{% endif %}">{{ field.type }}</FONT>
     </TD></TR>
+  {% endif %}
   {% endfor %}{% endif %}
     </TABLE>
     >]